accountant.py

import sys

from manager import FileHandler, Manager

reader = FileHandler("in.txt")
manager = Manager(reader)


@manager.action("saldo", 2)
def saldo(manager, rows):
    price = float(rows[0])
    comment = rows[1]
    manager.modify_account(price)
    return True


@manager.action("zakup", 3)
def zakup(manager, rows):
    name = rows[0]
    price = float(rows[1])
    qty = float(rows[2])
    manager.modify_account(-price * qty)
    manager.modify_stock(name, qty)
    return True


@manager.action("sprzedaz", 3)
def sprzedaz(manager, rows):
    name = rows[0]
    price = float(rows[1])
    qty = float(rows[2])
    manager.modify_account(price * qty)
    manager.modify_stock(name, -qty)
    return True


@manager.action("przeglad", 2)
def przeglad(manager, rows):
    for row in manager.review(rows[0], rows[1]):
        print("\n".join(row))


@manager.action("magazyn", 3)
def magazyn(manager, rows):
    for item in rows:
        if item not in manager.stock:
            manager.stock[item] = 0.0
    print(manager.stock)