pfbudget/cli/interactive.py

@@ -36,23 +36,23 @@ class Interactive:
             uncategorized = session.select(
                 Transaction, lambda: ~Transaction.category.has()
             )
-            list(uncategorized).sort()
+            uncategorized.sort()
             n = len(uncategorized)
             print(f"{n} left to categorize")
 
             i = 0
             new = []
 
-            while i < len(uncategorized):
+            while (command := input("$ ")) != "quit" and i < len(uncategorized):
                 current = uncategorized[i] if len(new) == 0 else new.pop()
                 print(current)
-                command = input("$ ")
 
                 match command:
                     case "help":
                         print(self.help)
 
                     case "skip":
+                        if len(uncategorized) == 0:
                             i += 1
 
                     case "quit":
@@ -89,7 +89,7 @@ class Interactive:
                             for tag in tags:
                                 if tag not in [t.name for t in self.tags]:
                                     session.insert([Tag(tag)])
-                                    self.tags = session.select(Tag)
+                                    self.tags = session.get(Tag)
 
                                 current.tags.add(TransactionTag(tag))
 
@@ -98,7 +98,7 @@ class Interactive:
 
     def split(self, original: Transaction) -> list[SplitTransaction]:
         total = original.amount
-        new: list[SplitTransaction] = []
+        new = []
 
         done = False
         while not done:
@@ -114,7 +114,7 @@ class Interactive:
             amount = decimal.Decimal(input("amount: "))
             new.append(
                 SplitTransaction(
-                    original.date, original.description, amount, original=original.id
+                    original.date, original.description, amount, original.id
                 )
             )
 

pfbudget/extract/nordigen.py

@@ -1,7 +1,6 @@
 from dataclasses import dataclass
 import datetime as dt
 import dotenv
-import json
 import nordigen
 import os
 import requests
@@ -61,33 +60,11 @@ class NordigenClient:
                     retries += 1
                     print(f"Request #{retries} timed-out, retrying in 1s")
                     time.sleep(1)
-                except requests.HTTPError as e:
-                    if (
-                        e.response.status_code == 409
-                        and e.response.type == "AccountProcessing"
-                    ):
-                        timeout = 1
-                        while account.get_metadata()["status"] != "READY":
-                            print(f"Waiting for status ready, retrying in {timeout}s")
-                            time.sleep(timeout)
-                            timeout *= 2
-                    elif e.response.status_code == 429:
-                        print("Rate limit exceeded for today, aborting")
-                        break
-                    else:
-                        raise DownloadError(e)
 
             if not downloaded:
                 print(f"Couldn't download transactions for {account.get_metadata()}")
                 continue
 
-            with open(
-                f"logs/{dt.datetime.now().isoformat()}_{requisition_id}.json",
-                "w",
-                encoding="utf-8",
-            ) as f:
-                json.dump(downloaded, f, ensure_ascii=False, indent=4)
-
             if (
                 "transactions" not in downloaded
                 or "booked" not in downloaded["transactions"]

pyproject.toml

@@ -27,7 +27,7 @@ pytest = "^7.3.0"
 pytest-cov = "^4.0.0"
 pytest-mock = "^3.10.0"
 sqlalchemy = {extras = ["mypy"], version = "^2.0.9"}
-ruff = "^0.11.2"
+ruff = "^0.0.267"
 
 [build-system]
 requires = ["poetry-core"]