import json
from pathlib import Path
class PlannerLoad:
base_dir: Path
account_values: list = None
account_info: list = None
plan_values: list = None
marketing_values: list = None
new_structure: list
structure: list
config = {
'department': [1, 2, 3, 10, 30, 40, 50, 55, 81, 82],
'translation': [10, 11, 12, 13, 14, 15, 16, 17, 18, 19,
20, 21, 22, 23, 3, 4, 5, 6, 7, 4, 0, 4,
4, 4, 1, 4, 2, 4, 8, 9]
}
def __init__(self, base_dir: str):
self.base_dir = Path(base_dir)
def new_file(self, year: str):
self.load_values(year)
return self.convert_file(self.new_structure)
def load_values(self, year: str):
with open(self.base_dir.joinpath(f'accounts_{year}.json'), 'r') as frh:
self.account_values = json.load(frh)['values']
with open(self.base_dir.joinpath(f'planning_{year}.json'), 'r') as frh:
self.plan_values = json.load(frh)['values']
with open(self.base_dir.joinpath(f'marketing_{year}.json'), 'r') as frh:
self.marketing_values = json.load(frh)
with open(self.base_dir.joinpath('gcstruct.json'), 'r') as frh:
gcstruct = json.load(frh)
self.account_info = dict([(a['Konto_Nr'], a) for a in gcstruct['accounts']])
self.new_structure = gcstruct['flat']['Struktur_FB']
def set_structure(self, structure: list):
self.structure = structure
def convert_file(self, structure_source):
self.structure = []
for s in structure_source:
accts = s['accounts']
s['accounts'] = []
s['planlevel'] = (len(accts) > 0)
self.structure.append(s)
for a in accts:
if type(a) == str:
acct = {
'text': a + ' - ' + self.account_info[a]['Konto_Bezeichnung'],
'id': a,
'costcenter': self.account_info[a]['Konto_KST'],
'values': [],
'values2': None
}
else:
acct = a
a = acct['id']
acct['id'] = s['id'].replace(';;', ';' + acct['text'] + ';', 1)
s['children'].append(acct['id'])
new_account = {
"id": acct['id'],
"text": acct['text'],
"children": [],
"children2": [],
"parents": [s['id']] + s['parents'],
"accounts": [a],
"costcenter": acct['costcenter'],
"level": s['level'] + 1,
"drilldown": False,
"form": s['form'],
"accountlevel": False,
"absolute": True,
"seasonal": True,
"status": "0",
"values": acct['values'],
"values2": acct['values2']
}
self.structure.append(new_account)
for s in self.structure:
s['values2'] = self.get_values2(s)
s['options'] = self.get_options(s)
del s['accountlevel']
del s['absolute']
del s['seasonal']
del s['status']
s['sumvalues'] = [0] * 30
return self.structure
def get_values2(self, s):
if 'values2' not in s or s['values2'] is None or len(s['values2'].keys()) == 0:
s['values2'] = dict([(str(d), [0] * 30) for d in self.config['department']])
else:
for d in s['values2'].keys():
if len(s['values2'][d]) < 30:
s['values2'][d] = [s['values2'][d][i] if 0 <= i < len(s['values2'][d]) else 0
for i in self.config['translation']]
self.update_account_values(s)
self.update_plan_values(s)
self.update_marketing_values(s)
def update_values(self, value_type):
v_types = {
'accounts': self.update_account_values,
'plan': self.update_plan_values,
'marketing': self.update_marketing_values
}
parents = []
for s in self.structure:
if v_types[value_type](s):
parents.append(s['parents'][0])
if value_type == 'marketing':
for p_id in set(parents):
parent = self.get_structure_by_id(p_id)
parent['form'] = '9'
for d in parent['options'].keys():
parent['options'][d]['status'] = '0'
for d in parent['values2'].keys():
for i in range(15):
parent['values2'][d][i] = 0.0
return self.structure
def get_structure_by_id(self, id):
for s in self.structure:
if s['id'] == id:
return s
return None
def update_account_values(self, s):
if len(s['accounts']) == 0:
return False
a_values = self.account_values.get(s['accounts'][0], dict())
for d in s['values2'].keys():
if d in a_values:
for i, v in enumerate(a_values[d], 20):
s['values2'][d][i] = v
return s['values2']
def update_plan_values(self, s):
if self.plan_values is None:
return False
p_values = self.plan_values.get(s['id'], dict())
for d in s['values2'].keys():
if d in p_values:
for i, v in enumerate(p_values[d], 28):
s['values2'][d][i] = v
return True
def update_marketing_values(self, s):
if len(s['accounts']) == 0:
return False
m_values = self.marketing_values.get(s['accounts'][0], dict())
if not m_values:
return False
for d in s['values2'].keys():
if d in m_values:
for i, v in enumerate(m_values[d], 1):
s['values2'][d][i] = v
s['values2'][d][0] = sum(m_values[d])
s['values2'][d][14] = sum(m_values[d])
if 'options' in s:
s['options'][d]['seasonal'] = False
return True
def convert_values2(self, s):
a_values = {}
if s['accounts']:
a_values = self.account_values.get(s['accounts'][0], dict())
p_values = self.plan_values.get(s['id'], dict())
values = dict([(str(d), [0] * 30) for d in self.config['department']])
for d in values.keys():
if d in s['values2']:
if len(s['values2'][d]) == 30:
values[d] = s['values2'][d]
else:
values[d] = [s['values2'][d].get(i, 0) for i in self.config['translation']]
if d in p_values:
for i, v in enumerate(p_values[d], 28):
values[d][i] = v
if d in a_values:
for i, v in enumerate(a_values[d], 20):
values[d][i] = v
return values
def get_options(self, s):
status = "0"
seasonal = True
absolute = True
planlevel = False
if isinstance(s['absolute'], bool):
absolute = s['absolute']
s['absolute'] = {}
if isinstance(s['seasonal'], bool):
seasonal = s['seasonal']
s['seasonal'] = {}
if isinstance(s['status'], str):
status = s['status']
s['status'] = {}
if 'planlevel' in s:
planlevel = s['planlevel']
opts = dict([(str(d), {
'absolute': absolute,
'seasonal': seasonal,
'status': status,
'planlevel': planlevel
}) for d in self.config['department']])
for d in opts.keys():
if d in s['absolute']:
opts[d]['absolute'] = s['absolute'][d]
if d in s['seasonal']:
opts[d]['seasonal'] = s['seasonal'][d]
if d in s['status']:
opts[d]['status'] = s['status'][d]
return opts
if __name__ == '__main__':
planner_dir = Path(__file__).parent.parent.joinpath('export')
p_load = PlannerLoad(planner_dir)
print(p_load.convert_file(
json.load(open(planner_dir.joinpath('2022_V1.json'), 'r'))
)[12])