modification rendu diagramme

This commit is contained in:
pb 2025-06-21 21:01:03 +02:00
parent ac65921498
commit 86b7f2437b
1 changed files with 16 additions and 11 deletions

27
main.py
View File

@ -62,7 +62,11 @@ def create_accounting_sankey(cursor, save_dir):
year = datetime.datetime.now().year year = datetime.datetime.now().year
first_day = rf'{year}-01-01' first_day = rf'{year}-01-01'
reserves = int(get_bank_reserves(cursor, first_day)) reserves = int(get_bank_reserves(cursor, first_day))
NR.append(('reserves au début de l\'année', reserves, {'label_pos':'left'})) ro_name = 'reserves au début de l\'année'
ra_name = 'reserves aujourd\'hui'
t_name = 'Total'
NR.append((ro_name, reserves, {'label_pos':'left'}))
flows.append((ro_name, t_name, reserves))
# lister les comptes de charges et produit # lister les comptes de charges et produit
accounts = get_income_and_expense_accounts(cursor) accounts = get_income_and_expense_accounts(cursor)
@ -78,23 +82,24 @@ def create_accounting_sankey(cursor, save_dir):
if a.type == 6 : if a.type == 6 :
tmp_tot = tot_deb-tot_cred tmp_tot = tot_deb-tot_cred
if tmp_tot > 0 : if tmp_tot > 0 :
NC.append((a.label, tmp_tot, {'label_pos':'right'})) NC.append((a.label, tmp_tot, {'label_pos':'right', 'color':'#007e97' }))
flows.append(('Total', a.label, tmp_tot)) flows.append((t_name, a.label, tmp_tot))
else : else :
tmp_tot = tot_cred-tot_deb tmp_tot = tot_cred-tot_deb
if tmp_tot > 0 : if tmp_tot > 0 :
tot += tmp_tot tot += tmp_tot
NR.append((a.label, tmp_tot, {'label_pos':'left'})) NR.append((a.label, tmp_tot, {'label_pos':'left', 'color':'#007e97' }))
flows.append((a.label, 'Total', tmp_tot)) flows.append((a.label, t_name, tmp_tot))
NC.insert(0, ('reserves aujourd\'hui', tmp_res, {'label_pos':'right'})) NC.insert(0, (ra_name, tmp_res, {'label_pos':'right'}))
flows.append((t_name, ra_name, tmp_res))
nodes = [ nodes = [
NR, NR,
[('Total', tot, {'label_pos':'top'})], [('Total', tot, {'label_pos':'top', 'color':'#007e97'})],
NC NC
] ]
plt.figure(figsize=(25, 10), dpi=144) plt.figure(figsize=(30, 10), dpi=144)
s = Sankey( s = Sankey(
flows=flows, flows=flows,
nodes=nodes, nodes=nodes,
@ -217,16 +222,16 @@ def get_budget_elements(budget:list[BudgetLine]):
revenue = line.revenue revenue = line.revenue
if charge != 0 and revenue == 0: if charge != 0 and revenue == 0:
total += charge total += charge
CL.append((desc, charge, {'label_pos':'right'})) CL.append((desc, charge, {'label_pos':'right', 'color':'#007e97' }))
flows.append(('Total', desc, charge)) flows.append(('Total', desc, charge))
elif charge == 0 and revenue != 0: elif charge == 0 and revenue != 0:
RL.append((desc, revenue, {'label_pos':'left'})) RL.append((desc, revenue, {'label_pos':'left', 'color':'#007e97' }))
flows.append((desc, 'Total', revenue)) flows.append((desc, 'Total', revenue))
nodes = [ nodes = [
RL, RL,
[('Total', total)], [('Total', total, {'label_pos':'top', 'color':'#007e97' })],
CL CL
] ]