add a way to allow to get budgets information

actual/database.py

@@ -98,7 +98,8 @@ def before_flush(sess, flush_context, instances):
     @event.listens_for(session, "after_commit")
     @event.listens_for(session, "after_soft_rollback")
     def after_commit_or_rollback(
-        sess, previous_transaction=None  # noqa: previous_transaction needed for soft rollback
+        sess,
+        previous_transaction=None,  # noqa: previous_transaction needed for soft rollback
     ):
         if sess.info.get("messages"):
             del sess.info["messages"]
@@ -236,6 +237,8 @@ class Categories(BaseModel, table=True):
     tombstone: Optional[int] = Field(default=None, sa_column=Column("tombstone", Integer, server_default=text("0")))
     goal_def: Optional[str] = Field(default=None, sa_column=Column("goal_def", Text, server_default=text("null")))
 
+    zero_budgets: "ZeroBudgets" = Relationship(back_populates="category_item")
+
     transactions: List["Transactions"] = Relationship(
         back_populates="category",
         sa_relationship_kwargs={
@@ -571,7 +574,14 @@ class ZeroBudgets(SQLModel, table=True):
 
     id: Optional[str] = Field(default=None, sa_column=Column("id", Text, primary_key=True))
     month: Optional[int] = Field(default=None, sa_column=Column("month", Integer))
-    category: Optional[str] = Field(default=None, sa_column=Column("category", Text))
+    category: Optional[str] = Field(default=None, sa_column=Column("category", ForeignKey("categories.id")))
+    category_item: "Categories" = Relationship(
+        back_populates="zero_budgets",
+        sa_relationship_kwargs={
+            "uselist": False,
+            "primaryjoin": "and_(ZeroBudgets.category == Categories.id,Categories.tombstone == 0)",
+        },
+    )
     amount: Optional[int] = Field(default=None, sa_column=Column("amount", Integer, server_default=text("0")))
     carryover: Optional[int] = Field(default=None, sa_column=Column("carryover", Integer, server_default=text("0")))
     goal: Optional[int] = Field(default=None, sa_column=Column("goal", Integer, server_default=text("null")))

actual/queries.py

@@ -545,6 +545,22 @@ def get_or_create_account(s: Session, name: str | Accounts) -> Accounts:
     return account
 
 
+def get_budgets(s: Session) -> typing.Sequence[ZeroBudgets]:
+    """
+    Returns a list of all available ZeroBudgets.
+    :param s: session from Actual local database.
+    :return: list of ZeroBudgets
+    """
+    query = select(ZeroBudgets).options(joinedload(ZeroBudgets.category_item))
+    return s.exec(query).unique().all()
+
+
+def get_budget(s: Session, category_name: str) -> typing.Optional[ZeroBudgets]:
+    """Gets an existing budget by category name, returns `None` if not found"""
+    query = select(ZeroBudgets).join(Categories).filter(Categories.name == category_name)
+    return s.exec(query).unique().all()
+
+
 def create_transfer(
     s: Session,
     date: datetime.date,