Luís Murta
be67612f67
Categorizer will work directly on ORM classes, which will cleanup the code, since changes will automatically be persisted when change the objects. Adds wrapper session class inside the DbClient for the manager to use. The manager will have to have some DB session knowledge, which adds some unfortunate coupling. Removes some unnecessary relations between tables that were added by mistake. category CLI option now uses the manager.
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
from pfbudget.db.model import Transaction, TransactionCategory
|
|
|
|
from datetime import timedelta
|
|
|
|
|
|
class Categorizer:
|
|
options = {}
|
|
|
|
def __init__(self, args: dict):
|
|
self.options["null_days"] = 4
|
|
|
|
def categorize(self, transactions: list[Transaction]):
|
|
"""Overarching categorization tool
|
|
|
|
Receives a list of transactions (by ref) and updates their category
|
|
|
|
Args:
|
|
transactions (list[Transaction]): uncategorized transactions
|
|
"""
|
|
|
|
self._nullify(transactions)
|
|
|
|
def _nullify(self, transactions: list[Transaction]):
|
|
matching = []
|
|
for transaction in transactions:
|
|
for cancel in (
|
|
cancel
|
|
for cancel in transactions
|
|
if (
|
|
transaction.date - timedelta(days=self.options["null_days"])
|
|
<= cancel.date
|
|
<= transaction.date + timedelta(days=self.options["null_days"])
|
|
and transaction not in matching
|
|
and cancel not in matching
|
|
and cancel != transaction
|
|
and cancel.bank != transaction.bank
|
|
and cancel.amount == -transaction.amount
|
|
)
|
|
):
|
|
transaction.category = TransactionCategory(name="null")
|
|
cancel.category = TransactionCategory(name="null")
|
|
matching.extend([transaction, cancel])
|
|
break
|