database.py

from __future__ import annotations

from pathlib import Path

from derio.bootstrap import create_application


def setup_db(db_path: str | Path | None = None) -> None:
    create_application(db_path=db_path)


def add_transaction(
    type: str,
    category: str,
    amount: float,
    date: str,
    db_path: str | Path | None = None,
) -> int:
    app = create_application(db_path=db_path)
    transaction = app.transaction_service.add_transaction(
        transaction_type=type,
        category=category,
        amount=amount,
        transaction_date=date,
    )
    return int(transaction.id or 0)


def get_all_transactions(db_path: str | Path | None = None) -> list[tuple]:
    app = create_application(db_path=db_path)
    transactions = app.transaction_service.list_transactions()
    return [
        (
            transaction.id,
            transaction.transaction_type.value,
            transaction.category,
            transaction.amount,
            transaction.transaction_date.isoformat(),
        )
        for transaction in transactions
    ]


def get_summary(db_path: str | Path | None = None) -> list[tuple]:
    app = create_application(db_path=db_path)
    summary_items = app.transaction_service.get_summary()
    return [
        (item.transaction_type.value, item.total)
        for item in summary_items
    ]