fix: pre commit fixes

Author: bidhan-nagarro

api/data_ingestion/api.py

@@ -2,11 +2,6 @@
 import sys
 from datetime import timedelta
 
-from fastapi import FastAPI
-from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import FileResponse, ORJSONResponse
-from starlette.middleware.sessions import SessionMiddleware
-
 from data_ingestion.constants import __version__
 from data_ingestion.db.primary import get_db_context
 from data_ingestion.internal.auth import azure_scheme, local_auth_bypass
@@ -26,6 +21,10 @@
     utils,
 )
 from data_ingestion.settings import DeploymentEnvironment, initialize_sentry, settings
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import FileResponse, ORJSONResponse
+from starlette.middleware.sessions import SessionMiddleware
 
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.DEBUG)
@@ -90,11 +89,10 @@ async def _ensure_local_dev_user():
     """Create the local dev user with Admin role if it doesn't already have it."""
     from uuid import uuid4
 
+    from data_ingestion.models import Role, User
     from sqlalchemy import select
     from sqlalchemy.orm import selectinload
 
-    from data_ingestion.models import Role, User
-
     local_email = "dev@example.com"
 
     async with get_db_context() as db:

api/data_ingestion/cache/cache.py

@@ -1,9 +1,8 @@
 from contextlib import AbstractAsyncContextManager, asynccontextmanager
 
 import redis.asyncio as redis
-from loguru import logger
-
 from data_ingestion.settings import settings
+from loguru import logger
 
 pool = redis.ConnectionPool.from_url(settings.REDIS_CACHE_URL)
 

api/data_ingestion/celery.py

@@ -1,7 +1,6 @@
 from datetime import timedelta
 
 from celery import Celery
-
 from data_ingestion.settings import settings
 
 celery = Celery(

api/data_ingestion/constants.py

@@ -1,11 +1,10 @@
 import tomllib
 from functools import lru_cache
 
-from pydantic import computed_field
-from pydantic_settings import BaseSettings
-
 from data_ingestion.internal.utils import megabytes_to_bytes
 from data_ingestion.settings import settings
+from pydantic import computed_field
+from pydantic_settings import BaseSettings
 
 
 class Constants(BaseSettings):

api/data_ingestion/db/primary.py

@@ -5,14 +5,13 @@
     contextmanager,
 )
 
+from data_ingestion.settings import settings
 from loguru import logger
 from sqlalchemy import create_engine
 from sqlalchemy.exc import DatabaseError
 from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
 from sqlalchemy.orm import Session, sessionmaker
 
-from data_ingestion.settings import settings
-
 engine = create_async_engine(
     settings.ASYNC_DATABASE_URL,
     echo=not settings.IN_PRODUCTION,

api/data_ingestion/db/trino.py

@@ -1,12 +1,11 @@
 from contextlib import AbstractContextManager, contextmanager
 
+from data_ingestion.settings import settings
 from loguru import logger
 from sqlalchemy import create_engine
 from sqlalchemy.exc import DatabaseError
 from sqlalchemy.orm import Session, sessionmaker
 
-from data_ingestion.settings import settings
-
 engine = create_engine(
     settings.TRINO_URL,
     echo=not settings.IN_PRODUCTION,

api/data_ingestion/internal/auth.py

@@ -1,11 +1,11 @@
 from dataclasses import dataclass, field
 
+from data_ingestion.settings import settings
 from fastapi.security import HTTPBearer
 from fastapi_azure_auth import B2CMultiTenantAuthorizationCodeBearer
 from msgraph import GraphServiceClient
 
 from azure.identity import ClientSecretCredential
-from data_ingestion.settings import settings
 
 
 @dataclass

api/data_ingestion/internal/data_quality_checks.py

@@ -2,15 +2,15 @@
 from io import BytesIO, StringIO
 
 import pandas as pd
+from data_ingestion.internal.storage import storage_client
+from data_ingestion.utils.data_quality import process_n_columns
 from fastapi import (
     HTTPException,
     status,
 )
 from loguru import logger
 
 from azure.storage.blob import BlobProperties
-from data_ingestion.internal.storage import storage_client
-from data_ingestion.utils.data_quality import process_n_columns
 
 
 def get_data_quality_summary(dq_report_path: str):

api/data_ingestion/internal/email.py

@@ -1,11 +1,6 @@
 from typing import Any
 
 import requests
-from fastapi.encoders import jsonable_encoder
-from loguru import logger
-from mailjet_rest import Client
-from requests import HTTPError, JSONDecodeError
-
 from data_ingestion.schemas.email import (
     DataCheckSuccessRenderRequest,
     DqReportRenderRequest,
@@ -15,6 +10,10 @@
 )
 from data_ingestion.schemas.invitation import InviteEmailRenderRequest
 from data_ingestion.settings import DeploymentEnvironment, settings
+from fastapi.encoders import jsonable_encoder
+from loguru import logger
+from mailjet_rest import Client
+from requests import HTTPError, JSONDecodeError
 
 
 def send_email_base(

api/data_ingestion/internal/groups.py

@@ -1,6 +1,15 @@
 import json
 
 import requests
+from data_ingestion.internal.utils import chunks
+from data_ingestion.models.users import Role
+from data_ingestion.schemas.group import (
+    CreateGroupRequest,
+    GraphGroup,
+    ModifyUserAccessRequest,
+    UpdateGroupRequest,
+)
+from data_ingestion.schemas.user import GraphUser
 from fastapi import HTTPException
 from fastapi.encoders import jsonable_encoder
 from msgraph.generated.groups.groups_request_builder import (
@@ -14,16 +23,6 @@
 from sqlalchemy import select
 from sqlalchemy.ext.asyncio import AsyncSession
 
-from data_ingestion.internal.utils import chunks
-from data_ingestion.models.users import Role
-from data_ingestion.schemas.group import (
-    CreateGroupRequest,
-    GraphGroup,
-    ModifyUserAccessRequest,
-    UpdateGroupRequest,
-)
-from data_ingestion.schemas.user import GraphUser
-
 from .auth import graph_client, graph_credentials