Luís Murta
37c97453a9
Bring report up-to-date with sqlite3 database and yaml configuration files. Updates report command in runnable.py. Moves date arg parsing to utils module Report and graph now share an ArgumentParser with shared date options.
77 lines
1.8 KiB
Python
77 lines
1.8 KiB
Python
from datetime import date, datetime, timedelta
|
|
from decimal import Decimal
|
|
from pathlib import Path
|
|
|
|
|
|
class WrongFilenameError(Exception):
|
|
pass
|
|
|
|
|
|
class BankNotAvailableError(Exception):
|
|
pass
|
|
|
|
|
|
class CreditCardNotAvailableError(Exception):
|
|
pass
|
|
|
|
|
|
def parse_decimal(s: str) -> Decimal:
|
|
try:
|
|
float(s)
|
|
return Decimal(s)
|
|
except ValueError:
|
|
pass
|
|
s = s.strip().replace(u"\xa0", "").replace(" ", "")
|
|
s = s.strip().replace("€", "").replace("+", "")
|
|
if s.rfind(",") > s.rfind("."):
|
|
s = s.replace(".", "")
|
|
i = s.rfind(",")
|
|
li = list(s)
|
|
li[i] = "."
|
|
s = "".join(li)
|
|
return Decimal(s.replace(",", ""))
|
|
|
|
|
|
def find_credit_institution(fn, banks, creditcards):
|
|
name = Path(fn).stem.split("_")
|
|
bank, cc = None, None
|
|
for i in name:
|
|
try:
|
|
int(i)
|
|
except ValueError:
|
|
if not bank:
|
|
bank = i
|
|
else:
|
|
cc = i
|
|
|
|
if not bank:
|
|
raise WrongFilenameError
|
|
|
|
if bank not in banks:
|
|
raise BankNotAvailableError
|
|
if cc and cc not in creditcards:
|
|
raise CreditCardNotAvailableError
|
|
|
|
return bank, cc
|
|
|
|
|
|
def parse_args_period(args):
|
|
start, end = date.min, date.max
|
|
if args.start or args.interval:
|
|
start = datetime.strptime(args.start[0], "%Y/%m/%d").date()
|
|
|
|
if args.end or args.interval:
|
|
end = datetime.strptime(args.end[0], "%Y/%m/%d").date()
|
|
|
|
if args.interval:
|
|
start = datetime.strptime(args.interval[0], "%Y/%m/%d").date()
|
|
end = datetime.strptime(args.interval[1], "%Y/%m/%d").date()
|
|
|
|
if args.year:
|
|
start = datetime.strptime(args.year[0], "%Y").date()
|
|
end = datetime.strptime(str(int(args.year[0]) + 1), "%Y").date() - timedelta(
|
|
days=1
|
|
)
|
|
|
|
return start, end
|