49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
from pfbudget.input.input import Input
|
|
from pfbudget.input.parsers import parse_data
|
|
from pfbudget.common.types import Bank, Transaction
|
|
from pfbudget.db.client import DatabaseClient
|
|
from pfbudget.utils import convert
|
|
|
|
|
|
class Manager:
|
|
def __init__(self, db: str):
|
|
self.db = db
|
|
|
|
def init(self):
|
|
client = DatabaseClient(self.db)
|
|
client.init()
|
|
|
|
def register(self, args: dict):
|
|
print(args)
|
|
client = DatabaseClient(self.db)
|
|
client.register_bank(
|
|
Bank(
|
|
(
|
|
args["bank"][0],
|
|
args["requisition"][0]
|
|
if args["requisition"]
|
|
else args["requisition"],
|
|
args["invert"],
|
|
args["description"],
|
|
)
|
|
)
|
|
)
|
|
|
|
def unregister(self, args: dict):
|
|
client = DatabaseClient(self.db)
|
|
client.unregister_bank(args["bank"][0])
|
|
|
|
def parser(self, parser: Input):
|
|
print(parser.parse())
|
|
|
|
def parse(self, filename: str, args: dict):
|
|
transactions = parse_data(filename, args)
|
|
self.add_transactions(transactions)
|
|
|
|
def transactions() -> list[Transaction]:
|
|
pass
|
|
|
|
def add_transactions(self, transactions: list[Transaction]):
|
|
converted = convert(transactions)
|
|
self.__db.insert_transactions(converted)
|