add routers and schemas

Author: ks6088ts

backend/fastapi.py

@@ -2,6 +2,7 @@
 from fastapi.openapi.utils import get_openapi
 
 from backend.routers import azure_openai as azure_openai_router
+from backend.routers import azure_storage as azure_storage_router
 from backend.routers import document_intelligence as document_intelligence_router
 
 app = FastAPI(
@@ -10,6 +11,7 @@
 
 app.include_router(azure_openai_router.router)
 app.include_router(document_intelligence_router.router)
+app.include_router(azure_storage_router.router)
 
 
 def custom_openapi():

backend/internals/azure_storage.py

@@ -19,49 +19,46 @@ def get_blob_service_client(self) -> BlobServiceClient:
 
     def upload_blob_stream(
         self,
-        container_name: str,
         blob_name: str,
         stream: bytes,
     ):
         blob_service_client = self.get_blob_service_client()
         blob_client = blob_service_client.get_blob_client(
-            container=container_name,
+            container=self.settings.azure_storage_blob_container_name,
             blob=blob_name,
         )
         blob_client.upload_blob(stream, overwrite=True)
-        logger.info(f"Uploaded blob {blob_name} to container {container_name}")
+        logger.info(f"Uploaded blob {blob_name} to container {self.settings.azure_storage_blob_container_name}")
 
     def download_blob_stream(
         self,
-        container_name: str,
         blob_name: str,
     ) -> bytes:
         blob_service_client = self.get_blob_service_client()
         blob_client = blob_service_client.get_blob_client(
-            container=container_name,
+            container=self.settings.azure_storage_blob_container_name,
             blob=blob_name,
         )
         stream = blob_client.download_blob().readall()
-        logger.info(f"Downloaded blob {blob_name} from container {container_name}")
+        logger.info(f"Downloaded blob {blob_name} from container {self.settings.azure_storage_blob_container_name}")
         return stream
 
     def delete_blob(
         self,
-        container_name: str,
         blob_name: str,
     ):
         blob_service_client = self.get_blob_service_client()
         blob_client = blob_service_client.get_blob_client(
-            container=container_name,
+            container=self.settings.azure_storage_blob_container_name,
             blob=blob_name,
         )
         blob_client.delete_blob()
-        logger.info(f"Deleted blob {blob_name} from container {container_name}")
+        logger.info(f"Deleted blob {blob_name} from container {self.settings.azure_storage_blob_container_name}")
 
     def list_blobs(
         self,
-        container_name: str,
     ) -> list:
         blob_service_client = self.get_blob_service_client()
-        container_client = blob_service_client.get_container_client(container_name)
-        return container_client.list_blobs()
+        container_client = blob_service_client.get_container_client(self.settings.azure_storage_blob_container_name)
+        logger.info(f"Listed blobs in container {self.settings.azure_storage_blob_container_name}")
+        return [blob.name for blob in container_client.list_blobs()]

backend/routers/azure_storage.py

@@ -0,0 +1,78 @@
+from logging import getLogger
+
+from fastapi import APIRouter, UploadFile, status
+from fastapi.responses import JSONResponse
+
+from backend.internals import azure_storage
+from backend.schemas import azure_storage as azure_storage_schemas
+from backend.settings.azure_storage import Settings as AzureStorageSettings
+
+logger = getLogger(__name__)
+blob_storage_client = azure_storage.BlobStorageClient(
+    settings=AzureStorageSettings(),
+)
+
+router = APIRouter(
+    prefix="/azure_storage",
+    tags=["azure_storage"],
+    responses={404: {"description": "Not found"}},
+)
+
+
+@router.post(
+    "/blobs/upload/",
+    response_model=azure_storage_schemas.BlobUploadResponse,
+    status_code=200,
+)
+async def upload_blob(
+    file: UploadFile,
+    blob_name: str,
+):
+    try:
+        content = await file.read()
+        blob_storage_client.upload_blob_stream(
+            blob_name=blob_name,
+            stream=content,
+        )
+    except Exception as e:
+        logger.error(f"Failed to upload blob: {e}")
+        raise
+    return azure_storage_schemas.BlobUploadResponse(
+        blob_name=blob_name,
+    )
+
+
+@router.delete(
+    "/blobs/delete/",
+    status_code=200,
+)
+async def delete_blob(
+    blob_name: str,
+):
+    try:
+        blob_storage_client.delete_blob(
+            blob_name=blob_name,
+        )
+    except Exception as e:
+        logger.error(f"Failed to delete blob: {e}")
+        raise
+    return JSONResponse(
+        status_code=status.HTTP_200_OK,
+        content={"blob_name": blob_name},
+    )
+
+
+@router.get(
+    "/blobs/",
+    status_code=200,
+)
+async def list_blobs():
+    try:
+        blobs = blob_storage_client.list_blobs()
+    except Exception as e:
+        logger.error(f"Failed to upload blob: {e}")
+        raise
+    return JSONResponse(
+        status_code=status.HTTP_200_OK,
+        content={"blobs": blobs},
+    )

backend/schemas/azure_storage.py

@@ -0,0 +1,9 @@
+from logging import getLogger
+
+from pydantic import BaseModel
+
+logger = getLogger(__name__)
+
+
+class BlobUploadResponse(BaseModel):
+    blob_name: str

backend/settings/azure_storage.py

@@ -4,7 +4,7 @@
 class Settings(BaseSettings):
     azure_storage_account_name: str = "<account-name>"
     azure_storage_sas_token: str = "<sas-token>"
-    azure_storage_container_name: str = "<container-name>"
+    azure_storage_blob_container_name: str = "<blob-container-name>"
 
     model_config = SettingsConfigDict(
         env_file="azure_storage.env",