beancount-gocardless 0.1.10__tar.gz → 0.1.11__tar.gz

{beancount_gocardless-0.1.10 → beancount_gocardless-0.1.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beancount-gocardless
-Version: 0.1.10
+Version: 0.1.11
 License-Expression: Unlicense
 License-File: LICENSE
 Requires-Python: <4,>=3.12

{beancount_gocardless-0.1.10 → beancount_gocardless-0.1.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "beancount-gocardless"
-version = "0.1.10"
+version = "0.1.11"
 description = ""
 authors = []
 readme = "README.md"

{beancount_gocardless-0.1.10 → beancount_gocardless-0.1.11}/src/beancount_gocardless/cli.py

@@ -4,6 +4,7 @@ import os
 import logging
 
 from beancount_gocardless.models import AccountInfo
+from beancount_gocardless.client import GoCardlessClient
 
 
 logging.basicConfig(level=os.environ.get("LOGLEVEL", logging.INFO))
@@ -83,7 +84,52 @@ def main():
     try:
         logger.debug("Initializing GoCardlessClient")
 
-        # TODO: Implement client initialization with cache options
+        cache_options = (
+            {
+                "backend": args.cache_backend,
+                "expire_after": args.cache_expire,
+                "cache_name": args.cache_name,
+            }
+            if args.cache
+            else {}
+        )
+
+        client = GoCardlessClient(args.secret_id, args.secret_key, cache_options)
+
+        if args.mode == "list_banks":
+            banks = client.list_banks(args.country)
+            for bank in banks:
+                logger.info(bank)
+        elif args.mode == "create_link":
+            if not args.bank:
+                logger.error("Error: --bank is required for create_link")
+                sys.exit(1)
+            link = client.create_bank_link(args.reference, args.bank)
+            if link:
+                logger.info(f"Bank link created: {link}")
+            else:
+                logger.info(f"Link already exists for reference '{args.reference}'")
+        elif args.mode == "list_accounts":
+            accounts = client.list_accounts()
+            for i, account in enumerate(accounts, 1):
+                display_account(i, account)
+        elif args.mode == "delete_link":
+            req = client.find_requisition_by_reference(args.reference)
+            if req:
+                client.delete_requisition(req.id)
+                logger.info(f"Deleted requisition '{args.reference}'")
+            else:
+                logger.error(f"No requisition found with reference '{args.reference}'")
+                sys.exit(1)
+        elif args.mode == "balance":
+            if not args.account:
+                logger.error("Error: --account is required for balance")
+                sys.exit(1)
+            balances = client.get_account_balances(args.account)
+            for balance in balances.balances:
+                logger.info(
+                    f"{balance.balance_type}: {balance.balance_amount.amount} {balance.balance_amount.currency}"
+                )
 
     except Exception as e:
         logger.error(f"Error: {e}")

{beancount_gocardless-0.1.10 → beancount_gocardless-0.1.11}/src/beancount_gocardless/client.py

@@ -239,9 +239,9 @@ class GoCardlessClient:
         """Get institutions for a country"""
         logger.debug("Getting institutions for country %s", country)
         params = {"country": country} if country else {}
-        data = self.get("/institutions/", params=params)
-        logger.debug("Fetched %d institutions", len(data))
-        return [Institution(**inst) for inst in data]
+        institutions_data = self.get("/institutions/", params=params)
+        logger.debug("Fetched %d institutions", len(institutions_data))
+        return [Institution(**inst) for inst in institutions_data]
 
     def get_institution(self, institution_id: str) -> Institution:
         """Get specific institution"""
@@ -422,3 +422,7 @@ class GoCardlessClient:
                     # Skip accounts that can't be accessed
                     continue
         return accounts
+
+    def list_accounts(self) -> List[AccountInfo]:
+        """Alias for get_all_accounts"""
+        return self.get_all_accounts()

{beancount_gocardless-0.1.10 → beancount_gocardless-0.1.11}/src/beancount_gocardless/models.py

@@ -351,7 +351,7 @@ class Requisition(BaseModel):
     id: str
     created: str
     redirect: str
-    status: StatusEnum
+    status: str
     institution_id: str
     agreement: Optional[str] = None
     reference: str
@@ -376,7 +376,7 @@ class SpectacularRequisition(BaseModel):
     id: str
     created: str
     redirect: str
-    status: StatusEnum
+    status: str
     institution_id: str
     agreement: Optional[str] = None
     reference: str

{beancount_gocardless-0.1.10 → beancount_gocardless-0.1.11}/src/beancount_gocardless/tui.py

@@ -113,7 +113,7 @@ class BanksView(BaseSubView):
                 ]
 
             self.banks_table.update(
-                f"Loading accounts... {name_filter} {country_code} {len(self.all_banks)}"
+                f"Loading banks... {name_filter} {country_code} {len(self.all_banks)}"
             )
 
             name_filtered = [
@@ -180,6 +180,8 @@ class AccountsView(BaseSubView):
             "Loading accounts...", id="accounts_status_message"
         )
         self.accounts_table = Static("", id="accounts_table")
+        self.pending_delete_ref = None
+        self.accounts_list = []
 
     def compose_content(self) -> ComposeResult:
         yield Label("Linked Accounts", classes="view_title")
@@ -189,6 +191,23 @@ class AccountsView(BaseSubView):
             id="accounts_scrollable_area",
             classes="horizontal-scroll",
         )
+        yield Vertical(
+            Label("Select Account by # and Delete Link"),
+            Input(placeholder="Enter account #", id="select_account_input"),
+            Button("Select", id="select_account_button"),
+            Input(placeholder="Reference", id="delete_ref_input", disabled=True),
+            Button(
+                "Delete Link", id="delete_link_button", variant="error", disabled=True
+            ),
+            Button(
+                "Confirm Delete",
+                id="confirm_delete_button",
+                variant="error",
+                disabled=True,
+            ),
+            Button("Cancel", id="cancel_delete_button", disabled=True),
+            id="delete_section",
+        )
         yield Button("Back to Menu", id="back_to_menu", classes="back_button")
 
     # Override the base class compose to avoid duplicate back button
@@ -206,6 +225,7 @@ class AccountsView(BaseSubView):
         try:
             self.status_message.update("Loading accounts...")
             accounts = self.client.get_all_accounts()
+            self.accounts_list = accounts
             if not accounts:
                 self.accounts_table.update("No accounts found.")
                 self.status_message.update("")
@@ -215,7 +235,7 @@ class AccountsView(BaseSubView):
             table = Table(title=None, expand=True, min_width=100)
             table.add_column("#", style="dim", width=3)
             table.add_column("Bank", overflow="ellipsis", min_width=20)
-            table.add_column("Name", overflow="crop", min_width=20)
+            table.add_column("Name", overflow="ellipsis", min_width=20)
             table.add_column("ID", overflow="ellipsis", min_width=20)
             table.add_column("IBAN", overflow="ellipsis", min_width=20)
             table.add_column("Reference", overflow="ellipsis", min_width=15)
@@ -224,14 +244,12 @@ class AccountsView(BaseSubView):
 
             # Format and add accounts to the table
             for idx, account in enumerate(accounts, start=1):
-                # import pdb
-                # pdb.set_trace()
                 # Get values with fallbacks for any missing keys
                 institution_id = account.get("institution_id", "N/A")
                 name = account.get("name", account.get("owner_name", "N/A"))
                 iban = account.get("iban", "N/A")
-                account_id = account.get("account_id", "N/A")
-                reference = account.get("reference", "N/A")
+                account_id = account.get("id", "N/A")
+                reference = account.get("requisition_reference", "N/A")
                 status = account.get("status", "N/A")
                 last_accessed = account.get("last_accessed", "")
                 if last_accessed:
@@ -253,6 +271,39 @@ class AccountsView(BaseSubView):
 
             # Update the table content
             self.accounts_table.update(table)
+            if not accounts:
+                self.accounts_table.add_row(
+                    "No accounts found.", "", "", "", "", "", "", ""
+                )
+                self.status_message.update("")
+                return
+
+            # Format and add accounts to the table
+            for idx, account in enumerate(accounts, start=1):
+                # Get values with fallbacks for any missing keys
+                institution_id = account.get("institution_id", "N/A")
+                name = account.get("name", account.get("owner_name", "N/A"))
+                iban = account.get("iban", "N/A")
+                account_id = account.get("id", "N/A")
+                reference = account.get("requisition_reference", "N/A")
+                status = account.get("status", "N/A")
+                last_accessed = account.get("last_accessed", "")
+                if last_accessed:
+                    try:
+                        last_accessed = last_accessed.split("T")[0]
+                    except Exception:
+                        pass
+
+                self.accounts_table.add_row(
+                    str(idx),
+                    institution_id,
+                    name,
+                    account_id,
+                    iban,
+                    reference,
+                    status,
+                    last_accessed,
+                )
             self.status_message.update("")
 
         except HttpServiceException as e:
@@ -265,245 +316,74 @@ class AccountsView(BaseSubView):
         if event.button.id == "back_to_menu":
             self.post_message(ActionMessage("show_menu"))
 
-
-# ... existing code ...
-
-
-class BalanceView(BaseSubView):
-    """View to get and display account balance."""
-
-    def __init__(
-        self, client: "GoCardlessClient"
-    ):  # Forward reference GoCardlessClient if not imported yet
-        super().__init__()
-        self.client = client
-        self.accounts_data = []
-
-    def compose(self) -> ComposeResult:
-        """Override compose to yield widgets from compose_content."""
-        for widget in self.compose_content():
-            yield widget
-
-    def compose_content(self) -> ComposeResult:
-        yield Label("Account Balance", classes="view_title")
-        # Moved status message to be a direct child of BalanceView, and sibling to content_area
-        yield Static(
-            "", id="balance_status_message"
-        )  # Initialize empty or with a generic placeholder
-        yield Vertical(
-            # The initial Static message that was here is removed
-            id="balance_content_area",
-        )
-        yield Button("Back to Menu", id="back_to_menu", classes="back_button")
-
-    async def on_mount(self) -> None:
-        # Query elements only after they are mounted
-        content_area = self.query_one("#balance_content_area", Vertical)
-        status_message = self.query_one("#balance_status_message", Static)
-        status_message.update(
-            "Loading accounts for selection..."
-        )  # Set initial operational message
-        await self.load_accounts_for_selection(content_area, status_message)
-
-    async def load_accounts_for_selection(
-        self, content_area: Vertical, status_message: Static
-    ):
-        try:
-            # status_message.update("Loading accounts...") # Message set by on_mount or can be refined here
-            self.accounts_data = self.client.get_all_accounts()
-            await (
-                content_area.remove_children()
-            )  # Clear "Loading..." or previous content from content_area
-
-            if not self.accounts_data:
-                await content_area.mount(
-                    Static("No accounts linked. Cannot fetch balance.")
-                )
-                status_message.update(
-                    "No accounts available to display."
-                )  # Update persistent status
-                return
-
-            table = Table(title="Select Account by Index for Balance", expand=True)
-            table.add_column("#", style="dim")
-            table.add_column("Name", overflow="fold")
-            table.add_column("Institution ID", overflow="fold")
-            for idx, a in enumerate(self.accounts_data):
-                table.add_row(str(idx), a["name"], a["institution_id"])
-
-            await content_area.mount(table)  # Mount Table widget directly
-            idx_input = Input(
-                placeholder="Enter account # and press Enter", id="acct_idx_input"
-            )
-            await content_area.mount(idx_input)
-            idx_input.focus()
-            status_message.update(
-                "Please select an account from the list."
-            )  # Clear or update status message
-        except HttpServiceException as e:  # Make sure HttpServiceException is defined
-            status_message.update(
-                Text(f"Error loading accounts: {e}", style="bold red")
-            )
-            await content_area.remove_children()
-            await content_area.mount(
-                Static(
-                    Text(
-                        "Failed to load accounts. Check connection or try again later.",
-                        style="yellow",
-                    )
-                )
-            )
-        except Exception as e:
-            status_message.update(Text(f"Unexpected error: {e}", style="bold red"))
-            await content_area.remove_children()
-            await content_area.mount(
-                Static(
-                    Text(
-                        "An unexpected error occurred while loading accounts.",
-                        style="yellow",
-                    )
-                )
-            )
-
-    async def on_input_submitted(self, event: Input.Submitted) -> None:
-        if event.input.id == "acct_idx_input":
-            content_area = self.query_one("#balance_content_area", Vertical)
-            # status_message is now persistent and queried once
-            status_message = self.query_one("#balance_status_message", Static)
-
-            await (
-                content_area.remove_children()
-            )  # Clear account list and input from content_area
-
+    async def on_button_pressed(self, event: Button.Pressed) -> None:
+        """Handle button presses within this view."""
+        if event.button.id == "back_to_menu":
+            self.post_message(ActionMessage("show_menu"))
+        elif event.button.id == "select_account_button":
+            select_input = self.query_one("#select_account_input", Input)
             try:
-                idx = int(event.value)
-                if not (0 <= idx < len(self.accounts_data)):
-                    # Display temporary error message in content_area
-                    await content_area.mount(
-                        Static(Text("Invalid account index.", style="bold red"))
+                idx = int(select_input.value.strip()) - 1  # 1-based to 0-based
+                if 0 <= idx < len(self.accounts_list):
+                    reference = self.accounts_list[idx].get("requisition_reference", "")
+                    ref_input = self.query_one("#delete_ref_input", Input)
+                    ref_input.value = reference
+                    delete_btn = self.query_one("#delete_link_button", Button)
+                    delete_btn.disabled = False
+                    self.status_message.update(
+                        f"Selected account {idx + 1}, reference: {reference}"
                     )
-                    status_message.update(
-                        Text(
-                            "Selection out of range. Please try again.", style="yellow"
-                        )
-                    )
-                    # Reload account selection UI
-                    await self.load_accounts_for_selection(content_area, status_message)
-                    return
-
-                account_to_check = self.accounts_data[idx]
-                acct_id = account_to_check["id"]
-
-                status_message.update(
-                    f"Fetching balances for {account_to_check['name']} ({acct_id})..."
-                )
-                # Optionally, display a "Fetching..." message in content_area as well
-                await content_area.mount(Static("Fetching balances..."))
-
-                balances_data = self.client.get_account_balances(acct_id)
-                await content_area.remove_children()  # Clear "Fetching..." message
-
-                table = Table(
-                    title=f"Balances for {account_to_check['name']}", expand=True
-                )
-                table.add_column("Type", overflow="fold")
-                table.add_column("Amount")
-                table.add_column("Currency")
-                if balances_data.balances:
-                    for b in balances_data.balances:
-                        table.add_row(
-                            b.balance_type,
-                            b.balance_amount.amount,
-                            b.balance_amount.currency,
-                        )
                 else:
-                    table.add_row("N/A", "No balance data found", "N/A")
-
-                await content_area.mount(table)  # Mount the results table
-                status_message.update(
-                    f"Balances for {account_to_check['name']} displayed."
-                )
-                # Consider how the user navigates back to account selection (e.g., a button, or re-selecting the view)
-
+                    self.status_message.update("Invalid account number.")
             except ValueError:
-                # Display temporary error message in content_area
-                await content_area.mount(
-                    Static(
-                        Text("Invalid input. Please enter a number.", style="bold red")
-                    )
-                )
-                status_message.update(
-                    Text("Numeric input required for account index.", style="yellow")
-                )
-                await self.load_accounts_for_selection(
-                    content_area, status_message
-                )  # Reload selection
-            except (
-                HttpServiceException
-            ) as e:  # Make sure HttpServiceException is defined
-                status_message.update(
-                    Text(f"API Error fetching balances: {e}", style="bold red")
-                )
-                await content_area.mount(
-                    Static(
-                        Text(
-                            "Could not retrieve balances due to an API error.",
-                            style="red",
-                        )
+                self.status_message.update("Please enter a valid number.")
+        elif event.button.id == "delete_link_button":
+            ref_input = self.query_one("#delete_ref_input", Input)
+            reference = ref_input.value.strip()
+            if reference:
+                self.pending_delete_ref = reference
+                self.status_message.update(
+                    Text(
+                        f"Are you sure to delete link with reference '{reference}'?",
+                        style="bold yellow",
                     )
                 )
-                # Optionally, reload selection UI so user can try again or select a different account
-                # await self.load_accounts_for_selection(content_area, status_message)
-            except Exception as e:
-                status_message.update(
-                    Text(f"Error processing balance request: {e}", style="bold red")
-                )
-                await content_area.mount(
-                    Static(
-                        Text(
-                            "An unexpected error occurred while fetching balances.",
-                            style="red",
-                        )
-                    )
+                confirm_btn = self.query_one("#confirm_delete_button", Button)
+                cancel_btn = self.query_one("#cancel_delete_button", Button)
+                confirm_btn.disabled = False
+                cancel_btn.disabled = False
+            else:
+                self.status_message.update(
+                    Text("No reference selected.", style="bold yellow")
                 )
-                # Optionally, reload selection UI
-                # await self.load_accounts_for_selection(content_area, status_message)
-
-
-class LinkView(BaseSubView):
-    """View to create a new bank link."""
-
-    def __init__(self, client: GoCardlessClient):
-        super().__init__()
-        self.client = client
-
-    def compose(self) -> ComposeResult:
-        """Override compose to yield widgets from compose_content."""
-        for widget in self.compose_content():
-            yield widget
-
-    def compose_content(self) -> ComposeResult:
-        yield Label("Create New Bank Link", classes="view_title")
-        yield Vertical(
-            Input(
-                placeholder="Bank ID (e.g., SANDBOXFINANCE_SFIN0000)",
-                id="bank_id_input",
-            ),
-            Input(
-                placeholder="Unique Reference (e.g., mypc-savings-ref)", id="ref_input"
-            ),
-            Button("Create Link", id="do_create_link_button", variant="success"),
-            Static(id="link_result_message", classes="result_message_area"),
-            id="link_form_area",
-        )
-        yield Button("Back to Menu", id="back_to_menu", classes="back_button")
-
-    async def on_mount(self) -> None:
-        self.query_one("#bank_id_input", Input).focus()
-
-    async def on_button_pressed(self, event: Button.Pressed) -> None:
-        if event.button.id == "back_to_menu":
-            self.post_message(ActionMessage("show_menu"))
+        elif event.button.id == "confirm_delete_button":
+            if self.pending_delete_ref:
+                await self.process_delete_link(self.pending_delete_ref)
+                select_input = self.query_one("#select_account_input", Input)
+                ref_input = self.query_one("#delete_ref_input", Input)
+                select_input.value = ""
+                ref_input.value = ""
+                delete_btn = self.query_one("#delete_link_button", Button)
+                delete_btn.disabled = True
+                self.pending_delete_ref = None
+                confirm_btn = self.query_one("#confirm_delete_button", Button)
+                cancel_btn = self.query_one("#cancel_delete_button", Button)
+                confirm_btn.disabled = True
+                cancel_btn.disabled = True
+        elif event.button.id == "cancel_delete_button":
+            self.pending_delete_ref = None
+            self.status_message.update("Delete cancelled.")
+            confirm_btn = self.query_one("#confirm_delete_button", Button)
+            cancel_btn = self.query_one("#cancel_delete_button", Button)
+            confirm_btn.disabled = True
+            cancel_btn.disabled = True
+        elif event.button.id == "cancel_delete_button":
+            self.pending_delete_ref = None
+            self.status_message.update("Delete cancelled.")
+            confirm_btn = self.query_one("#confirm_delete_button", Button)
+            cancel_btn = self.query_one("#cancel_delete_button", Button)
+            confirm_btn.disabled = True
+            cancel_btn.disabled = True
         elif event.button.id == "do_create_link_button":
             await self.process_create_link()