data-sourcerer 0.5.0__tar.gz → 0.6.0__tar.gz

{data_sourcerer-0.5.0 → data_sourcerer-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: data-sourcerer
-Version: 0.5.0
+Version: 0.6.0
 Summary: Sourcerer is a terminal cloud storage navigator.
 Author-email: Bohdana Kuzmenko <bohdana.kuzmenko.dev@gmail.com>
 License: MIT
@@ -59,6 +59,10 @@ engineers to view and manage files across multiple cloud providers like
 
 ## 🪄 Installation
 
+```bash
+pipx install data-sourcerer
+```
+or
 ```bash
 pip install data-sourcerer
 ```

{data_sourcerer-0.5.0 → data_sourcerer-0.6.0}/README.md

@@ -22,6 +22,10 @@ engineers to view and manage files across multiple cloud providers like
 
 ## 🪄 Installation
 
+```bash
+pipx install data-sourcerer
+```
+or
 ```bash
 pip install data-sourcerer
 ```

{data_sourcerer-0.5.0 → data_sourcerer-0.6.0}/sourcerer/__init__.py

@@ -14,4 +14,4 @@ The application is structured using a clean architecture approach with:
 
 name = "sourcerer"
 package_name = "data_sourcerer"
-__version__ = "0.5.0"
+__version__ = "0.6.0"

{data_sourcerer-0.5.0 → data_sourcerer-0.6.0}/sourcerer/infrastructure/storage_provider/services/azure.py

@@ -6,12 +6,14 @@ interface for various cloud storage providers.
 """
 
 import os.path
+import threading
 from collections.abc import Callable
 from pathlib import Path
 from typing import Any
 
 from azure.mgmt.storage import StorageManagementClient
 from azure.storage.blob import BlobServiceClient
+from cachetools import LRUCache
 from platformdirs import user_downloads_dir
 
 from sourcerer.domain.shared.entities import StorageProvider
@@ -37,6 +39,8 @@ from sourcerer.infrastructure.utils import generate_uuid, is_text_file
 
 @storage_provider(StorageProvider.AzureStorage)
 class AzureStorageProviderService(BaseStorageProviderService):
+    MAX_CACHE_SIZE = 10
+
     def __init__(self, credentials: Any):
         """
         Initialize the service with Azure credentials.
@@ -48,6 +52,12 @@ class AzureStorageProviderService(BaseStorageProviderService):
         self.subscription_id = credentials.subscription_id
         self.cloud_suffix = credentials.cloud_suffix
 
+        self._storage_management_client: StorageManagementClient | None = None
+        self._blob_service_clients_lock = threading.Lock()
+        self._blob_service_clients: LRUCache[str, BlobServiceClient] = LRUCache(
+            maxsize=self.MAX_CACHE_SIZE
+        )
+
     def get_accounts_client(self) -> StorageManagementClient:
         """
         Get the Azure accounts client.
@@ -55,7 +65,13 @@ class AzureStorageProviderService(BaseStorageProviderService):
         Returns:
             Any: Azure accounts client
         """
-        return StorageManagementClient(self.credentials, self.subscription_id)
+        if self._storage_management_client:
+            return self._storage_management_client
+
+        self._storage_management_client = StorageManagementClient(
+            self.credentials, self.subscription_id
+        )
+        return self._storage_management_client
 
     def get_containers_client(self, storage: str):
         """
@@ -69,12 +85,19 @@ class AzureStorageProviderService(BaseStorageProviderService):
             BlobServiceClient: An instance of the BlobServiceClient, configured with the
             account URL and credentials.
         """
+        with self._blob_service_clients_lock:
+            if (client := self._blob_service_clients.get(storage)) is not None:
+                return client
+
         account_url = "https://{account}.{cloud_suffix}"
-        return BlobServiceClient(
+        client = BlobServiceClient(
             account_url.format(account=storage, cloud_suffix=self.cloud_suffix),
             credential=self.credentials,
             retry_connect=0,
         )
+        with self._blob_service_clients_lock:
+            self._blob_service_clients[storage] = client
+        return client
 
     def list_storages(self) -> list[Storage]:
         """

{data_sourcerer-0.5.0 → data_sourcerer-0.6.0}/sourcerer/presentation/screens/question/styles.tcss

@@ -6,7 +6,7 @@ QuestionScreen {
     padding: 1 2 0 2;
     margin: 0 0;
     width: 50;
-    height: 8;
+    height: 10;
     border: solid $warning-darken-1;
 
     & > Static {

{data_sourcerer-0.5.0 → data_sourcerer-0.6.0}/sourcerer/presentation/utils.py

@@ -4,7 +4,9 @@ Utility functions for the presentation layer.
 This module provides helper functions for the presentation layer,
 particularly for retrieving and initializing storage provider services.
 """
+from threading import Lock
 
+from cachetools import LRUCache
 from dependency_injector.wiring import Provide
 
 from sourcerer.domain.access_credentials.repositories import BaseCredentialsRepository
@@ -15,6 +17,10 @@ from sourcerer.infrastructure.access_credentials.registry import (
 )
 from sourcerer.infrastructure.storage_provider.registry import storage_provider_registry
 from sourcerer.presentation.di_container import DiContainer
+from sourcerer.settings import MAX_CREDENTIALS_CACHE_SIZE
+
+_provider_service_cache: LRUCache = LRUCache(maxsize=MAX_CREDENTIALS_CACHE_SIZE)
+_provider_service_cache_lock = Lock()
 
 
 def get_provider_service_by_access_uuid(
@@ -60,6 +66,10 @@ def get_provider_service_by_access_credentials(
            authenticated credentials.
     """
 
+    with _provider_service_cache_lock:
+        if credentials.uuid in _provider_service_cache:
+            return _provider_service_cache[credentials.uuid]
+
     credentials_service = access_credential_method_registry.get_by_provider_and_name(
         credentials.provider, credentials.credentials_type
     )
@@ -79,4 +89,7 @@ def get_provider_service_by_access_credentials(
         )
     except CredentialsAuthError:
         return None
-    return provider_service_class(auth_credentials)
+    service = provider_service_class(auth_credentials)
+    with _provider_service_cache_lock:
+        _provider_service_cache[credentials.uuid] = service
+    return service

{data_sourcerer-0.5.0 → data_sourcerer-0.6.0}/sourcerer/settings.py

@@ -76,3 +76,5 @@ TEXT_EXTENSIONS = {
 PAGE_SIZE = 100
 PREVIEW_LENGTH_LIMIT = 10_000
 PREVIEW_LIMIT_SIZE = 2 * 1024 * 1024  # 2 MB
+
+MAX_CREDENTIALS_CACHE_SIZE = 30