saviialib 0.11.2__py3-none-any.whl → 1.0.1__py3-none-any.whl

saviialib/libs/ftp_client/clients/ftplib_client.py

@@ -0,0 +1,58 @@
+import ftplib
+import asyncio
+from io import BytesIO
+from saviialib.libs.ftp_client.ftp_client_contract import (
+    FTPClientContract,
+)
+from saviialib.libs.ftp_client.types.ftp_client_types import (
+    FtpClientInitArgs,
+    FtpListFilesArgs,
+    FtpReadFileArgs,
+)
+
+
+class FtplibClient(FTPClientContract):
+    def __init__(self, args: FtpClientInitArgs) -> None:
+        self.host = args.config.ftp_host
+        self.port = args.config.ftp_port
+        self.password = args.config.ftp_password
+        self.user = args.config.ftp_user
+        self.client = ftplib.FTP(host=self.host, user=self.user, passwd=self.password)
+
+    async def _async_start(self) -> None:
+        try:
+            await asyncio.to_thread(self.client.login, self.user, self.password)
+        except OSError:
+            raise ConnectionRefusedError(
+                f"{self.host}:{self.port} isn't active. "
+                "Please ensure the server is running and accessible."
+            )
+        except Exception as error:
+            raise ConnectionError(
+                f"General connection for {self.host}:{self.port}.", error.__str__()
+            )
+
+    async def list_files(self, args: FtpListFilesArgs) -> list[str]:
+        try:
+            EXCLUDED_NAMES = [".", ".."]
+            await self._async_start()
+            await asyncio.to_thread(self.client.cwd, args.path)
+            filenames = await asyncio.to_thread(self.client.nlst, args.path)
+            return [
+                filename for filename in filenames if filename not in EXCLUDED_NAMES
+            ]
+        except Exception as error:
+            raise ConnectionAbortedError(error)
+
+    async def read_file(self, args: FtpReadFileArgs) -> bytes:
+        await self._async_start()
+        try:
+            file_content = BytesIO()
+            await asyncio.to_thread(
+                self.client.retrbinary, "RETR " + args.file_path, file_content.write
+            )
+            await asyncio.to_thread(file_content.seek, 0)
+            file_bytes = await asyncio.to_thread(file_content.read)
+            return file_bytes
+        except Exception as error:
+            raise FileNotFoundError(f"File not found: {args.file_path}") from error

saviialib/libs/ftp_client/ftp_client.py

@@ -1,10 +1,11 @@
 from .clients.aioftp_client import AioFTPClient
+from .clients.ftplib_client import FtplibClient
 from .ftp_client_contract import FTPClientContract
 from .types.ftp_client_types import FtpClientInitArgs, FtpListFilesArgs, FtpReadFileArgs
 
 
 class FTPClient(FTPClientContract):
-    CLIENTS = {"aioftp_client"}
+    CLIENTS = {"aioftp_client", "ftplib_client"}
 
     def __init__(self, args: FtpClientInitArgs) -> None:
         if args.client_name not in FTPClient.CLIENTS:
@@ -13,6 +14,8 @@ class FTPClient(FTPClientContract):
 
         if args.client_name == "aioftp_client":
             self.client_obj = AioFTPClient(args)
+        elif args.client_name == "ftplib_client":
+            self.client_obj = FtplibClient(args)
         self.client_name = args.client_name
 
     async def list_files(self, args: FtpListFilesArgs) -> list[str]:

saviialib/libs/sharepoint_client/__init__.py

@@ -4,6 +4,7 @@ from .types.sharepoint_client_types import (
     SpListFilesArgs,
     SpListFoldersArgs,
     SpUploadFileArgs,
+    SpCreateFolderArgs,
 )
 
 __all__ = [
@@ -12,4 +13,5 @@ __all__ = [
     "SpListFilesArgs",
     "SpListFoldersArgs",
     "SpUploadFileArgs",
+    "SpCreateFolderArgs",
 ]

saviialib/libs/sharepoint_client/clients/sharepoint_rest_api.py

@@ -1,5 +1,5 @@
 from typing import Any
-
+import json
 from aiohttp import ClientError, ClientSession
 from dotenv import load_dotenv
 
@@ -10,6 +10,7 @@ from saviialib.libs.sharepoint_client.types.sharepoint_client_types import (
     SpListFilesArgs,
     SpListFoldersArgs,
     SpUploadFileArgs,
+    SpCreateFolderArgs,
     SharepointClientInitArgs,
 )
 
@@ -134,3 +135,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,
 )
 
 
@@ -52,3 +53,6 @@ class SharepointClient(SharepointClientContract):
 
     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/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/services/epii/use_cases/update_thies_data.py

@@ -59,7 +59,7 @@ class UpdateThiesDataUseCase:
     def _initialize_thies_ftp_client(self, config: FtpClientConfig) -> FTPClient:
         """Initialize the FTP client."""
         try:
-            return FTPClient(FtpClientInitArgs(config, client_name="aioftp_client"))
+            return FTPClient(FtpClientInitArgs(config, client_name="ftplib_client"))
         except RuntimeError as error:
             raise FtpClientError(error)
 

saviialib/services/epii/use_cases/upload_backup_to_sharepoint.py

@@ -1,5 +1,6 @@
 import asyncio
 from time import time
+from saviialib.libs.zero_dependency.utils.datetime_utils import today, datetime_to_str
 from logging import Logger
 from saviialib.general_types.error_types.api.epii_api_error_types import (
     BackupEmptyError,
@@ -20,7 +21,7 @@ from saviialib.libs.sharepoint_client import (
     SharepointClient,
     SharepointClientInitArgs,
     SpUploadFileArgs,
-    SpListFoldersArgs,
+    SpCreateFolderArgs,
 )
 from saviialib.services.epii.utils.upload_backup_to_sharepoint_utils import (
     calculate_percentage_uploaded,
@@ -40,6 +41,7 @@ class UploadBackupToSharepointUsecase:
         self.sharepoint_config = input.sharepoint_config
         self.local_backup_source_path = input.local_backup_source_path
         self.sharepoint_destination_path = input.sharepoint_destination_path
+
         self.files_client = self._initialize_files_client()
         self.dir_client = self._initialize_directory_client()
         self.log_history = []
@@ -61,7 +63,28 @@ class UploadBackupToSharepointUsecase:
     def _initialize_files_client(self):
         return FilesClient(FilesClientInitArgs(client_name="aiofiles_client"))
 
+    async def _initialize_backup_base_folder(self):
+        local_backup_name = (
+            f"/local-backup-{datetime_to_str(today(), date_format='%m-%d-%Y')}"
+        )
+        local_backup_destination_path = (
+            self.sharepoint_destination_path + local_backup_name
+        )
+        async with self.sharepoint_client:
+            await self.sharepoint_client.create_folder(
+                SpCreateFolderArgs(folder_relative_url=local_backup_destination_path)
+            )
+        self.sharepoint_destination_path = local_backup_destination_path
+        base_folder_message = (
+            "[local_backup_lib] Creating base folder" + local_backup_name
+        )
+        self.logger.info(base_folder_message)
+        self.log_history.append(base_folder_message)
+
     async def _validate_backup_structure(self):
+        # Initialize the backup folder
+        await self._initialize_backup_base_folder()
+
         # Check if the local path exists in the main directory
         if not await self.dir_client.path_exists(self.local_backup_source_path):
             raise BackupSourcePathError(
@@ -74,20 +97,21 @@ class UploadBackupToSharepointUsecase:
             else []
         )
         async with self.sharepoint_client:  # type: ignore
-            response = await self.sharepoint_client.list_folders(
-                SpListFoldersArgs(folder_relative_url=self.sharepoint_destination_path)
-            )
-            sharepoint_directories = [x["Name"] for x in response["value"]]  # type: ignore
+            for local_dir in local_directories:
+                create_message = (
+                    f"[local_backup_lib] Creating a new directory '{local_dir}'."
+                )
 
-        # Verify that all local directories exist in SharePoint
-        for local_dir in local_directories:
-            if local_dir not in sharepoint_directories:
-                error_message = f"Folder '{local_dir}' does not exist in SharePoint destination path '{self.sharepoint_destination_path}'."
-                self.log_history.append(error_message)
-                self.logger.error("[local_backup_lib] %s", error_message)
-                raise BackupSourcePathError(
-                    reason=f"Folder '{local_dir}' does not exist in SharePoint destination path '{self.sharepoint_destination_path}'."
+                self.log_history.append(create_message)
+                await self.sharepoint_client.create_folder(
+                    SpCreateFolderArgs(
+                        folder_relative_url=self.sharepoint_destination_path
+                        + "/"
+                        + local_dir
+                    )
                 )
+                self.logger.info("[local_backup_lib] %s", create_message)
+                self.log_history.append(f"[local_backup_lib] {create_message}")
 
         # Check if the current folder only have files and each folder exist in Microsoft Sharepoint.
         if self.total_files == 0:

{saviialib-0.11.2.dist-info → saviialib-1.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: saviialib
-Version: 0.11.2
+Version: 1.0.1
 Summary: A client library for IoT projects in the RCER initiative
 License: MIT
 Author: pedropablozavalat
@@ -11,12 +11,12 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: aiofiles (==24.1.0)
-Requires-Dist: aioftp (==0.25.1)
-Requires-Dist: aiohttp (==3.11.16)
+Requires-Dist: aiofiles
+Requires-Dist: aioftp
+Requires-Dist: aiohttp
 Requires-Dist: build
-Requires-Dist: dotenv (==0.9.9)
-Requires-Dist: pytest-cov (==6.1.1)
+Requires-Dist: dotenv (>=0.9.9,<0.10.0)
+Requires-Dist: pytest-cov (>=6.1.1,<7.0.0)
 Description-Content-Type: text/markdown
 
 # SAVIIA Library 

{saviialib-0.11.2.dist-info → saviialib-1.0.1.dist-info}/RECORD

@@ -20,16 +20,17 @@ saviialib/libs/files_client/types/files_client_types.py,sha256=8OHm4nvZTW9K3ryeI
 saviialib/libs/ftp_client/__init__.py,sha256=dW2Yutgc7mJJJzgKLhWKXMgQ6KIWJYfFa1sGpjHH5xU,191
 saviialib/libs/ftp_client/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 saviialib/libs/ftp_client/clients/aioftp_client.py,sha256=5sgr3PMETgaBRlKeu_avxHIh6tr1gb7t2mkxmpRas_k,1925
-saviialib/libs/ftp_client/ftp_client.py,sha256=UWpuIrfO27A1M40YaBvAUdkdwPYJ4Ty4q55x7H4y1F8,844
+saviialib/libs/ftp_client/clients/ftplib_client.py,sha256=VcpAn23r03iyCUAj35xHU_AVNrll_45uHFN8Z9OnBkE,2203
+saviialib/libs/ftp_client/ftp_client.py,sha256=Naj9p0yYWlXt9um0zKfvuHSoycM8JZg2SPBm8110pYk,1008
 saviialib/libs/ftp_client/ftp_client_contract.py,sha256=tymkugDzsJ5PzUXIaSkwX1h7T0naR15qAkjrqswqDyM,338
 saviialib/libs/ftp_client/types/__init__.py,sha256=syfwf9feP4QK7fkCTfl4j8l11ic-jHtfi1DE2chaWbs,155
 saviialib/libs/ftp_client/types/ftp_client_types.py,sha256=e4SmYkewldulaD8ms2q75zVgLFXyBxBqoa_L-IQOmso,256
-saviialib/libs/sharepoint_client/__init__.py,sha256=v7h-cNsK-BaPp-hTU3NWRNXYRD9ztU-hsCk0eNRPIKA,334
-saviialib/libs/sharepoint_client/clients/sharepoint_rest_api.py,sha256=oGqNQcPL0HX-WYqO26JuIi9PDYPIZAoyDYMl965L_0c,5295
-saviialib/libs/sharepoint_client/sharepoint_client.py,sha256=LMQFti0Ulc4FCPQy17mrTLNJkO2Qp3i-KR0VbjPyPG4,1665
-saviialib/libs/sharepoint_client/sharepoint_client_contract.py,sha256=xqNHzCjp7GvUGGUox9YTJj2QJgTc5819t2etOk8X26o,485
-saviialib/libs/sharepoint_client/types/sharepoint_client_types.py,sha256=OmPlCJ9rLrAFBeG6aDp5cxMiQ5BZlDyGVx5S4GN4aqg,414
-saviialib/libs/zero_dependency/utils/datetime_utils.py,sha256=NFPHxOTuCtfkYjUnsbRqw9ZK87UGAK-Eira2CwG8rJ8,754
+saviialib/libs/sharepoint_client/__init__.py,sha256=6RbEzFtCfMrrY8F0JOCVcK6Gqt1FBJ09m0NuWHMCmCI,384
+saviialib/libs/sharepoint_client/clients/sharepoint_rest_api.py,sha256=Gq__zrYtSyiwi5NguMYCqJ5Bbs84UqzC9L5qvOMhfVI,6100
+saviialib/libs/sharepoint_client/sharepoint_client.py,sha256=gXFisWWCk6pKhoEeaNN8dcHA5ywrGDLqfiKjchrHRy0,1816
+saviialib/libs/sharepoint_client/sharepoint_client_contract.py,sha256=H-WsXR5f50jy5RRYNlgV61y_YGjOUq4U_E1DWUUrDYw,612
+saviialib/libs/sharepoint_client/types/sharepoint_client_types.py,sha256=JruUCn6o16w00t4zxDrxmv_Bxa52UyShGYmwUFfBCCQ,482
+saviialib/libs/zero_dependency/utils/datetime_utils.py,sha256=c2H_JpUKpunCzDw6I4alDWNftEqaciWGtEUklPepm_s,759
 saviialib/services/epii/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 saviialib/services/epii/api.py,sha256=99G-iC8Nz8vv9TuNPpdFsHxDAH8DXVCHnUQzwAUO9Ao,4036
 saviialib/services/epii/controllers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -42,12 +43,12 @@ saviialib/services/epii/use_cases/constants/upload_backup_to_sharepoint_constant
 saviialib/services/epii/use_cases/types/__init__.py,sha256=u6fyodOEJE2j6FMqJux40Xf9ccYAi-UUYxqT-Kzc0kE,199
 saviialib/services/epii/use_cases/types/update_thies_data_types.py,sha256=3lJzG1nuZoP1mqFlvQ0-aFJp80SECaeiROlvucVhaSY,539
 saviialib/services/epii/use_cases/types/upload_backup_to_sharepoint_types.py,sha256=J_sGhqSaPoMZA0GIrzCx6pmgSKgIkGi0uyrsltU_H8w,320
-saviialib/services/epii/use_cases/update_thies_data.py,sha256=SKW9_Z2pbvE0rgyFHmpBynGVqfQ30OMqK3zofW_tY8Y,9866
-saviialib/services/epii/use_cases/upload_backup_to_sharepoint.py,sha256=UZ2Pe_mavdquYG00_GYtANY4nn0wHjT_Z0HLUvtfW2c,11143
+saviialib/services/epii/use_cases/update_thies_data.py,sha256=0dkk45qPkvPpxS8uERPDrPN4woWz4vze3FR0gA-P6wI,9866
+saviialib/services/epii/use_cases/upload_backup_to_sharepoint.py,sha256=-0MvCzkFM7T2y7XLgFevWAXmqAsthzuY68ReiiL2zDU,11886
 saviialib/services/epii/utils/__init__.py,sha256=cYt2tvq65_OMjFaqb8-CCC7IGCQgFd4ziEUWJV7s1iY,98
 saviialib/services/epii/utils/update_thies_data_utils.py,sha256=EpjYWXqyHxJ-dO3MHhdXp-rGV7WyUckeFko-nnfnNac,555
 saviialib/services/epii/utils/upload_backup_to_sharepoint_utils.py,sha256=hEeV4_kcG8YL6t_3V8AlhgDHtHiUNsdYpilfgTLaQMc,3528
-saviialib-0.11.2.dist-info/LICENSE,sha256=NWpf6b38xgBWPBo5HZsCbdfp9hZSliEbRqWQgm0fkOo,1076
-saviialib-0.11.2.dist-info/METADATA,sha256=Af3aG28BKcG0Fsfds49sWCWJTBuWs4xo--RhfBsAjSE,4083
-saviialib-0.11.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-saviialib-0.11.2.dist-info/RECORD,,
+saviialib-1.0.1.dist-info/LICENSE,sha256=NWpf6b38xgBWPBo5HZsCbdfp9hZSliEbRqWQgm0fkOo,1076
+saviialib-1.0.1.dist-info/METADATA,sha256=qxj1sUazmjA38ue9RFtljIuW4xt9IJGyPSkBpS_LXqQ,4063
+saviialib-1.0.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+saviialib-1.0.1.dist-info/RECORD,,