aiverify-moonshot 0.4.0__py3-none-any.whl

moonshot/integrations/web_api/routes/context_strategy.py

@@ -0,0 +1,129 @@
+from dependency_injector.wiring import Provide, inject
+from fastapi import APIRouter, Depends, HTTPException
+
+from ..container import Container
+from ..services.context_strategy_service import ContextStrategyService
+from ..services.utils.exceptions_handler import ServiceException
+
+router = APIRouter(tags=["Context Strategy"])
+
+@router.get("/api/v1/context-strategies")
+@inject
+def get_all_context_strategies(
+    context_strategy_service: ContextStrategyService = Depends(
+        Provide[Container.context_strategy_service]
+    ),
+) -> list[dict]:
+    """
+    Retrieve all context strategies from the database.
+
+    Args:
+        context_strategy_service (ContextStrategyService): The service responsible for retrieving context strategies.
+
+    Returns:
+        list[dict]: A list of context strategies with details.
+
+    Raises:
+        HTTPException: An error with status code 404 if no context strategies are found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    try:
+        return context_strategy_service.get_ctx_strategy()
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404,
+                detail=f"Failed to retrieve context strategies: {e.msg}",
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400,
+                detail=f"Failed to retrieve context strategies: {e.msg}",
+            )
+        else:
+            raise HTTPException(
+                status_code=500,
+                detail=f"Failed to retrieve context strategies: {e.msg}",
+            )
+        
+@router.get("/api/v1/context-strategies/name")
+@inject
+def get_all_context_strategies_name(
+    context_strategy_service: ContextStrategyService = Depends(
+        Provide[Container.context_strategy_service]
+    ),
+) -> list[str]:
+    """
+    Retrieve all context strategies from the database.
+
+    Args:
+        context_strategy_service (ContextStrategyService): The service responsible for retrieving context strategies.
+
+    Returns:
+        list[str]: A list of context strategies.
+
+    Raises:
+        HTTPException: An error with status code 404 if no context strategies are found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    try:
+        return context_strategy_service.get_ctx_strategy_name()
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404,
+                detail=f"Failed to retrieve context strategies: {e.msg}",
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400,
+                detail=f"Failed to retrieve context strategies: {e.msg}",
+            )
+        else:
+            raise HTTPException(
+                status_code=500,
+                detail=f"Failed to retrieve context strategies: {e.msg}",
+            )
+
+
+@router.delete("/api/v1/context-strategies/{ctx_strategy_name}")
+@inject
+def delete_context_strategy(
+    ctx_strategy_name: str,
+    context_strategy_service: ContextStrategyService = Depends(
+        Provide[Container.context_strategy_service]
+    ),
+) -> dict[str, bool]:
+    """
+    Delete a context strategy from the database by its name.
+
+    Args:
+        ctx_strategy_name (str): The name of the context strategy to delete.
+        context_strategy_service (ContextStrategyService): The service responsible for deleting the context strategy.
+
+    Returns:
+        dict[str, bool]: A dictionary with a key 'success' indicating the result of the deletion operation.
+
+    Raises:
+        HTTPException: An error with status code 404 if the context strategy is not found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    try:
+        context_strategy_service.delete_ctx_strategy(ctx_strategy_name)
+        return {"success": True}
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to delete context strategy: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to delete context strategy: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to delete context strategy: {e.msg}"
+            )

moonshot/integrations/web_api/routes/cookbook.py

@@ -0,0 +1,225 @@
+from typing import Optional
+
+from dependency_injector.wiring import Provide, inject
+from fastapi import APIRouter, Depends, HTTPException, Query
+
+from ..container import Container
+from ..schemas.cookbook_create_dto import CookbookCreateDTO, CookbookUpdateDTO
+from ..schemas.cookbook_response_model import CookbookResponseModel
+from ..services.cookbook_service import CookbookService
+from ..services.utils.exceptions_handler import ServiceException
+
+router = APIRouter(tags=["Cookbook"])
+
+
+@router.post("/api/v1/cookbooks")
+@inject
+def create_cookbook(
+    cookbook_data: CookbookCreateDTO,
+    cookbook_service: CookbookService = Depends(Provide[Container.cookbook_service]),
+) -> dict[str, str]:
+    """
+    Endpoint to create a new cookbook entry in the database.
+
+    Parameters:
+        cookbook_data (CookbookCreateDTO): The DTO containing the details for the new cookbook.
+        cookbook_service (CookbookService): The service layer responsible for the creation logic.
+
+    Returns:
+        dict[str, str]: A dictionary with a message key indicating successful creation.
+
+    Raises:
+        HTTPException: 404 if the cookbook cannot be found.
+                       400 if there is a validation error with the provided data.
+                       500 for any other internal server error.
+    """
+    try:
+        cookbook_service.create_cookbook(cookbook_data)
+        return {"message": "Cookbook created successfully"}
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to create cookbook: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to create cookbook: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to create cookbook: {e.msg}"
+            )
+
+
+@router.get("/api/v1/cookbooks")
+@inject
+def get_all_cookbooks(
+    ids: Optional[str] = Query(None, description="Get recipes to query"),
+    tags: Optional[str] = Query(None, description="Filter cookbooks by tags"),
+    categories: Optional[str] = Query(
+        None, description="Filter cookbooks by categories"
+    ),
+    categories_excluded: Optional[str] = Query(
+        None, description="Filter out (exlude) cookbooks by categories"
+    ),
+    count: bool = Query(False, description="Whether to include the count of recipes"),
+    cookbook_service: CookbookService = Depends(Provide[Container.cookbook_service]),
+) -> list[CookbookResponseModel]:
+    """
+    Endpoint to retrieve a list of all cookbooks, with optional filtering.
+
+    Parameters:
+        ids (Optional[str]): A string to filter cookbooks by ids.
+        tags (Optional[str]): A string to filter cookbooks by tags.
+        categories (Optional[str]): A string to filter cookbooks by categories.
+        categories_excluded (Optional[str]): A string to filter out (exclude) cookbooks by categories.
+        count (bool): A flag to decide if the count of recipes should be included.
+        cookbook_service (CookbookService): The service layer responsible for retrieval logic.
+
+    Returns:
+        list[CookbookResponseModel]: A list of CookbookResponseModel instances that match the filters.
+
+    Raises:
+        HTTPException: 404 if no cookbooks could be found.
+                       400 if there is a validation error with the filters.
+                       500 for any other internal server error.
+    """
+    try:
+        cookbooks = cookbook_service.get_all_cookbooks(
+            tags=tags,
+            categories=categories,
+            count=count,
+            ids=ids,
+            categories_excluded=categories_excluded,
+        )
+        return cookbooks
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to retrieve cookbooks: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to retrieve cookbooks: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to retrieve cookbooks: {e.msg}"
+            )
+
+
+@router.get("/api/v1/cookbooks/name")
+@inject
+def get_all_cookbooks_name(
+    cookbook_service: CookbookService = Depends(Provide[Container.cookbook_service]),
+) -> list[str]:
+    """
+    Endpoint to retrieve the names of all cookbooks in the database.
+
+    Parameters:
+        cookbook_service (CookbookService): The service layer responsible for retrieving the names.
+
+    Returns:
+        list[str]: A list of strings representing the names of all cookbooks.
+
+    Raises:
+        HTTPException: 404 if no cookbook names could be found.
+                       400 if there is a validation error.
+                       500 for any other internal server error.
+    """
+    try:
+        cookbooks = cookbook_service.get_all_cookbooks_names()
+        return cookbooks
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to retrieve cookbooks: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to retrieve cookbooks: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to retrieve cookbooks: {e.msg}"
+            )
+
+
+@router.put("/api/v1/cookbooks/{cookbook_id}")
+@inject
+def update_cookbook(
+    cookbook_id: str,
+    cookbook_data: CookbookUpdateDTO,
+    cookbook_service: CookbookService = Depends(Provide[Container.cookbook_service]),
+) -> dict[str, str]:
+    """
+    Endpoint to update the details of an existing cookbook in the database.
+
+    Parameters:
+        cookbook_id (str): The unique identifier of the cookbook to be updated.
+        cookbook_data (CookbookCreateDTO): The DTO containing the updated details for the cookbook.
+        cookbook_service (CookbookService): The service layer responsible for the update logic.
+
+    Returns:
+        dict[str, str]: A dictionary with a message key indicating successful update.
+
+    Raises:
+        HTTPException: 404 if the cookbook with the given ID cannot be found.
+                       400 if there is a validation error with the provided data.
+                       500 for any other internal server error.
+    """
+    try:
+        cookbook_service.update_cookbook(cookbook_data, cookbook_id)
+        return {"message": "Cookbook updated successfully"}
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to update cookbook: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to update cookbook: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to update cookbook: {e.msg}"
+            )
+
+
+@router.delete("/api/v1/cookbooks/{cb_id}")
+@inject
+def delete_cookbook(
+    cb_id: str,
+    cookbook_service: CookbookService = Depends(Provide[Container.cookbook_service]),
+) -> dict[str, str]:
+    """
+    Endpoint to delete a cookbook entry from the database using its ID.
+
+    Parameters:
+        cb_id (str): The unique identifier of the cookbook to be deleted.
+        cookbook_service (CookbookService): The service layer responsible for the deletion logic.
+
+    Returns:
+        dict[str, str]: A dictionary with a message key indicating successful deletion.
+
+    Raises:
+        HTTPException: 404 if the cookbook with the given ID cannot be found.
+                       400 if there is a validation error.
+                       500 for any other internal server error.
+    """
+    try:
+        cookbook_service.delete_cookbook(cb_id)
+        return {"message": "Cookbook deleted successfully"}
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to delete cookbook: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to delete cookbook: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to delete cookbook: {e.msg}"
+            )

moonshot/integrations/web_api/routes/dataset.py

@@ -0,0 +1,120 @@
+from dependency_injector.wiring import Provide, inject
+from fastapi import APIRouter, Depends, HTTPException
+
+from ..container import Container
+from ..schemas.dataset_response_dto import DatasetResponseDTO
+from ..services.dataset_service import DatasetService
+from ..services.utils.exceptions_handler import ServiceException
+
+router = APIRouter(tags=["Datasets"])
+
+
+@router.get("/api/v1/datasets")
+@inject
+def get_all_datasets(
+    dataset_service: DatasetService = Depends(Provide[Container.dataset_service]),
+) -> list[DatasetResponseDTO]:
+    """
+    Retrieve all datasets from the database.
+
+    Args:
+        dataset_service (DatasetService): The service responsible for retrieving datasets.
+
+    Returns:
+        list: A list of all datasets.
+
+    Raises:
+        HTTPException: An error with status code 404 if no datasets are found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    try:
+        return dataset_service.get_all_datasets()
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to retrieve datasets: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to retrieve datasets: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to retrieve datasets: {e.msg}"
+            )
+
+
+@router.get("/api/v1/datasets/name")
+@inject
+def get_all_datasets_name(
+    dataset_service: DatasetService = Depends(Provide[Container.dataset_service]),
+) -> list[str]:
+    """
+    Retrieve the names of all datasets from the database.
+
+    Args:
+        dataset_service (DatasetService): The service responsible for retrieving dataset names.
+
+    Returns:
+        list[str]: A list of dataset names.
+
+    Raises:
+        HTTPException: An error with status code 404 if no dataset names are found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    try:
+        return dataset_service.get_all_datasets_name()
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to retrieve dataset names: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to retrieve dataset names: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to retrieve dataset names: {e.msg}"
+            )
+
+
+@router.delete("/api/v1/datasets/{dataset_id}")
+@inject
+def delete_dataset(
+    dataset_id: str,
+    dataset_service: DatasetService = Depends(Provide[Container.dataset_service]),
+) -> dict[str, str]:
+    """
+    Delete a dataset from the database by its ID.
+
+    Args:
+        dataset_id (str): The unique identifier of the dataset to delete.
+        dataset_service (DatasetService): The service responsible for deleting the dataset.
+
+    Returns:
+        dict[str, str]: A message indicating the successful deletion of the dataset.
+
+    Raises:
+        HTTPException: An error with status code 404 if the dataset is not found.
+                       An error with status code 400 if there is a validation error.
+                       An error with status code 500 for any other server-side error.
+    """
+    try:
+        dataset_service.delete_dataset(dataset_id)
+        return {"message": "Dataset deleted successfully"}
+    except ServiceException as e:
+        if e.error_code == "FileNotFound":
+            raise HTTPException(
+                status_code=404, detail=f"Failed to delete dataset: {e.msg}"
+            )
+        elif e.error_code == "ValidationError":
+            raise HTTPException(
+                status_code=400, detail=f"Failed to delete dataset: {e.msg}"
+            )
+        else:
+            raise HTTPException(
+                status_code=500, detail=f"Failed to delete dataset: {e.msg}"
+            )