data-sourcerer 0.3.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

sourcerer/presentation/screens/shared/modal_screens.py

@@ -0,0 +1,37 @@
+from typing import ClassVar
+
+from textual.binding import Binding, BindingType
+from textual.screen import ModalScreen
+
+
+class ExitBoundModalScreen(ModalScreen):
+    """
+    A base class for modal screens that can be exited.
+    It provides a method to exit the screen and a flag to indicate if the screen should be exited.
+    """
+
+    BINDINGS: ClassVar[list[BindingType]] = [
+        Binding("escape", "cancel_screen", "Pop screen"),
+    ]
+
+    def action_cancel_screen(self):
+        """
+        Action to exit the screen.
+        """
+        self.dismiss()
+
+
+class RefreshTriggerableModalScreen(ExitBoundModalScreen):
+    """
+    A base class for modal screens that can be refreshed.
+    It provides a method to refresh the screen and a flag to indicate if the screen should be refreshed.
+    """
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self._requires_storage_refresh = False
+
+    def action_cancel_screen(self):
+        requires_storage_refresh = self._requires_storage_refresh
+        self._requires_storage_refresh = False
+        self.dismiss(requires_storage_refresh)

sourcerer/presentation/screens/shared/widgets/button.py

@@ -28,6 +28,7 @@ class Button(Label):
         }
     }
     """
+    can_focus = True
 
     @dataclass
     class Click(Message):
@@ -52,3 +53,13 @@ class Button(Label):
             _: An instance of events.Click representing the click event.
         """
         self.post_message(self.Click(self.name))  # type: ignore
+
+    def on_key(self, event: events.Key) -> None:
+        """
+        Handle key events to trigger click action when the button is focused and activated.
+
+        Args:
+            event (events.Key): The key event to handle.
+        """
+        if event.key == "enter":
+            self.post_message(self.Click(self.name))  # type: ignore

sourcerer/presentation/screens/shared/widgets/labeled_input.py

@@ -77,8 +77,6 @@ class LabeledInput(Container):
         """
         input_area = self.query_one(".form_input")
         text = (
-            input_area.document.text
-            if isinstance(input_area, TextArea)
-            else input_area.value  # type: ignore
+            input_area.document.text if isinstance(input_area, TextArea) else input_area.value  # type: ignore
         )
         return self.Value(name=self.key, value=text)

sourcerer/presentation/screens/storage_action_progress/main.py

@@ -369,7 +369,7 @@ class StorageActionProgressScreen(ModalScreen):
             key (str): The key/path of the file to delete
             uuid (str): Unique identifier for the file
             main_progress_bar (ProgressBar): The main progress bar to update
-        """ ""
+        """
         if not self.provider_service:
             self.notify(f"Failed to delete {key}", severity="error")
             main_progress_bar.advance(1)
@@ -418,7 +418,6 @@ class StorageActionProgressScreen(ModalScreen):
                 finally:
                     self.files_has_been_processed = True
             elif source_path.is_dir():
-
                 files_n = len([i for i in source_path.rglob("*") if i.is_file()])
                 progress_bar = self.query_one(f"#progress_bar_{key.uuid}", ProgressBar)
                 progress_bar.total = files_n

sourcerer/presentation/screens/storages_list/main.py

@@ -1,13 +1,14 @@
 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.screen import ModalScreen
 from textual.widgets import Label
 
 from sourcerer.domain.storage.entities import Storage
@@ -15,6 +16,9 @@ from sourcerer.infrastructure.access_credentials.services import CredentialsServ
 from sourcerer.infrastructure.storage.services import StoragesService
 from sourcerer.presentation.di_container import DiContainer
 from sourcerer.presentation.screens.question.main import QuestionScreen
+from sourcerer.presentation.screens.shared.modal_screens import (
+    RefreshTriggerableModalScreen,
+)
 from sourcerer.presentation.screens.shared.widgets.button import Button
 from sourcerer.presentation.screens.storages_list.messages.reload_storages_request import (
     ReloadStoragesRequest,
@@ -74,12 +78,17 @@ class StorageRow(Horizontal):
         self.post_message(ReloadStoragesRequest())
 
 
-class StoragesListScreen(ModalScreen):
+class StoragesListScreen(RefreshTriggerableModalScreen):
     CSS_PATH = "styles.tcss"
 
     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__(
@@ -102,6 +111,7 @@ class StoragesListScreen(ModalScreen):
                     "+Add new storage",
                     name=ControlsEnum.ADD_STORAGE.name,
                     classes="add_storage_button",
+                    id="add_storage_button",
                 ),
                 id="right-top",
             )
@@ -119,13 +129,15 @@ class StoragesListScreen(ModalScreen):
         """
         Initialize the screen by refreshing the credentials list when the screen is composed.
         """
-        self.refresh_storages_list()
+        self.refresh_storages_list(set_refresh_flag=False)
 
-    def refresh_storages_list(self):
+    def refresh_storages_list(self, set_refresh_flag: bool = True):
         """
         Refresh the storages list by retrieving the latest storages from the storage service.
         """
         self.storages_list = self.storage_service.list()
+        if set_refresh_flag:
+            self._requires_storage_refresh = True
 
     @on(ReloadStoragesRequest)
     def on_reload_storages_request(self, _: ReloadStoragesRequest):
@@ -149,12 +161,9 @@ class StoragesListScreen(ModalScreen):
             event (Button.Click): The button click event.
         """
         if event.action == ControlsEnum.CANCEL.name:
-            self.dismiss()
+            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):
         """
@@ -178,3 +187,9 @@ class StoragesListScreen(ModalScreen):
             )
         )
         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/screens/storages_registration/main.py

@@ -5,11 +5,11 @@ from dependency_injector.wiring import Provide
 from textual import on
 from textual.app import ComposeResult
 from textual.containers import Container, Horizontal, VerticalScroll
-from textual.screen import ModalScreen
 from textual.widgets import Label, Select
 
 from sourcerer.infrastructure.access_credentials.services import CredentialsService
 from sourcerer.presentation.di_container import DiContainer
+from sourcerer.presentation.screens.shared.modal_screens import ExitBoundModalScreen
 from sourcerer.presentation.screens.shared.widgets.button import Button
 from sourcerer.presentation.screens.shared.widgets.labeled_input import LabeledInput
 
@@ -25,7 +25,7 @@ class StorageEntry:
     credentials_uuid: str
 
 
-class StoragesRegistrationScreen(ModalScreen):
+class StoragesRegistrationScreen(ExitBoundModalScreen):
     CSS_PATH = "styles.tcss"
 
     MAIN_CONTAINER_ID = "StoragesRegistrationScreen"
@@ -84,7 +84,7 @@ class StoragesRegistrationScreen(ModalScreen):
                collected authentication fields.
         """
         if event.action == ControlsEnum.CANCEL.name:
-            self.dismiss()
+            self.action_cancel_screen()
         elif event.action == ControlsEnum.CREATE.name:
             storage_name = self.query_one("#storage_name", LabeledInput).get().value
             if not storage_name:

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,7 @@ Utility functions for the presentation layer.
 This module provides helper functions for the presentation layer,
 particularly for retrieving and initializing storage provider services.
 """
+
 from dependency_injector.wiring import Provide
 
 from sourcerer.domain.access_credentials.repositories import BaseCredentialsRepository

sourcerer/settings.py

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

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