data-sourcerer 0.4.0__py3-none-any.whl → 0.6.0__py3-none-any.whl

sourcerer/presentation/screens/storages_list/main.py

@@ -1,10 +1,12 @@
 import datetime
 import uuid
 from enum import Enum
+from typing import ClassVar
 
 from dependency_injector.wiring import Provide
 from textual import on
 from textual.app import ComposeResult
+from textual.binding import Binding, BindingType
 from textual.containers import Container, Horizontal, VerticalScroll
 from textual.reactive import reactive
 from textual.widgets import Label
@@ -82,6 +84,11 @@ class StoragesListScreen(RefreshTriggerableModalScreen):
     MAIN_CONTAINER_ID = "StoragesListScreen"
     SETTINGS_CONTAINER_ID = "settings"
 
+    BINDINGS: ClassVar[list[BindingType]] = [
+        *RefreshTriggerableModalScreen.BINDINGS,
+        Binding("ctrl+n", "add_storage", "Add new storage"),
+    ]
+
     storages_list = reactive([], recompose=True)
 
     def __init__(
@@ -104,6 +111,7 @@ class StoragesListScreen(RefreshTriggerableModalScreen):
                     "+Add new storage",
                     name=ControlsEnum.ADD_STORAGE.name,
                     classes="add_storage_button",
+                    id="add_storage_button",
                 ),
                 id="right-top",
             )
@@ -155,10 +163,7 @@ class StoragesListScreen(RefreshTriggerableModalScreen):
         if event.action == ControlsEnum.CANCEL.name:
             self.action_cancel_screen()
         if event.action == ControlsEnum.ADD_STORAGE.name:
-            self.app.push_screen(
-                StoragesRegistrationScreen(),
-                callback=self.create_storage_entry,  # type: ignore
-            )
+            self.action_add_storage()
 
     def create_storage_entry(self, storage: StorageEntry | None):
         """
@@ -182,3 +187,9 @@ class StoragesListScreen(RefreshTriggerableModalScreen):
             )
         )
         self.refresh_storages_list()
+
+    def action_add_storage(self):
+        self.app.push_screen(
+            StoragesRegistrationScreen(),
+            callback=self.create_storage_entry,  # type: ignore
+        )

sourcerer/presentation/screens/storages_list/styles.tcss

@@ -23,6 +23,13 @@ StoragesListScreen {
 
       .add_storage_button {
         color: $success-darken-2;
+
+        &:hover {
+            color: $primary;
+        }
+        &:focus {
+            color: $primary;
+        }
       }
 
       margin-bottom: 1;

sourcerer/presentation/settings.py

@@ -3,6 +3,7 @@
 This module contains various configuration constants and settings used throughout
 the application, including UI elements and display configurations.
 """
+
 from enum import Enum
 
 NO_DATA_LOGO = """

sourcerer/presentation/utils.py

@@ -4,6 +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
@@ -14,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(
@@ -59,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
     )
@@ -78,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

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

sourcerer/utils.py

@@ -1,8 +1,10 @@
-# Generate and store encryption key in a file
+import contextlib
 import os
 import uuid
 from pathlib import Path
 
+import requests
+
 
 def get_encryption_key(path: Path) -> str:
     """
@@ -30,3 +32,19 @@ def get_encryption_key(path: Path) -> str:
         f.write(new_key)
 
     return new_key
+
+
+def get_last_package_version(name):
+    """
+    Fetch the latest version of a package from PyPI.
+    Args:
+        name (str): The name of the package.
+    Returns:
+        str: The latest version of the package, or None if an error occurs.
+    """
+    with contextlib.suppress(Exception):
+        url = f"https://pypi.org/pypi/{name}/json"
+        response = requests.get(url, timeout=5)
+        response.raise_for_status()
+        return response.json()["info"]["version"]
+    return None