trovesuite 1.0.1__py3-none-any.whl → 1.0.31__py3-none-any.whl

trovesuite/notification/notification_controller.py

@@ -0,0 +1,21 @@
+from .notification_write_dto import (
+    NotificationEmailControllerWriteDto,
+    NotificationSMSControllerWriteDto
+)
+from .notification_read_dto import (
+    NotificationEmailControllerReadDto,
+    NotificationSMSControllerReadDto
+)
+from .notification_service import NotificationService
+from ..entities.sh_response import Respons
+from fastapi import APIRouter
+
+notification_router = APIRouter(tags=["Notification"])
+
+@notification_router.post("/send_email", response_model=Respons[NotificationEmailControllerReadDto])
+async def send_email(data: NotificationEmailControllerWriteDto):
+    return NotificationService.send_email(data=data)
+
+@notification_router.post("/send_sms", response_model=Respons[NotificationSMSControllerReadDto])
+async def send_sms(data: NotificationSMSControllerWriteDto):
+    return NotificationService.send_sms(data=data)

trovesuite/notification/notification_read_dto.py

@@ -0,0 +1,21 @@
+from .notification_base import (
+    NotificationEmailBase,
+    NotificationSMSBase
+)
+from pydantic import BaseModel
+
+# EMAIL Notification
+
+class NotificationEmailControllerReadDto(NotificationEmailBase):
+    pass
+
+class NotificationEmailServiceReadDto(BaseModel):
+    pass
+
+# SMS Notification
+
+class NotificationSMSControllerReadDto(NotificationSMSBase):
+    pass
+
+class NotificationSMSServiceReadDto(BaseModel):
+    pass

trovesuite/notification/notification_service.py

@@ -0,0 +1,73 @@
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+from ..entities.sh_response import Respons
+from .notification_read_dto import (
+    NotificationEmailServiceReadDto,
+    NotificationSMSServiceReadDto
+)
+from .notification_write_dto import (
+    NotificationEmailServiceWriteDto,
+    NotificationSMSServiceWriteDto
+)
+
+class NotificationService:
+
+    @staticmethod
+    def send_email(data: NotificationEmailServiceWriteDto) -> Respons[NotificationEmailServiceReadDto]:
+        """
+        Send an email (single or multiple recipients) via Gmail SMTP.
+        Supports both plain text and HTML email bodies.
+        """
+
+        # Extract input data
+        receiver_email = data.receiver_email
+        text_message = data.text_message
+        html_message = getattr(data, "html_message", None)
+        sender_email = data.sender_email
+        password = data.password
+        subject = data.subject
+
+        # Allow single email or list
+        if isinstance(receiver_email, str):
+            receiver_email = [receiver_email]
+
+        # Create the email container
+        message = MIMEMultipart("alternative")
+        message["From"] = sender_email
+        message["To"] = ", ".join(receiver_email)
+        message["Subject"] = subject
+
+        # Attach plain text
+        message.attach(MIMEText(text_message, "plain"))
+
+        # Attach HTML if provided
+        if html_message:
+            message.attach(MIMEText(html_message, "html"))
+
+        try:
+            with smtplib.SMTP_SSL("smtp.gmail.com", 465) as server:
+                server.login(sender_email, password)
+                server.sendmail(sender_email, receiver_email, message.as_string())
+
+            return Respons[NotificationEmailServiceReadDto](
+                detail=f"Email successfully sent to {len(receiver_email)} recipient(s)",
+                error=None,
+                data=[],
+                status_code=200,
+                success=True,
+            )
+
+        except Exception as e:
+            print(e)
+            return Respons[NotificationEmailServiceReadDto](
+                detail="An error occurred while sending the email",
+                error=str(e),
+                data=[],
+                status_code=500,
+                success=False,
+            )
+
+    @staticmethod
+    def send_sms(data: NotificationSMSServiceWriteDto) -> Respons[NotificationSMSServiceReadDto]:
+        pass

trovesuite/notification/notification_write_dto.py

@@ -0,0 +1,21 @@
+from pydantic import BaseModel
+from .notification_base import (
+    NotificationEmailBase,
+    NotificationSMSBase
+)
+
+# Email Notification
+
+class NotificationEmailControllerWriteDto(NotificationEmailBase):
+    pass
+
+class NotificationEmailServiceWriteDto(NotificationEmailControllerWriteDto):
+    pass
+
+# SMS Notification
+
+class NotificationSMSControllerWriteDto(NotificationSMSBase):
+    pass
+
+class NotificationSMSServiceWriteDto(BaseModel):
+    pass

trovesuite/storage/__init__.py

@@ -0,0 +1,42 @@
+"""
+TroveSuite Storage Service
+
+Provides Azure Storage blob management capabilities for TroveSuite applications.
+Includes container creation, file upload/download/update/delete, and presigned URL generation.
+"""
+
+from .storage_service import StorageService
+from .storage_write_dto import (
+    StorageContainerCreateServiceWriteDto,
+    StorageFileUploadServiceWriteDto,
+    StorageFileUpdateServiceWriteDto,
+    StorageFileDeleteServiceWriteDto,
+    StorageFileDownloadServiceWriteDto,
+    StorageFileUrlServiceWriteDto
+)
+from .storage_read_dto import (
+    StorageContainerCreateServiceReadDto,
+    StorageFileUploadServiceReadDto,
+    StorageFileUpdateServiceReadDto,
+    StorageFileDeleteServiceReadDto,
+    StorageFileDownloadServiceReadDto,
+    StorageFileUrlServiceReadDto
+)
+
+__all__ = [
+    "StorageService",
+    # Write DTOs
+    "StorageContainerCreateServiceWriteDto",
+    "StorageFileUploadServiceWriteDto",
+    "StorageFileUpdateServiceWriteDto",
+    "StorageFileDeleteServiceWriteDto",
+    "StorageFileDownloadServiceWriteDto",
+    "StorageFileUrlServiceWriteDto",
+    # Read DTOs
+    "StorageContainerCreateServiceReadDto",
+    "StorageFileUploadServiceReadDto",
+    "StorageFileUpdateServiceReadDto",
+    "StorageFileDeleteServiceReadDto",
+    "StorageFileDownloadServiceReadDto",
+    "StorageFileUrlServiceReadDto",
+]

trovesuite/storage/storage_base.py

@@ -0,0 +1,63 @@
+from typing import Optional
+from pydantic import BaseModel
+
+
+class StorageConnectionBase(BaseModel):
+    """Base model for Azure Storage connection using Managed Identity"""
+    storage_account_url: str  # e.g., https://<account-name>.blob.core.windows.net
+    container_name: str
+    managed_identity_client_id: Optional[str] = None  # Optional: For user-assigned managed identity
+
+
+class StorageFileUploadBase(BaseModel):
+    """Base model for file upload operations"""
+    storage_account_url: str
+    container_name: str
+    file_content: bytes
+    blob_name: str
+    directory_path: Optional[str] = None
+    content_type: Optional[str] = None
+    managed_identity_client_id: Optional[str] = None
+
+
+class StorageFileUpdateBase(BaseModel):
+    """Base model for file update operations"""
+    storage_account_url: str
+    container_name: str
+    blob_name: str
+    file_content: bytes
+    content_type: Optional[str] = None
+    managed_identity_client_id: Optional[str] = None
+
+
+class StorageFileDeleteBase(BaseModel):
+    """Base model for file delete operations"""
+    storage_account_url: str
+    container_name: str
+    blob_name: str
+    managed_identity_client_id: Optional[str] = None
+
+
+class StorageFileDownloadBase(BaseModel):
+    """Base model for file download operations"""
+    storage_account_url: str
+    container_name: str
+    blob_name: str
+    managed_identity_client_id: Optional[str] = None
+
+
+class StorageFileUrlBase(BaseModel):
+    """Base model for getting presigned URL"""
+    storage_account_url: str
+    container_name: str
+    blob_name: str
+    expiry_hours: Optional[int] = 1
+    managed_identity_client_id: Optional[str] = None
+
+
+class StorageContainerCreateBase(BaseModel):
+    """Base model for creating a container"""
+    storage_account_url: str
+    container_name: str
+    public_access: Optional[str] = None
+    managed_identity_client_id: Optional[str] = None

trovesuite/storage/storage_controller.py

@@ -0,0 +1,198 @@
+from typing import List
+from fastapi import APIRouter, File, UploadFile, Form
+from fastapi.responses import StreamingResponse
+from io import BytesIO
+from .storage_write_dto import (
+    StorageContainerCreateControllerWriteDto,
+    StorageFileUploadControllerWriteDto,
+    StorageFileUpdateControllerWriteDto,
+    StorageFileDeleteControllerWriteDto,
+    StorageFileDownloadControllerWriteDto,
+    StorageFileUrlControllerWriteDto
+)
+from .storage_read_dto import (
+    StorageContainerCreateControllerReadDto,
+    StorageFileUploadControllerReadDto,
+    StorageFileUpdateControllerReadDto,
+    StorageFileDeleteControllerReadDto,
+    StorageFileDownloadControllerReadDto,
+    StorageFileUrlControllerReadDto
+)
+from .storage_service import StorageService
+from ..entities.sh_response import Respons
+
+storage_router = APIRouter(tags=["File Storage"])
+
+
+@storage_router.post("/create-container", response_model=Respons[StorageContainerCreateControllerReadDto])
+async def create_container(data: StorageContainerCreateControllerWriteDto):
+    """
+    Create a new Azure Storage container.
+
+    Example request body:
+    {
+        "storage_account_url": "https://myaccount.blob.core.windows.net",
+        "container_name": "my-container",
+        "public_access": null,
+        "managed_identity_client_id": "your-client-id"  // optional
+    }
+    """
+    return StorageService.create_container(data=data)
+
+
+@storage_router.post("/upload", response_model=Respons[StorageFileUploadControllerReadDto])
+async def upload_file(
+    storage_account_url: str = Form(...),
+    container_name: str = Form(...),
+    blob_name: str = Form(...),
+    file: UploadFile = File(...),
+    directory_path: str = Form(None),
+    managed_identity_client_id: str = Form(None)
+):
+    """
+    Upload a file to Azure Storage.
+
+    Use form-data with the following fields:
+    - storage_account_url: Your Azure storage URL
+    - container_name: Container name
+    - blob_name: Name for the blob
+    - file: The file to upload
+    - directory_path: Optional directory path (e.g., "uploads/2024")
+    - managed_identity_client_id: Optional client ID for user-assigned managed identity
+    """
+    content = await file.read()
+
+    upload_data = StorageFileUploadControllerWriteDto(
+        storage_account_url=storage_account_url,
+        container_name=container_name,
+        file_content=content,
+        blob_name=blob_name,
+        directory_path=directory_path,
+        content_type=file.content_type,
+        managed_identity_client_id=managed_identity_client_id
+    )
+
+    return StorageService.upload_file(data=upload_data)
+
+
+@storage_router.put("/update", response_model=Respons[StorageFileUpdateControllerReadDto])
+async def update_file(
+    storage_account_url: str = Form(...),
+    container_name: str = Form(...),
+    blob_name: str = Form(...),
+    file: UploadFile = File(...),
+    managed_identity_client_id: str = Form(None)
+):
+    """
+    Update an existing file in Azure Storage.
+
+    Use form-data with the following fields:
+    - storage_account_url: Your Azure storage URL
+    - container_name: Container name
+    - blob_name: Full blob name including path (e.g., "uploads/2024/file.pdf")
+    - file: The new file content
+    - managed_identity_client_id: Optional client ID for user-assigned managed identity
+    """
+    content = await file.read()
+
+    update_data = StorageFileUpdateControllerWriteDto(
+        storage_account_url=storage_account_url,
+        container_name=container_name,
+        blob_name=blob_name,
+        file_content=content,
+        content_type=file.content_type,
+        managed_identity_client_id=managed_identity_client_id
+    )
+
+    return StorageService.update_file(data=update_data)
+
+
+@storage_router.delete("/delete", response_model=Respons[StorageFileDeleteControllerReadDto])
+async def delete_file(data: StorageFileDeleteControllerWriteDto):
+    """
+    Delete a file from Azure Storage.
+
+    Example request body:
+    {
+        "storage_account_url": "https://myaccount.blob.core.windows.net",
+        "container_name": "my-container",
+        "blob_name": "uploads/2024/file.pdf",
+        "managed_identity_client_id": "your-client-id"  // optional
+    }
+    """
+    return StorageService.delete_file(data=data)
+
+
+@storage_router.delete("/delete-multiple", response_model=Respons[StorageFileDeleteControllerReadDto])
+async def delete_multiple_files(
+    storage_account_url: str,
+    container_name: str,
+    blob_names: List[str],
+    managed_identity_client_id: str = None
+):
+    """
+    Delete multiple files from Azure Storage.
+
+    Example request body:
+    {
+        "storage_account_url": "https://myaccount.blob.core.windows.net",
+        "container_name": "my-container",
+        "blob_names": ["file1.pdf", "file2.pdf", "folder/file3.jpg"],
+        "managed_identity_client_id": "your-client-id"  // optional
+    }
+    """
+    return StorageService.delete_multiple_files(
+        storage_account_url=storage_account_url,
+        container_name=container_name,
+        blob_names=blob_names,
+        managed_identity_client_id=managed_identity_client_id
+    )
+
+
+@storage_router.post("/download")
+async def download_file(data: StorageFileDownloadControllerWriteDto):
+    """
+    Download a file from Azure Storage.
+
+    Returns the file as a streaming response.
+
+    Example request body:
+    {
+        "storage_account_url": "https://myaccount.blob.core.windows.net",
+        "container_name": "my-container",
+        "blob_name": "uploads/2024/file.pdf",
+        "managed_identity_client_id": "your-client-id"  // optional
+    }
+    """
+    result = StorageService.download_file(data=data)
+
+    if not result.success:
+        return result
+
+    file_data = result.data[0]
+
+    # Return as streaming response
+    return StreamingResponse(
+        BytesIO(file_data.content),
+        media_type=file_data.content_type or "application/octet-stream",
+        headers={
+            "Content-Disposition": f"attachment; filename={data.blob_name.split('/')[-1]}"
+        }
+    )
+
+
+@storage_router.post("/get-url", response_model=Respons[StorageFileUrlControllerReadDto])
+async def get_file_url(data: StorageFileUrlControllerWriteDto):
+    """
+    Generate a presigned URL for a file.
+
+    Example request body:
+    {
+        "storage_account_url": "https://myaccount.blob.core.windows.net",
+        "container_name": "my-container",
+        "blob_name": "uploads/2024/file.pdf",
+        "expiry_hours": 2,
+        "managed_identity_client_id": "your-client-id"  // optional
+    }
+    """
+    return StorageService.get_file_url(data=data)

trovesuite/storage/storage_read_dto.py

@@ -0,0 +1,74 @@
+from pydantic import BaseModel
+from typing import Optional
+
+
+# Container Creation
+
+class StorageContainerCreateControllerReadDto(BaseModel):
+    container_name: str
+    container_url: Optional[str] = None
+
+
+class StorageContainerCreateServiceReadDto(StorageContainerCreateControllerReadDto):
+    pass
+
+
+# File Upload
+
+class StorageFileUploadControllerReadDto(BaseModel):
+    blob_name: str
+    blob_url: str
+    content_type: Optional[str] = None
+    size: Optional[int] = None
+
+
+class StorageFileUploadServiceReadDto(StorageFileUploadControllerReadDto):
+    pass
+
+
+# File Update
+
+class StorageFileUpdateControllerReadDto(BaseModel):
+    blob_name: str
+    blob_url: str
+    updated_at: Optional[str] = None
+
+
+class StorageFileUpdateServiceReadDto(StorageFileUpdateControllerReadDto):
+    pass
+
+
+# File Delete
+
+class StorageFileDeleteControllerReadDto(BaseModel):
+    blob_name: str
+    deleted: bool
+
+
+class StorageFileDeleteServiceReadDto(StorageFileDeleteControllerReadDto):
+    pass
+
+
+# File Download
+
+class StorageFileDownloadControllerReadDto(BaseModel):
+    blob_name: str
+    content: bytes
+    content_type: Optional[str] = None
+    size: Optional[int] = None
+
+
+class StorageFileDownloadServiceReadDto(StorageFileDownloadControllerReadDto):
+    pass
+
+
+# File URL
+
+class StorageFileUrlControllerReadDto(BaseModel):
+    blob_name: str
+    presigned_url: str
+    expires_in_hours: int
+
+
+class StorageFileUrlServiceReadDto(StorageFileUrlControllerReadDto):
+    pass