azure-explorer 0.1.0__tar.gz

azure_explorer-0.1.0/PKG-INFO

@@ -0,0 +1,80 @@
+Metadata-Version: 2.4
+Name: azure-explorer
+Version: 0.1.0
+Summary: 
+Author: uch
+Author-email: uch@energinet.dk
+Requires-Python: >=3.10.8,<4.0.0
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Requires-Dist: azure-data-tables (>=12.7.0,<13.0.0)
+Requires-Dist: azure-identity (>=1.25.1,<2.0.0)
+Requires-Dist: azure-mgmt-resource (>=24.0.0,<25.0.0)
+Requires-Dist: azure-mgmt-servicebus (>=9.0.0,<10.0.0)
+Requires-Dist: azure-mgmt-storage (>=24.0.0,<25.0.0)
+Requires-Dist: azure-mgmt-subscription (>=3.1.1,<4.0.0)
+Requires-Dist: azure-servicebus (>=7.14.3,<8.0.0)
+Requires-Dist: azure-storage-blob (>=12.27.1,<13.0.0)
+Requires-Dist: azure-storage-file-datalake (>=12.22.0,<13.0.0)
+Requires-Dist: textual (>=7.0.0,<8.0.0)
+Description-Content-Type: text/markdown
+
+# 🔎 Azure Explorer
+
+The purpose of this package is to provide a simple interface for exploring resources in your Azure Cloud environment. The package consists of a CLI tool and a Python library.
+
+---
+
+## Setup
+
+Install the latest version of Azure Explorer with
+
+```console
+pip install azure-explorer
+```
+
+---
+
+## Usage
+
+##### CLI tool
+
+The CLI tool provides a way of navigating your Azure Cloud resources in the command line. Run the CLI tool with
+
+```console
+ax
+```
+
+![cli-tool](docs/cli-tool.png)
+
+##### Python library
+
+The Python library provides a way of interacting with our Azure Cloud resources in in Python code. Use the Python library with
+
+```python
+>>> import azure_explorer as ax
+>>> tenant = ax.get_tenant_manager()
+>>> tenant.list_subscription_names()
+[
+    'development-subscription',
+    'preproduction-subscription',
+    'production-subscription',
+    'test-subscription',
+]
+
+>>> container = ax.get_container_manager(
+    storage_account_name='sadataprod',
+    container_name='documents',
+)
+>>> container.list_dir()
+[
+    'dir1',
+    'dir2',
+    'file1.txt',
+    'file2.jpg',
+    'file3.pdf',
+]
+```

azure_explorer-0.1.0/README.md

@@ -0,0 +1,56 @@
+# 🔎 Azure Explorer
+
+The purpose of this package is to provide a simple interface for exploring resources in your Azure Cloud environment. The package consists of a CLI tool and a Python library.
+
+---
+
+## Setup
+
+Install the latest version of Azure Explorer with
+
+```console
+pip install azure-explorer
+```
+
+---
+
+## Usage
+
+##### CLI tool
+
+The CLI tool provides a way of navigating your Azure Cloud resources in the command line. Run the CLI tool with
+
+```console
+ax
+```
+
+![cli-tool](docs/cli-tool.png)
+
+##### Python library
+
+The Python library provides a way of interacting with our Azure Cloud resources in in Python code. Use the Python library with
+
+```python
+>>> import azure_explorer as ax
+>>> tenant = ax.get_tenant_manager()
+>>> tenant.list_subscription_names()
+[
+    'development-subscription',
+    'preproduction-subscription',
+    'production-subscription',
+    'test-subscription',
+]
+
+>>> container = ax.get_container_manager(
+    storage_account_name='sadataprod',
+    container_name='documents',
+)
+>>> container.list_dir()
+[
+    'dir1',
+    'dir2',
+    'file1.txt',
+    'file2.jpg',
+    'file3.pdf',
+]
+```

azure_explorer-0.1.0/azure_explorer/__init__.py

@@ -0,0 +1,13 @@
+from .managers import (
+    ContainerManager,
+    StorageAccountManager,
+    SubscriptionManager,
+    TenantManager,
+)
+from .short_cuts import (
+    get_container_manager,
+    get_storage_account_manager,
+    get_subscription_manager,
+    get_tenant_manager,
+    read_file,
+)

azure_explorer-0.1.0/azure_explorer/config.py

@@ -0,0 +1,5 @@
+from azure.identity import DefaultAzureCredential
+
+
+class Config:
+    credential = DefaultAzureCredential()

azure_explorer-0.1.0/azure_explorer/managers/__init__.py

@@ -0,0 +1,6 @@
+from .base import Manager
+from .container import ContainerManager
+from .models import ResourceProperties, SubscriptionProperties
+from .storage_account import StorageAccountManager
+from .subscription import SubscriptionManager
+from .tenant import TenantManager

azure_explorer-0.1.0/azure_explorer/managers/base.py

@@ -0,0 +1,5 @@
+from abc import ABC
+
+
+class Manager(ABC):
+    pass

azure_explorer-0.1.0/azure_explorer/managers/container/__init__.py

@@ -0,0 +1,3 @@
+from .base import ContainerManager
+from .blob_storage import BlobStorageContainerManager
+from .data_lake import DataLakeContainerManager

azure_explorer-0.1.0/azure_explorer/managers/container/base.py

@@ -0,0 +1,131 @@
+from abc import ABC, abstractmethod
+from io import BytesIO
+from pathlib import Path
+from typing import Iterator
+
+from azure.core.credentials import TokenCredential
+from azure.storage.blob import StorageStreamDownloader
+
+
+class FolderNotFoundError(Exception):
+    pass
+
+
+class FileOrFolderNotFoundError(Exception):
+    pass
+
+
+class ContainerManager(ABC):
+    @staticmethod
+    def create(
+        storage_account_name: str,
+        container_name: str,
+        is_data_lake: bool,
+        credential: TokenCredential,
+    ):
+        from . import BlobStorageContainerManager, DataLakeContainerManager
+
+        if is_data_lake:
+            return DataLakeContainerManager(
+                storage_account_name=storage_account_name,
+                container_name=container_name,
+                credential=credential,
+            )
+        else:
+            return BlobStorageContainerManager(
+                storage_account_name=storage_account_name,
+                container_name=container_name,
+                credential=credential,
+            )
+
+    @abstractmethod
+    def _get_blob_stream_downloader(self, path: Path | str) -> StorageStreamDownloader:
+        ...
+
+    @abstractmethod
+    def _write_blob_data(
+        self, path: Path | str, data: bytes | BytesIO, overwrite: bool = False
+    ):
+        ...
+
+    @abstractmethod
+    def _iter_dir(self, path: Path | str) -> Iterator[str]:
+        ...
+
+    @abstractmethod
+    def is_dir(self, path: Path | str) -> bool:
+        ...
+
+    @abstractmethod
+    def is_file(self, path: Path | str) -> bool:
+        ...
+
+    def list_dir(self, path: Path | str = "") -> list[str]:
+        if not self.is_dir(path):
+            raise FolderNotFoundError(path)
+
+        subpaths = list(self._iter_dir(path))
+
+        return sorted(subpaths)
+
+    def read_file(self, path: Path | str) -> bytes:
+        if not self.is_file(path):
+            raise FileNotFoundError(path)
+
+        blob_stream_downloader = self._get_blob_stream_downloader(path)
+
+        return blob_stream_downloader.readall()
+
+    def download_file(self, source: Path | str, destination: Path | str):
+        destination = Path(destination)
+
+        if not self.is_file(source):
+            raise FileNotFoundError(source)
+
+        destination.parent.mkdir(parents=True, exist_ok=True)
+
+        blob_stream_downloader = self._get_blob_stream_downloader(source)
+
+        with open(destination, "wb") as f:
+            blob_stream_downloader.readinto(f)
+
+    def download_dir(self, source: Path | str, destination: Path):
+        destination = Path(destination)
+
+        if not self.is_dir(source):
+            raise FolderNotFoundError(source)
+
+        for subsource in self.list_dir(source):
+            basename = Path(subsource).relative_to(source)
+
+            subdestination = destination / basename
+
+            if self.is_dir(subsource):
+                self.download_dir(subsource, subdestination)
+            else:
+                self.download_file(subsource, subdestination)
+
+    def write_blob(self, data: bytes, path: Path | str, overwrite: bool = False):
+        self._write_blob_data(path, data, overwrite=overwrite)
+
+    def upload_file(self, source: Path | str, path: str, overwrite: bool = False):
+        with open(source, "rb") as f:
+            self._write_blob_data(path, f, overwrite=overwrite)
+
+    def upload_dir(
+        self, source: Path | str, destination: Path | str, overwrite: bool = False
+    ):
+        # NOTE: Will not create empty folders
+        destination = Path(destination)
+        source = Path(source)
+
+        for subsource in source.iterdir():
+
+            basename = subsource.name
+
+            subdestination = str(destination / basename)
+
+            if subsource.is_file():
+                self.upload_file(subsource, subdestination, overwrite)
+            else:
+                self.upload_file(subsource, subdestination, overwrite)

azure_explorer-0.1.0/azure_explorer/managers/container/blob_storage.py

@@ -0,0 +1,64 @@
+from io import BytesIO
+from pathlib import Path
+from typing import Iterator
+
+from azure.core.credentials import TokenCredential
+from azure.storage.blob import ContainerClient, StorageStreamDownloader
+
+from . import ContainerManager
+
+
+class BlobStorageContainerManager(ContainerManager):
+    def __init__(
+        self,
+        storage_account_name: str,
+        container_name: str,
+        credential: TokenCredential,
+    ):
+        self.storage_account_name = storage_account_name
+        self.container_name = container_name
+        self.credential = credential
+        self.container_client = ContainerClient(
+            f"https://{storage_account_name}.blob.core.windows.net",
+            container_name=container_name,
+            credential=credential,
+        )
+
+    def is_dir(self, path: Path | str) -> bool:
+        if path == "":
+            return True
+
+        prefix = str(path).rstrip("/") + "/"
+        items = list(self.container_client.walk_blobs(prefix, delimiter="/"))
+
+        if len(items) == 0:
+            return False
+
+        if len(items) == 1 and items[0].name == path:
+            return False
+
+        return True
+
+    def is_file(self, path: Path | str) -> bool:
+        blob_client = self.container_client.get_blob_client(path)
+        return blob_client.exists()
+
+    def _iter_dir(self, path: Path | str = "") -> Iterator[str]:
+        prefix = str(path).rstrip("/") + "/"
+        for item in self.container_client.walk_blobs(prefix, delimiter="/"):
+            yield item.name.rstrip("/")
+
+    def _get_blob_stream_downloader(self, path: Path | str) -> StorageStreamDownloader:
+        return self.container_client.download_blob(path)
+
+    def _write_blob_data(
+        self, path: Path | str, data: bytes | BytesIO, overwrite: bool = False
+    ):
+        self.container_client.upload_blob(path, data, overwrite=overwrite)
+
+    def __repr__(self) -> str:
+        return (
+            "BlobStorageContainerManager("
+            f"storage_account_name='{self.storage_account_name}', "
+            f"container_name='{self.container_name}')"
+        )

azure_explorer-0.1.0/azure_explorer/managers/container/data_lake.py

@@ -0,0 +1,79 @@
+from io import BytesIO
+from pathlib import Path
+from typing import Iterator
+
+from azure.core.credentials import TokenCredential
+from azure.storage.filedatalake import FileSystemClient, StorageStreamDownloader
+
+from . import ContainerManager
+
+
+class DataLakeContainerManager(ContainerManager):
+    def __init__(
+        self,
+        storage_account_name: str,
+        container_name: str,
+        credential: TokenCredential,
+    ):
+        self.storage_account_name = storage_account_name
+        self.container_name = container_name
+        self.credential = credential
+        self.file_system_client = FileSystemClient(
+            f"https://{storage_account_name}.dfs.core.windows.net",
+            file_system_name=container_name,
+            credential=credential,
+        )
+
+    def is_dir(self, folder: Path | str) -> bool:
+        # checking if a folder exist is a bit tricky, but
+        # we use the solution proposed here in
+        # https://github.com/Azure/azure-sdk-for-python/issues/24814
+        if folder == "":
+            return True
+
+        file_client = self.file_system_client.get_file_client(folder)
+        if not file_client.exists():
+            return False
+
+        file_metadata = file_client.get_file_properties().metadata
+        is_dir_blob = file_metadata.get("hdi_isfolder", "false").lower() == "true"
+        if not is_dir_blob:
+            return False
+
+        return True
+
+    def is_file(self, path):
+        if path == "":
+            return False
+
+        file_client = self.file_system_client.get_file_client(path)
+        if not file_client.exists():
+            return False
+
+        file_metadata = file_client.get_file_properties().metadata
+        is_dir_blob = file_metadata.get("hdi_isfolder", "false").lower() == "true"
+        if is_dir_blob:
+            return False
+
+        return True
+
+    def _iter_dir(self, folder: Path | str = "") -> Iterator[str]:
+        for path in self.file_system_client.get_paths(folder, recursive=False):
+            yield path.name
+
+    def _get_blob_stream_downloader(self, path: Path | str) -> StorageStreamDownloader:
+        file_client = self.file_system_client.get_file_client(path)
+        return file_client.download_file()
+
+    def _write_blob_data(
+        self, path: Path | str, data: bytes | BytesIO, overwrite: bool = False
+    ):
+        file_client = self.file_system_client.get_file_client(path)
+        file_client.upload_data(data, overwrite=overwrite)
+
+    def __repr__(self) -> str:
+        return (
+            "DataLakeContainerManager("
+            f"storage_account_name='{self.storage_account_name}', "
+            f"container_name='{self.container_name}')"
+        )

azure_explorer-0.1.0/azure_explorer/managers/models.py

@@ -0,0 +1,14 @@
+from dataclasses import dataclass, field
+
+
+@dataclass
+class ResourceProperties:
+    name: str
+    id: str = field(repr=False)
+    type: str
+
+
+@dataclass
+class SubscriptionProperties:
+    id: str
+    name: str

azure_explorer-0.1.0/azure_explorer/managers/storage_account.py

@@ -0,0 +1,47 @@
+from azure.core.credentials import TokenCredential
+from azure.data.tables import TableServiceClient
+from azure.storage.blob import BlobServiceClient
+
+from . import ContainerManager, Manager
+
+
+class StorageAccountManager(Manager):
+    def __init__(self, storage_account_name: str, credential: TokenCredential):
+        self.storage_account_name = storage_account_name
+        self.credential = credential
+        self.blob_service_client = BlobServiceClient(
+            f"https://{storage_account_name}.blob.core.windows.net",
+            credential=credential,
+        )
+
+        self.is_data_lake = self.blob_service_client.get_account_information().get(
+            "is_hns_enabled", False
+        )
+
+        self.table_service_client = TableServiceClient(
+            f"https://{storage_account_name}.table.core.windows.net",
+            credential=credential,
+        )
+
+    def get_container_manager(self, container_name: str) -> ContainerManager:
+        return ContainerManager.create(
+            storage_account_name=self.storage_account_name,
+            container_name=container_name,
+            is_data_lake=self.is_data_lake,
+            credential=self.credential,
+        )
+
+    def list_container_names(self) -> list[str]:
+        return [
+            container.name for container in self.blob_service_client.list_containers()
+        ]
+
+    def list_table_names(self) -> list[str]:
+        return [table.name for table in self.table_service_client.list_tables()]
+
+    def __repr__(self) -> str:
+        return (
+            "StorageAccountManager("
+            f"storage_account_name='{self.storage_account_name}', "
+            f"is_data_lake={self.is_data_lake})"
+        )

azure_explorer-0.1.0/azure_explorer/managers/subscription.py

@@ -0,0 +1,76 @@
+from azure.core.credentials import TokenCredential
+from azure.mgmt.resource import ResourceManagementClient
+from azure.mgmt.servicebus import ServiceBusManagementClient
+from azure.mgmt.storage import StorageManagementClient
+
+from azure_explorer.managers.models import ResourceProperties
+
+from . import Manager, StorageAccountManager
+
+
+class SubscriptionManager(Manager):
+    def __init__(self, subscription_id: str, credential: TokenCredential):
+        self.subscription_id = subscription_id
+        self.resource_mgtm_client = ResourceManagementClient(
+            credential,
+            subscription_id,
+        )
+        self.storage_mgtm_client = StorageManagementClient(
+            credential,
+            subscription_id,
+        )
+        self.servicebus_mgtm_client = ServiceBusManagementClient(
+            credential,
+            subscription_id,
+        )
+        self.credential = credential
+
+    def list_storage_accounts(self) -> list[ResourceProperties]:
+        items = []
+        for storage_account in self.storage_mgtm_client.storage_accounts.list():
+            if storage_account.is_hns_enabled:
+                resource_type = "data_lake"
+            else:
+                resource_type = "storage_account"
+
+            item = ResourceProperties(
+                id=storage_account.id,
+                name=storage_account.name,
+                type=resource_type,
+            )
+            items.append(item)
+
+        return items
+
+    def list_storage_account_ids(self) -> list[str]:
+        return [item.id for item in self.list_storage_accounts()]
+
+    def list_storage_account_names(self) -> list[str]:
+        return [item.name for item in self.list_storage_accounts()]
+
+    def get_storage_account_manager(self, storage_account_name: str):
+        return StorageAccountManager(
+            storage_account_name=storage_account_name,
+            credential=self.credential,
+        )
+
+    def list_service_buses(self) -> list[ResourceProperties]:
+        items = []
+        for namespace in self.servicebus_mgtm_client.namespaces.list():
+            item = ResourceProperties(
+                id=namespace.id,
+                name=namespace.name,
+                type="service_bus",
+            )
+            items.append(item)
+
+        return items
+
+    def list_service_bus_ids(self) -> list[str]:
+        return [item.id for item in self.list_service_buses()]
+
+    def list_service_bus_names(self) -> list[str]:
+        return [item.name for item in self.list_service_buses()]
+
+    def __repr__(self) -> str:
+        return f"SubscriptionManager(subscription_id='{self.subscription_id}')"

azure_explorer-0.1.0/azure_explorer/managers/tenant.py

@@ -0,0 +1,35 @@
+from azure.core.credentials import TokenCredential
+from azure.mgmt.subscription import SubscriptionClient
+
+from . import Manager, SubscriptionManager, SubscriptionProperties
+
+
+class TenantManager(Manager):
+    def __init__(self, credential: TokenCredential):
+        self.sub_client = SubscriptionClient(credential)
+        self.credential = credential
+
+    def list_subscriptions(self) -> list[SubscriptionProperties]:
+        items = []
+        for sub in self.sub_client.subscriptions.list():
+            item = SubscriptionProperties(
+                id=sub.subscription_id,
+                name=sub.display_name,
+            )
+            items.append(item)
+        return items
+
+    def list_subscription_ids(self) -> list[str]:
+        return [item.id for item in self.list_subscriptions()]
+
+    def list_subscription_names(self) -> list[str]:
+        return [item.name for item in self.list_subscriptions()]
+
+    def get_subscription_manager(self, subscription_id: str):
+        return SubscriptionManager(
+            subscription_id=subscription_id,
+            credential=self.credential,
+        )
+
+    def __repr__(self) -> str:
+        return "TenantManager()"

azure_explorer-0.1.0/azure_explorer/short_cuts.py

@@ -0,0 +1,53 @@
+from azure_explorer.config import Config
+from azure_explorer.managers import (
+    ContainerManager,
+    StorageAccountManager,
+    SubscriptionManager,
+    TenantManager,
+)
+
+
+def get_tenant_manager() -> TenantManager:
+    return TenantManager(
+        credential=Config.credential,
+    )
+
+
+def get_subscription_manager(subscription_id: str) -> SubscriptionManager:
+    return SubscriptionManager(
+        subscription_id=subscription_id,
+        credential=Config.credential,
+    )
+
+
+def get_storage_account_manager(
+    storage_account_name: str,
+) -> StorageAccountManager:
+    return StorageAccountManager(
+        storage_account_name=storage_account_name,
+        credential=Config.credential,
+    )
+
+
+def get_container_manager(
+    storage_account_name: str,
+    container_name: str,
+    is_data_lake: bool = False,
+) -> ContainerManager:
+    return ContainerManager.create(
+        storage_account_name=storage_account_name,
+        container_name=container_name,
+        is_data_lake=is_data_lake,
+        credential=Config.credential,
+    )
+
+
+def read_file(
+    storage_account_name: str,
+    container_name: str,
+    path: str,
+) -> bytes:
+    container_manager = get_container_manager(
+        storage_account_name, container_name, is_data_lake=False
+    )
+    return container_manager.read_file(path)

azure_explorer-0.1.0/azure_explorer/widget/__init__.py

@@ -0,0 +1 @@
+from . import app

azure_explorer-0.1.0/azure_explorer/widget/app.py

@@ -0,0 +1,29 @@
+from textual.app import App, ComposeResult
+from textual.binding import Binding
+from textual.widgets import Footer, Header
+
+from ..config import Config
+from ..managers import TenantManager
+from .browsers import TenantBrowser
+
+
+class AzureExplorerApp(App):
+
+    BINDINGS = [
+        Binding("ctrl+x", "quit", "Exit", show=True, priority=True),
+    ]
+
+    TITLE = "Azure Explorer"
+
+    def compose(self) -> ComposeResult:
+        yield Header()
+        yield Footer()
+
+    def on_mount(self):
+        tenant_manager = TenantManager(credential=Config.credential)
+        tenant_browser = TenantBrowser(tenant_manager)
+        self.mount(tenant_browser)
+
+
+def run():
+    AzureExplorerApp().run()

azure_explorer-0.1.0/azure_explorer/widget/browsers/__init__.py

@@ -0,0 +1,5 @@
+from .base import Browser
+from .container import ContainerBrowser
+from .storage_account import StorageAccountTypeBrowser
+from .subscription import SubscriptionBrowser
+from .tenant import TenantBrowser

azure_explorer-0.1.0/azure_explorer/widget/browsers/base.py

@@ -0,0 +1,112 @@
+from abc import abstractmethod
+from typing import Any
+
+from textual.app import ComposeResult
+from textual.binding import Binding
+from textual.containers import VerticalScroll
+from textual.widget import Widget
+from textual.widgets import DataTable, Input, Tabs
+
+
+class Browser(Widget):
+
+    BINDINGS = [
+        Binding("up", "previous_row", "Previous row", show=False, priority=True),
+        Binding("down", "next_row", "Next row", show=False, priority=True),
+        Binding("escape", "go_back", "Go Back", show=True, priority=True),
+    ]
+
+    def __init__(self):
+        self.parent_browser = None
+        self.cached_rows = None
+        super().__init__()
+
+    def compose(self) -> ComposeResult:
+        input_ = Input(placeholder="Search...")
+        yield input_
+
+        data_table = DataTable(cursor_type="row")
+        data_table.can_focus = False
+        yield VerticalScroll(data_table)
+
+    def action_previous_row(self):
+        table = self.query_one(DataTable)
+        table.action_cursor_up()
+
+    def action_next_row(self):
+        table = self.query_one(DataTable)
+        table.action_cursor_down()
+
+    def action_go_back(self):
+        if self.parent_browser is not None:
+            self.remove()
+            self.app.mount(self.parent_browser)
+
+    def set_parent_browser(self, browser: "Browser"):
+        self.parent_browser = browser
+
+    def get_parent_browser(self) -> "Browser":
+        return getattr(self, "parent_browser", None)
+
+    def get_highlighted_item(self) -> Any:
+        table = self.query_one(DataTable)
+        if table.cursor_row is None:
+            return None
+        selected_id = table.coordinate_to_cell_key(
+            table.cursor_coordinate
+        ).row_key.value
+        return self.cached_rows[selected_id]
+
+    def _on_mount(self):
+        table = self.query_one(DataTable)
+        table.add_columns("Name")
+
+        self.populate_table()
+        self.query_one(Input).focus()
+
+    def on_data_table_row_selected(self, event: DataTable.RowSelected):
+        selected_id = event.row_key.value
+
+        selected_item = self.cached_rows[selected_id]
+
+        widget = self.get_item_widget(selected_item)
+        if widget is None:
+            return
+
+        widget.set_parent_browser(self)
+
+        self.remove()
+        self.app.mount(widget)
+
+    def on_input_changed(self, event: Input.Changed):
+        self.populate_table()
+
+    def on_tabs_tab_activated(self, event: Tabs.TabActivated):
+        self.populate_table()
+
+    def on_input_submitted(self):
+        self.query_one(DataTable).action_select_cursor()
+
+    def populate_table(self):
+        search_filter = self.query_one(Input).value.lower()
+
+        table = self.query_one(DataTable)
+        table.clear()
+
+        if self.cached_rows is None:
+            self.cached_rows = {
+                k: v for k, v in sorted(self.iter_rows(), key=lambda x: x[0].lower())
+            }
+
+        for id_, item in self.cached_rows.items():
+            if search_filter not in id_:
+                continue
+            table.add_row(id_, key=id_)
+
+    @abstractmethod
+    def iter_rows(self) -> list[tuple[str]]:
+        ...
+
+    @abstractmethod
+    def get_item_widget(self, id_: str) -> "Browser":
+        ...

azure_explorer-0.1.0/azure_explorer/widget/browsers/container.py

@@ -0,0 +1,74 @@
+from pathlib import Path
+from typing import Iterator
+
+from textual.binding import Binding
+from textual.widget import Widget
+
+from ...managers import ContainerManager
+from . import Browser
+
+
+class ContainerBrowser(Browser):
+
+    BINDINGS = [
+        Binding("ctrl+s", "download", "Download", show=True, priority=True),
+        Binding("ctrl+c", "copy_code", "Copy code", show=True, priority=True),
+    ]
+
+    def __init__(self, container_manager: ContainerManager, folder: str = ""):
+        self.container_manager = container_manager
+        self.folder = folder
+        self.blobs = None
+        super().__init__()
+
+    def check_action(self, action: str, parameters: tuple[object, ...]) -> bool | None:
+        """Check if an action may run."""
+        if action == "copy_code":
+            item = self.get_highlighted_item()
+            if self.container_manager.is_dir(item):
+                return None
+        return True
+
+    def action_download(self):
+        item = self.get_highlighted_item()
+
+        download_folder = Path.cwd() / ".downloads" / item
+
+        if self.container_manager.is_dir(item):
+            self.container_manager.download_dir(item, download_folder)
+        else:
+            self.container_manager.download_file(item, download_folder)
+
+    def action_copy_code(self):
+        item = self.get_highlighted_item()
+        code_snippet = f"""
+import azure_explorer as ax
+
+ax.read_file(
+    storage_account_name="{self.container_manager.storage_account_name}",
+    container_name="{self.container_manager.container_name}",
+    path="{item}",
+)
+"""
+        self.app.copy_to_clipboard(code_snippet)
+        self.notify("Code snippet copied to clipboard!", severity="info", timeout=3)
+
+    def iter_rows(self) -> Iterator[tuple[str, str]]:
+        for subpath in self.container_manager.list_dir(self.folder):
+            if self.container_manager.is_dir(subpath):
+                icon = "📁"
+            else:
+                icon = "📄"
+
+            base_name = subpath.split("/")[-1]
+
+            yield (f"{icon} {base_name}", subpath)
+
+    def get_item_widget(self, item: str) -> Widget:
+        if not self.container_manager.is_dir(item):
+            return
+
+        return ContainerBrowser(
+            self.container_manager,
+            item,
+        )

azure_explorer-0.1.0/azure_explorer/widget/browsers/storage_account.py

@@ -0,0 +1,50 @@
+from typing import Iterator
+
+from textual.widget import Widget
+
+from ...managers import StorageAccountManager
+from . import Browser, ContainerBrowser
+
+
+class StorageAccountTypeBrowser(Browser):
+    def __init__(self, sa_manager: StorageAccountManager):
+        self.sa_manager = sa_manager
+        super().__init__()
+
+    def iter_rows(self) -> Iterator[tuple[str, str]]:
+        yield ("Containers", "containers")
+        yield ("Tables", "tables")
+
+    def get_item_widget(self, item: str) -> Widget:
+        if item == "containers":
+            return StorageAccountContainerBrowser(self.sa_manager)
+        elif item == "tables":
+            return StorageAccountTableBrowser(self.sa_manager)
+        else:
+            return None
+
+
+class StorageAccountContainerBrowser(Browser):
+    def __init__(self, sa_manager: StorageAccountManager):
+        self.sa_manager = sa_manager
+        super().__init__()
+
+    def iter_rows(self) -> Iterator[tuple[str, str]]:
+        for container_name in self.sa_manager.list_container_names():
+            yield (container_name, container_name)
+
+    def get_item_widget(self, item: str) -> Widget:
+        container_manager = self.sa_manager.get_container_manager(
+            container_name=item,
+        )
+        return ContainerBrowser(container_manager)
+
+
+class StorageAccountTableBrowser(Browser):
+    def __init__(self, sa_manager: StorageAccountManager):
+        self.sa_manager = sa_manager
+        super().__init__()
+
+    def iter_rows(self) -> Iterator[tuple[str, str]]:
+        for table_name in self.sa_manager.list_table_names():
+            yield (table_name, table_name)

azure_explorer-0.1.0/azure_explorer/widget/browsers/subscription.py

@@ -0,0 +1,22 @@
+from typing import Iterator
+
+from textual.widget import Widget
+
+from ...managers import SubscriptionManager
+from . import Browser, StorageAccountTypeBrowser
+
+
+class SubscriptionBrowser(Browser):
+    def __init__(self, sub_manager: SubscriptionManager):
+        self.sub_manager = sub_manager
+        super().__init__()
+
+    def iter_rows(self) -> Iterator[tuple[str, str]]:
+        for storage_account in self.sub_manager.list_storage_accounts():
+            yield (storage_account.name, storage_account.name)
+
+    def get_item_widget(self, item: str) -> Widget:
+        sa_manager = self.sub_manager.get_storage_account_manager(
+            storage_account_name=item,
+        )
+        return StorageAccountTypeBrowser(sa_manager)

azure_explorer-0.1.0/azure_explorer/widget/browsers/tenant.py

@@ -0,0 +1,22 @@
+from typing import Iterator
+
+from textual.widget import Widget
+
+from ...managers import TenantManager
+from . import Browser, SubscriptionBrowser
+
+
+class TenantBrowser(Browser):
+    def __init__(self, tenant_manager: TenantManager):
+        self.tenant_manager = tenant_manager
+        super().__init__()
+
+    def iter_rows(self) -> Iterator[tuple[str, str]]:
+        for sub in self.tenant_manager.list_subscriptions():
+            yield (sub.name, sub.id)
+
+    def get_item_widget(self, item: str) -> Widget:
+        sub_manager = self.tenant_manager.get_subscription_manager(
+            subscription_id=item,
+        )
+        return SubscriptionBrowser(sub_manager)

azure_explorer-0.1.0/pyproject.toml

@@ -0,0 +1,45 @@
+[tool.poetry]
+name = "azure-explorer"
+version = "0.1.0"
+description = ""
+authors = ["uch <uch@energinet.dk>"]
+readme = "README.md"
+
+[tool.poetry.dependencies]
+python = "^3.10.8"
+azure-data-tables = "^12.7.0"
+azure-identity = "^1.25.1"
+azure-mgmt-resource = "^24.0.0"
+azure-mgmt-servicebus = "^9.0.0"
+azure-mgmt-storage = "^24.0.0"
+azure-mgmt-subscription = "^3.1.1"
+azure-servicebus = "^7.14.3"
+azure-storage-blob = "^12.27.1"
+azure-storage-file-datalake = "^12.22.0"
+textual = "^7.0.0"
+
+[tool.poetry.dev-dependencies]
+pytest = "^7"
+pytest-cov = "^3.0.0"
+pre-commit = "^2.20.0"
+pytest-asyncio = "^0.19.0"
+freezegun = "^1.2.2"
+mypy = "^1.19.0"
+
+[tool.poetry.scripts]
+ax = "azure_explorer.widget.app:run"
+
+[build-system]
+requires = ["poetry-core>=1.0.0"]
+build-backend = "poetry.core.masonry.api"
+
+[tool.isort]
+multi_line_output = 3
+line_length = 88
+include_trailing_comma = true
+
+[tool.black]
+line_length = 88
+
+[tool.mypy]
+ignore_missing_imports = true