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

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

@@ -0,0 +1,57 @@
+from collections.abc import Iterator
+from enum import Enum
+from itertools import cycle
+
+from textual.app import RenderResult
+from textual.widgets import Static
+
+
+class SpinnerType(Enum):
+    """Enumeration of various spinner animations."""
+
+    dots_1 = "⣷⣯⣟⡿⢿⣻⣽⣾"
+    dots_2 = "⠋⠙⠹⠸⠼⠴⠦⠧⠇⠏"
+    dots_3 = "⠋⠙⠚⠞⠖⠦⠴⠲⠳⠓"
+    dots_4 = "⠄⠆⠇⠋⠙⠸⠰⠠⠰⠸⠙⠋⠇⠆"
+    dots_5 = "⠈⠐⠠⢀⡀⠄⠂⠁"
+    dots_6 = "⋯⋱⋮⋰"
+    circles = "◐◓◑◒"
+    angles = "┐┤┘┴└├┌┬"
+    arrows = "←↖↑↗→↘↓↙"
+    moon = "🌑🌒🌓🌔🌕🌖🌗🌘"
+    clock = "🕛🕐🕑🕒🕓🕔🕕🕖🕗🕘🕙🕚"
+    histogram = "▁▃▄▅▆▇█▇▆▅▄▃"
+    shade = "░▒▓█▓▒░"
+    colors = "🟨🟨🟧🟧🟥🟥🟦🟦🟪🟪🟩🟩"
+    triangles = "◢◣◤◥"
+
+
+class Spinner(Static):
+    """A loading spinner widget for Textual apps."""
+
+    DEFAULT_CSS = """
+    Spinner {
+        color: #9E53E0;
+    }
+    """
+
+    def __init__(
+        self, spinner: SpinnerType = SpinnerType.dots_1, interval: float = 0.1, **kwargs
+    ):
+        """
+        Initialize the Loader widget.
+
+        Args:
+            spinner (SpinnerType): The spinner animation type.
+            interval (float): Time in seconds between frames.
+            **kwargs: Additional keyword arguments for Static.
+        """
+        super().__init__(**kwargs)
+        self._frames: Iterator[str] = cycle(spinner.value)
+        self._interval = interval
+
+    def render(self) -> RenderResult:
+        return next(self._frames)
+
+    def on_mount(self) -> None:
+        self.auto_refresh = self._interval

sourcerer/presentation/screens/storage_action_progress/main.py

@@ -298,7 +298,7 @@ class StorageActionProgressScreen(ModalScreen):
                 progress_bar.total = file_size
             except Exception as ex:
                 self.notify(
-                    f"Failed to get file size for {key}: {str(ex)}", severity="error"
+                    f"Failed to get file size for {key}: {ex}", severity="error"
                 )
                 self.log.error(f"Error getting file size: {ex}")
                 return
@@ -311,7 +311,7 @@ class StorageActionProgressScreen(ModalScreen):
                     lambda chunk: progress_callback(progress_bar, chunk),
                 )
             except Exception as ex:
-                self.notify(f"Failed to download {key}: {str(ex)}", severity="error")
+                self.notify(f"Failed to download {key}: {ex}", severity="error")
                 self.log.error(f"Error downloading file: {ex}")
                 return
 
@@ -324,9 +324,7 @@ class StorageActionProgressScreen(ModalScreen):
                     # Non-critical error, continue execution
         except Exception as ex:
             # Catch any unexpected exceptions
-            self.notify(
-                f"Unexpected error downloading {key}: {str(ex)}", severity="error"
-            )
+            self.notify(f"Unexpected error downloading {key}: {ex}", severity="error")
             self.log.error(f"Unexpected error: {ex}")
         finally:
             main_progress_bar.advance(1)

sourcerer/presentation/screens/storages_list/main.py

@@ -0,0 +1,184 @@
+import datetime
+import uuid
+from enum import Enum
+
+from dependency_injector.wiring import Provide
+from textual import on
+from textual.app import ComposeResult
+from textual.containers import Container, Horizontal, VerticalScroll
+from textual.reactive import reactive
+from textual.widgets import Label
+
+from sourcerer.domain.storage.entities import Storage
+from sourcerer.infrastructure.access_credentials.services import CredentialsService
+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,
+)
+from sourcerer.presentation.screens.storages_registration.main import (
+    StorageEntry,
+    StoragesRegistrationScreen,
+)
+
+
+class ControlsEnum(Enum):
+    CANCEL = "Cancel"
+    ADD_STORAGE = "Add Storage"
+
+
+class StorageRow(Horizontal):
+    def __init__(
+        self, storage: Storage, storages_service: StoragesService, *args, **kwargs
+    ):
+        super().__init__(*args, **kwargs)
+        self.storage = storage
+        self.storages_service = storages_service
+
+    def compose(self):
+        yield Label(self.storage.name, classes="storage_name")
+        yield Label(self.storage.credentials_name or "🚫", classes="credentials_name")
+        yield Button(
+            "❌",
+            name="delete_storage",
+            classes="storage_delete",
+        )
+
+    @on(Button.Click)
+    def on_button_click(self, _: Button.Click):
+        """
+        Handle delete button click events by showing a confirmation dialog for storage deletion.
+        Args:
+            _ (Button.Click): The button click event.
+        """
+        self.app.push_screen(
+            QuestionScreen(
+                f"Are you sure you want to delete {self.storage.credentials_name} {self.storage.name} storage?"
+            ),
+            callback=self.delete_callback,  # type: ignore
+        )
+
+    def delete_callback(self, result: bool):
+        """
+        Callback function to handle the result of the confirmation screen.
+
+        Args:
+            result (bool): True if the user confirmed, False otherwise.
+        """
+        if not result:
+            return
+        self.storages_service.delete(self.storage.uuid)
+        self.post_message(ReloadStoragesRequest())
+
+
+class StoragesListScreen(RefreshTriggerableModalScreen):
+    CSS_PATH = "styles.tcss"
+
+    MAIN_CONTAINER_ID = "StoragesListScreen"
+    SETTINGS_CONTAINER_ID = "settings"
+
+    storages_list = reactive([], recompose=True)
+
+    def __init__(
+        self,
+        credentials_service: CredentialsService = Provide[
+            DiContainer.credentials_service
+        ],
+        storages_service: StoragesService = Provide[DiContainer.storages_service],
+        *args,
+        **kwargs,
+    ):
+        super().__init__(*args, **kwargs)
+        self.storage_service = storages_service
+        self.credentials_service = credentials_service
+
+    def compose(self) -> ComposeResult:
+        with Container(id=self.MAIN_CONTAINER_ID):
+            yield Container(
+                Button(
+                    "+Add new storage",
+                    name=ControlsEnum.ADD_STORAGE.name,
+                    classes="add_storage_button",
+                ),
+                id="right-top",
+            )
+            with VerticalScroll(id=self.SETTINGS_CONTAINER_ID):
+                with Horizontal():
+                    yield Label("Storage Name", classes="storage_name")
+                    yield Label("Credentials Name", classes="credentials_name")
+                    yield Label("Delete", classes="storage_delete")
+                for storage in self.storages_list:
+                    yield StorageRow(storage, self.storage_service)
+            with Horizontal(id="controls"):
+                yield Button(ControlsEnum.CANCEL.value, name=ControlsEnum.CANCEL.name)
+
+    def on_compose(self):
+        """
+        Initialize the screen by refreshing the credentials list when the screen is composed.
+        """
+        self.refresh_storages_list(set_refresh_flag=False)
+
+    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):
+        """
+        Handle the reload storages request by refreshing the storages list.
+
+        Args:
+            _: ReloadStoragesRequest: The reload storages request message.
+        """
+        self.refresh_storages_list()
+
+    @on(Button.Click)
+    def on_control_button_click(self, event: Button.Click):
+        """
+        Handle click events for control buttons.
+
+        Dismisses the screen if the cancel button is clicked, or opens the provider credentials registration screen if
+        the add registration button is clicked.
+
+        Args:
+            event (Button.Click): The button click event.
+        """
+        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
+            )
+
+    def create_storage_entry(self, storage: StorageEntry | None):
+        """
+        Create a new storage entry.
+
+        Creates a new storage entry using the provided data and refreshes the storage list.
+        """
+        if storage is None:
+            return
+
+        credentials = self.credentials_service.get(storage.credentials_uuid)
+        if not credentials:
+            self.notify("Credentials not found", severity="error")
+            return
+        self.storage_service.create(
+            Storage(
+                uuid=str(uuid.uuid4()),
+                name=storage.name,
+                credentials_id=credentials.id,
+                date_created=datetime.datetime.now(),
+            )
+        )
+        self.refresh_storages_list()

sourcerer/presentation/screens/storages_list/messages/reload_storages_request.py

@@ -0,0 +1,8 @@
+from dataclasses import dataclass
+
+from textual.message import Message
+
+
+@dataclass
+class ReloadStoragesRequest(Message):
+    pass

sourcerer/presentation/screens/storages_list/styles.tcss

@@ -0,0 +1,55 @@
+
+Container {
+  height: auto;
+}
+
+Label {
+  padding-left: 1;
+}
+
+StoragesListScreen {
+  align: center middle;
+  content-align: center top;
+
+  & > #StoragesListScreen {
+    padding: 1 2 0 2;
+    margin: 0 0;
+    width: 60%;
+    height: 60%;
+    border: solid $secondary-background;
+
+    #right-top {
+      align: right top;
+
+      .add_storage_button {
+        color: $success-darken-2;
+      }
+
+      margin-bottom: 1;
+    }
+
+    ProviderCredentialsRow.active {
+      background: $primary-lighten-2;
+    }
+
+    ProviderCredentialsRow, Horizontal {
+      height: auto;
+    }
+
+    .storage_name {
+      width: 45%;
+    }
+
+    .credentials_name {
+      width: 35%;
+    }
+
+    .storage_delete {
+      width: 20%;
+    }
+
+    & > Select {
+      margin-bottom: 1;
+    }
+  }
+}

sourcerer/presentation/screens/storages_registration/main.py

@@ -0,0 +1,100 @@
+from dataclasses import dataclass
+from enum import Enum
+
+from dependency_injector.wiring import Provide
+from textual import on
+from textual.app import ComposeResult
+from textual.containers import Container, Horizontal, VerticalScroll
+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
+
+
+class ControlsEnum(Enum):
+    CANCEL = "Cancel"
+    CREATE = "Create"
+
+
+@dataclass
+class StorageEntry:
+    name: str
+    credentials_uuid: str
+
+
+class StoragesRegistrationScreen(ExitBoundModalScreen):
+    CSS_PATH = "styles.tcss"
+
+    MAIN_CONTAINER_ID = "StoragesRegistrationScreen"
+    SETTINGS_CONTAINER_ID = "settings"
+    CREDENTIALS_SELECTOR_ID = "credentials_selector"
+
+    PROVIDERS_NAME = "credentials"
+
+    def __init__(
+        self,
+        credentials_service: CredentialsService = Provide[
+            DiContainer.credentials_repository
+        ],
+        *args,
+        **kwargs,
+    ):
+        super().__init__(*args, **kwargs)
+        self.credentials = credentials_service.list()
+        self.auth_method = None
+
+    def compose(self) -> ComposeResult:
+        with Container(id=self.MAIN_CONTAINER_ID):
+            with VerticalScroll(id=self.SETTINGS_CONTAINER_ID):
+                yield LabeledInput(
+                    "storage_name",
+                    "Name:",
+                    True,
+                    multiline=False,
+                    id="storage_name",
+                )
+                yield Label("* Credentials:", classes="form_label")
+                yield Select(
+                    options=((item.name, item.uuid) for item in self.credentials),
+                    name=self.PROVIDERS_NAME,
+                    id=self.CREDENTIALS_SELECTOR_ID,
+                )
+            with Horizontal(id="controls"):
+                yield Button(ControlsEnum.CANCEL.value, name=ControlsEnum.CANCEL.name)
+                yield Button(ControlsEnum.CREATE.value, name=ControlsEnum.CREATE.name)
+
+    @on(Button.Click)
+    def on_control_button_click(self, event: Button.Click):
+        """
+        Handle click events for control buttons on the registration screen.
+
+        Depending on the action associated with the button click event, either dismiss
+        the screen or gather authentication fields and then dismiss the screen with
+        the collected data.
+
+        Args:
+            event (Button.Click): The click event containing the action to be performed.
+
+        Flow:
+            1. Check if the event.action is ControlsEnum.cancel.name. If true, dismiss the screen.
+            2. If event.action is ControlsEnum.create.name, gather authentication fields. Dismiss the screen with the
+               collected authentication fields.
+        """
+        if event.action == ControlsEnum.CANCEL.name:
+            self.action_cancel_screen()
+        elif event.action == ControlsEnum.CREATE.name:
+            storage_name = self.query_one("#storage_name", LabeledInput).get().value
+            if not storage_name:
+                self.notify("Storage name is required", severity="error")
+                return
+            credentials_uuid = self.query_one(
+                f"#{self.CREDENTIALS_SELECTOR_ID}", Select
+            ).value
+            if not credentials_uuid or credentials_uuid == Select.BLANK:
+                self.notify("Credentials are required", severity="error")
+                return
+
+            self.dismiss(StorageEntry(storage_name, credentials_uuid))  # type: ignore

sourcerer/presentation/screens/storages_registration/styles.tcss

@@ -0,0 +1,41 @@
+
+Container {
+  height: auto;
+}
+
+Label {
+  padding-left: 1;
+}
+
+StoragesRegistrationScreen {
+  align: center middle;
+  content-align: center top;
+
+  & > #StoragesRegistrationScreen {
+    padding: 1 2 0 2;
+    margin: 0 0;
+    width: 70;
+    height: 17;
+    border: solid $secondary-background;
+
+    & > #settings {
+      width: 100%;
+      height: 11;
+
+      & > .form_label {
+        padding-left: 1;
+        margin-bottom: 1;
+
+      }
+
+      & > #credentials_fields_container {
+        margin-top: 2;
+        padding-left: 2;
+      }
+
+      & > Select {
+         margin-bottom: 1;
+      }
+    }
+  }
+}

sourcerer/presentation/settings.py

@@ -3,29 +3,42 @@
 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 = """
-          ###                 ###      
-          ###                 ###      
-             ####         ####         
-             ####         ####         
-          #######################      
-       ######    #########    ######   
-       ######    #########    ######   
+          ###                 ###
+          ###                 ###
+             ####         ####
+             ####         ####
+          #######################
+       ######    #########    ######
+       ######    #########    ######
     ###################################
     ###################################
     ###   #######################   ###
     ###   #######################   ###
     ###   ###                 ###   ###
-             #######   #######         
-             #######   #######         
-             
-            
-###   ##                ##         #         
-####  ##  ####      ######  ####  #### ##### 
-## ## ## ##  ##    ##   ##   ####  #    #### 
-##  #### ##  ##    ##   ## ##  ##  #  ##  ## 
-##    ##  ####       ##### ######  ### ### # 
+             #######   #######
+             #######   #######
+
+
+###   ##                ##         #
+####  ##  ####      ######  ####  #### #####
+## ## ## ##  ##    ##   ##   ####  #    ####
+##  #### ##  ##    ##   ## ##  ##  #  ##  ##
+##    ##  ####       ##### ######  ### ### #
 """
 
 MIN_SECTION_DIMENSION = 10
+
+
+class KeyBindings(Enum):
+    """Enum representing key bindings for various actions in the application."""
+
+    ARROW_DOWN = "down"
+    ARROW_UP = "up"
+    ARROW_RIGHT = "right"
+    ARROW_LEFT = "left"
+    ENTER = "enter"
+    BACKSPACE = "backspace"
+    CTRL = "ctrl"

sourcerer/presentation/utils.py

@@ -4,13 +4,16 @@ 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
 from sourcerer.domain.storage_provider.services import BaseStorageProviderService
 from sourcerer.infrastructure.access_credentials.exceptions import CredentialsAuthError
 from sourcerer.infrastructure.access_credentials.registry import (
     access_credential_method_registry,
 )
 from sourcerer.infrastructure.storage_provider.registry import storage_provider_registry
+from sourcerer.presentation.di_container import DiContainer
 
 
 def get_provider_service_by_access_uuid(
@@ -32,6 +35,9 @@ def get_provider_service_by_access_uuid(
 
 def get_provider_service_by_access_credentials(
     credentials,
+    credentials_repo: BaseCredentialsRepository = Provide[
+        DiContainer.credentials_repository
+    ],
 ) -> BaseStorageProviderService | None:
     """
     Retrieves a storage provider service instance using the given access credentials.
@@ -67,7 +73,9 @@ def get_provider_service_by_access_credentials(
         return None
 
     try:
-        auth_credentials = credentials_service().authenticate(credentials.credentials)
+        auth_credentials = credentials_service(credentials_repo).authenticate(
+            credentials.credentials
+        )
     except CredentialsAuthError:
         return None
     return provider_service_class(auth_credentials)

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/presentation/screens/shared/widgets/loader.py

@@ -1,31 +0,0 @@
-from textual.app import ComposeResult
-from textual.containers import Container
-from textual.widgets import Static
-
-
-class Loader(Container):
-    stops = "⣷⣯⣟⡿⢿⣻⣽⣾"
-    index = 0
-
-    DEFAULT_CSS = """
-    Loader {
-        color: #9E53E0;
-    }
-    """
-
-    def __init__(self, *args, **kwargs):
-        self.label = Static(self.stops[self.index])
-        super().__init__(*args, **kwargs)
-
-    def compose(self) -> ComposeResult:
-        yield self.label
-
-    def on_mount(self) -> None:
-        self.auto_refresh = 1 / 10
-
-    def automatic_refresh(self):
-        self.index += 1
-        if self.index >= len(self.stops):
-            self.index = 0
-        value = self.stops[self.index]
-        self.label.update(value)