1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
from typing import List
from utils.read_file import entry
from utils.register import currencies, complete_prices
from utils.colored_output import *
class tree:
def __init__(self, name: str, value: currencies = None) -> None:
self.name = name
self.value = value
self.children = []
def compute_value(self):
if self.value is not None:
return
self.value = currencies()
for child in reversed(self.children):
child.compute_value()
for price in child.value:
self.value.add_money(price)
def add_child(self, name: str, value: currencies = None):
self.children.append(tree(name, value))
def add_value(self, price: str):
if self.value is None:
self.value = currencies()
self.value.add_money(price)
def get_child_with_name(self, name: str):
for child in self.children:
if name == child.name:
return child
return None
def __contains__(self, name: str) -> bool:
for child in self.children:
if name == child.name:
return True
return False
def __str__(self, level=0, ignore_level = False) -> str:
result = ''
if not ignore_level:
for price in self.value:
result += price_f(f'{price:>20}') + '\n'
result = result.rstrip()
result += ' ' + ' '*(2 * level)
if len(self.children) == 1:
result += account_f(self.name + ':')
for child in self.children:
result += child.__str__(level + 1, ignore_level = True)
else:
result += account_f(self.name) + '\n'
for child in self.children:
result += child.__str__(level + 1)
return result
class accounts_tree:
def __init__(self) -> None:
self.my_accounts = tree(name='root')
def add_transaction(self, name_acc: str, price: str):
search = self.my_accounts
for name in name_acc.split(':'):
if name not in search:
search.add_child(name)
search = search.get_child_with_name(name)
search.add_value(price)
def __str__(self) -> str:
self.my_accounts.compute_value()
result = ''
for child in self.my_accounts.children:
result += str(child)
result += '--------------------\n'
for price in self.my_accounts.value:
result += price_f(f'{price:>20}') + '\n'
return result.rstrip()
def print_balance(my_entries: List[entry]):
my_tree = accounts_tree()
for ent in my_entries:
complete_prices(ent)
for trans in ent.transactions:
my_tree.add_transaction(trans[0], str(trans[1]))
print(my_tree)
|