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
|
from typing import List
from utils.read_file import entry
from utils.register import complete_prices
import re
RE_NOT = r'not ((?:[^()\s]+)|(?:\(.+\)))'
RE_AND = r'((?:[^()\s]+)|(?:\(.+\))) and ((?:[^()\s]+)|(?:\(.+\)))'
RE_OR = r'((?:[^()\s]+)|(?:\(.+\))) or ((?:[^()\s]+)|(?:\(.+\)))'
def clean_pattern(pattern: str):
pattern = pattern.strip()
if pattern[0] == '(' and pattern[-1] == ')':
pattern = pattern[1:-1]
return pattern
def create_ors(string: str):
splitted = string.split()
result = ''
for left, right in zip(splitted[:-1], splitted[1:]):
if left not in ['not', 'and', 'or'] and right not in ['and', 'or']:
result += f'{left} or '
else:
result += f'{left} '
result += splitted[-1]
return result
def find_logic_patt(pattern: str):
# First search for the pattern "not {text}"
patt = re.findall(RE_NOT, pattern)
if patt:
return ('not', patt[0])
# If there wasn't a "not" pattern, try searching for "{text 1} or {text 2}"
patt = re.findall(RE_OR, pattern)
if patt:
return ('or', patt[0])
# Lastly, try searching for "{text 1} and {text 2}" pattern.
patt = re.findall(RE_AND, pattern)
if patt:
return ('and', patt[0])
# If there wasn't any pattern, return None.
return (None, None)
def re_ledger(pattern: str, account_name: str):
pattern = clean_pattern(pattern)
if ' ' not in pattern:
if pattern == '{False}':
return False
elif pattern == '{True}':
return True
elif re.search(pattern, account_name) is None:
return False
else:
return True
pattern = create_ors(pattern)
key, patt = find_logic_patt(pattern)
while key is not None:
if key == 'not':
meets_criteria = not re_ledger(patt, account_name)
pattern = pattern.replace(f'not {patt}', f'{{{meets_criteria}}}')
elif key == 'or':
meets_criteria = re_ledger(patt[0], account_name) or re_ledger(patt[1], account_name)
pattern = pattern.replace(f'{patt[0]} or {patt[1]}', f'{{{meets_criteria}}}')
elif key == 'and':
meets_criteria = re_ledger(patt[0], account_name) and re_ledger(patt[1], account_name)
pattern = pattern.replace(f'{patt[0]} and {patt[1]}', f'{{{meets_criteria}}}')
key, patt = find_logic_patt(pattern)
return re_ledger(pattern, account_name)
def filter_accounts(arguments: List[str], my_entries: List[entry]):
pattern = ' '.join(arguments)
result = []
for ent in my_entries:
complete_prices(ent)
for trans in ent.transactions:
if re_ledger(pattern, trans[0]):
result.append(entry(
date=ent.date.strftime('%Y-%m-%d'),
comment=ent.comment,
transactions=[[trans[0], str(trans[1])]]
))
return result
|
