saviialib 0.9.1__py3-none-any.whl → 1.6.0__py3-none-any.whl

saviialib/libs/sharepoint_client/clients/sharepoint_rest_api.py

@@ -1,7 +1,9 @@
 from typing import Any
-
-from aiohttp import ClientError, ClientSession
+import json
+from aiohttp import ClientError, ClientSession, TCPConnector
 from dotenv import load_dotenv
+import ssl
+import certifi
 
 from saviialib.libs.sharepoint_client.sharepoint_client_contract import (
     SharepointClientContract,
@@ -10,10 +12,12 @@ from saviialib.libs.sharepoint_client.types.sharepoint_client_types import (
     SpListFilesArgs,
     SpListFoldersArgs,
     SpUploadFileArgs,
+    SpCreateFolderArgs,
     SharepointClientInitArgs,
 )
 
 load_dotenv()
+ssl_context = ssl.create_default_context(cafile=certifi.where())
 
 
 class SharepointRestAPI(SharepointClientContract):
@@ -50,7 +54,7 @@ class SharepointRestAPI(SharepointClientContract):
             "Content-Type": "application/x-www-form-urlencoded",
         }
 
-        async with ClientSession() as session:
+        async with ClientSession(connector=TCPConnector(ssl=ssl_context)) as session:
             # Load access token
             response = await session.post(url, data=payload, headers=headers)
             if response.status != 200:
@@ -74,8 +78,9 @@ class SharepointRestAPI(SharepointClientContract):
                 "Content-Type": "application/json",
             }
             self.base_url = f"{site_url}/sites/{self.site_name}/_api/"
+            connector = TCPConnector(ssl=ssl_context)
             self.session = ClientSession(
-                headers=self.base_headers, base_url=self.base_url
+                headers=self.base_headers, base_url=self.base_url, connector=connector
             )
             return self
         except ClientError as error:
@@ -104,10 +109,11 @@ class SharepointRestAPI(SharepointClientContract):
             folder_relative_url = (
                 f"GetFolderByServerRelativeUrl('{args.folder_relative_url}')"
             )
-            endpoint = f"web/{folder_relative_url}/Folder"
+            endpoint = f"web/{folder_relative_url}/Folders"
             response = await self.session.get(endpoint.lstrip("/"))
             response.raise_for_status()
-            return await response.json()
+            response_json = await response.json()
+            return response_json
         except ClientError as error:
             raise ConnectionError(error) from error
 
@@ -133,3 +139,22 @@ class SharepointRestAPI(SharepointClientContract):
             return await response.json()
         except ClientError as error:
             raise ConnectionError(error) from error
+
+    async def create_folder(self, args: SpCreateFolderArgs):
+        try:
+            # Load form digest value
+            form_digest_value = await self._load_form_digest_value()
+            headers = {
+                **self.base_headers,
+                "X-RequestDigest": form_digest_value,
+            }
+            body = {"ServerRelativeUrl": f"{args.folder_relative_url}"}
+            endpoint = "web/folders"
+            response = await self.session.post(
+                endpoint.lstrip("/"), data=json.dumps(body), headers=headers
+            )
+            response.raise_for_status()
+            response_json = await response.json()
+            return response_json
+        except ClientError as error:
+            raise ConnectionError(error) from error

saviialib/libs/sharepoint_client/sharepoint_client.py

@@ -5,6 +5,7 @@ from .types.sharepoint_client_types import (
     SpListFilesArgs,
     SpListFoldersArgs,
     SpUploadFileArgs,
+    SpCreateFolderArgs,
 )
 
 
@@ -18,6 +19,26 @@ class SharepointClient(SharepointClientContract):
         elif args.client_name == "sharepoint_rest_api":
             self.client_obj = SharepointRestAPI(args)
 
+    @property
+    def tenant_id(self):
+        return self.client_obj.tenant_id
+
+    @property
+    def tenant_name(self):
+        return self.client_obj.tenant_name
+
+    @property
+    def site_name(self):
+        return self.client_obj.site_name
+
+    @property
+    def client_id(self):
+        return self.client_obj.client_id
+
+    @property
+    def client_secret(self):
+        return self.client_obj.client_secret
+
     async def __aenter__(self):
         return await self.client_obj.__aenter__()
 
@@ -28,7 +49,10 @@ class SharepointClient(SharepointClientContract):
         return await self.client_obj.list_files(args)
 
     async def list_folders(self, args: SpListFoldersArgs) -> list:
-        return await self.client_obj.list_files(args)
+        return await self.client_obj.list_folders(args)
 
     async def upload_file(self, args: SpUploadFileArgs) -> dict:
         return await self.client_obj.upload_file(args)
+
+    async def create_folder(self, args: SpCreateFolderArgs) -> dict:
+        return await self.client_obj.create_folder(args)

saviialib/libs/sharepoint_client/sharepoint_client_contract.py

@@ -4,6 +4,7 @@ from .types.sharepoint_client_types import (
     SpListFilesArgs,
     SpListFoldersArgs,
     SpUploadFileArgs,
+    SpCreateFolderArgs,
 )
 
 
@@ -19,3 +20,7 @@ class SharepointClientContract(ABC):
     @abstractmethod
     async def upload_file(self, args: SpUploadFileArgs) -> dict:
         pass
+
+    @abstractmethod
+    async def create_folder(self, args: SpCreateFolderArgs) -> dict:
+        pass

saviialib/libs/sharepoint_client/types/sharepoint_client_types.py

@@ -23,3 +23,8 @@ class SpUploadFileArgs:
     folder_relative_url: str
     file_name: str
     file_content: bytes = bytes()
+
+
+@dataclass
+class SpCreateFolderArgs:
+    folder_relative_url: str

saviialib/libs/zero_dependency/utils/booleans_utils.py

@@ -0,0 +1,2 @@
+def boolean_to_emoji(boolean: bool) -> str:
+    return "✅" if boolean else "❌"

saviialib/libs/zero_dependency/utils/datetime_utils.py

@@ -2,7 +2,7 @@ from datetime import datetime
 from zoneinfo import ZoneInfo
 
 
-def today(timezone: str = "America/Santiago") -> str:
+def today(timezone: str = "America/Santiago") -> datetime:
     """
     Return the current date.
 

saviialib/libs/zero_dependency/utils/strings_utils.py

@@ -0,0 +1,5 @@
+def are_equal(str1: str, str2: str) -> bool:
+    if str1.lower() == str2.lower():
+        return True
+    else:
+        return False

saviialib/services/backup/api.py

@@ -0,0 +1,36 @@
+from typing import Any, Dict
+
+from .controllers.upload_backup_to_sharepoint import (
+    UploadBackupToSharepointControllerInput,
+)
+from .controllers.upload_backup_to_sharepoint import UploadBackupToSharepointController
+from saviialib.general_types.api.saviia_backup_api_types import SaviiaBackupConfig
+
+
+class SaviiaBackupAPI:
+    """
+    EpiiAPI is a service class that provides methods to interact with Patagonia Center system.
+    """
+
+    def __init__(self, config: SaviiaBackupConfig):
+        self.config = config
+
+    async def upload_backup_to_sharepoint(
+        self, local_backup_source_path: str, sharepoint_destination_path: str
+    ) -> Dict[str, Any]:
+        """Migrate a backup folder from Home assistant to Sharepoint directory.
+        Args:
+            local_backup_source_path (str): Local path to backup.
+        Returns:
+            response (dict): A dictionary containing the response from the upload operation.
+                This dictionary will typically include information about the success or
+                failure of the upload, as well as any relevant metadata.
+        """
+
+        controller = UploadBackupToSharepointController(
+            UploadBackupToSharepointControllerInput(
+                self.config, local_backup_source_path, sharepoint_destination_path
+            )
+        )
+        response = await controller.execute()
+        return response.__dict__

saviialib/services/{epii → backup}/controllers/types/__init__.py

@@ -1,4 +1,4 @@
-from .update_thies_data_types import (
+from ....thies.controllers.types.update_thies_data_types import (
     UpdateThiesDataControllerInput,
     UpdateThiesDataControllerOutput,
 )

saviialib/services/{epii → backup}/controllers/types/upload_backup_to_sharepoint_types.py

@@ -1,12 +1,14 @@
 from dataclasses import dataclass, field
 from typing import Dict
 
-from saviialib.general_types.api.epii_api_types import EpiiSharepointBackupConfig
+from saviialib.general_types.api.saviia_backup_api_types import SaviiaBackupConfig
 
 
 @dataclass
 class UploadBackupToSharepointControllerInput:
-    config: EpiiSharepointBackupConfig
+    config: SaviiaBackupConfig
+    local_backup_source_path: str
+    sharepoint_destination_path: str
 
 
 @dataclass

saviialib/services/{epii → backup}/controllers/upload_backup_to_sharepoint.py

@@ -1,7 +1,7 @@
 from http import HTTPStatus
-from saviialib.general_types.api.epii_api_types import SharepointConfig
+from saviialib.general_types.api.saviia_api_types import SharepointConfig
 
-from saviialib.general_types.error_types.api.epii_api_error_types import (
+from saviialib.general_types.error_types.api.saviia_api_error_types import (
     BackupUploadError,
     BackupSourcePathError,
     BackupEmptyError,
@@ -10,14 +10,14 @@ from saviialib.general_types.error_types.common.common_types import (
     EmptyDataError,
     SharepointClientError,
 )
-from saviialib.services.epii.controllers.types.upload_backup_to_sharepoint_types import (
+from saviialib.services.backup.controllers.types.upload_backup_to_sharepoint_types import (
     UploadBackupToSharepointControllerInput,
     UploadBackupToSharepointControllerOutput,
 )
-from saviialib.services.epii.use_cases.types.upload_backup_to_sharepoint_types import (
+from saviialib.services.backup.use_cases.types.upload_backup_to_sharepoint_types import (
     UploadBackupToSharepointUseCaseInput,
 )
-from saviialib.services.epii.use_cases.upload_backup_to_sharepoint import (
+from saviialib.services.backup.use_cases.upload_backup_to_sharepoint import (
     UploadBackupToSharepointUsecase,
 )
 
@@ -33,8 +33,9 @@ class UploadBackupToSharepointController:
                     sharepoint_tenant_name=input.config.sharepoint_tenant_name,
                     sharepoint_tenant_id=input.config.sharepoint_tenant_id,
                 ),
-                local_backup_source_path=input.config.local_backup_source_path,
-                destination_folders=input.config.destination_folders
+                local_backup_source_path=input.local_backup_source_path,
+                sharepoint_destination_path=input.sharepoint_destination_path,
+                logger=input.config.logger,
             )
         )
 
@@ -44,7 +45,7 @@ class UploadBackupToSharepointController:
             return UploadBackupToSharepointControllerOutput(
                 message="Local backup was migrated successfully",
                 status=HTTPStatus.OK.value,
-                metadata={"data": data},
+                metadata={"data": data},  # type: ignore
             )
         except EmptyDataError:
             return UploadBackupToSharepointControllerOutput(

saviialib/services/backup/use_cases/constants/upload_backup_to_sharepoint_constants.py

@@ -0,0 +1,5 @@
+import logging
+
+
+logging.basicConfig(level=logging.DEBUG)
+LOGGER = logging.getLogger(__package__)

saviialib/services/{epii → backup}/use_cases/types/__init__.py

@@ -1,4 +1,4 @@
-from .update_thies_data_types import (
+from ....thies.use_cases.types.update_thies_data_types import (
     FtpClientConfig,
     SharepointConfig,
     UpdateThiesDataUseCaseInput,

saviialib/services/{epii → backup}/use_cases/types/upload_backup_to_sharepoint_types.py

@@ -1,9 +1,11 @@
 from dataclasses import dataclass
-from saviialib.general_types.api.epii_api_types import SharepointConfig
+from saviialib.general_types.api.saviia_api_types import SharepointConfig
+from logging import Logger
 
 
 @dataclass
 class UploadBackupToSharepointUseCaseInput:
     sharepoint_config: SharepointConfig
     local_backup_source_path: str
-    destination_folders: dict
+    sharepoint_destination_path: str
+    logger: Logger