New stacked table by_labels for a category

hitier
1 parent aadf1790
Showing 2 changed files with 71 additions and 1 deletions Show diff stats
app/db_mgr.py
app/main/routes.py
@@ -205,7 +205,62 @@ def charges_by_project_stacked(project_id, category=&quot;service&quot;):
     return all_charges
  
  
+def charges_by_labels_stacked(category_id):
+    """
+    Build the list of charges for all projects, one period at a time, for each label of a category.
+
+    Will return a table like:
+
+            period,   label_0,       label_1,       ....., label_n,
+            period_0, tot_charge_00, tot_charge_01, ....., tot_charge_0n,
+            period_1, tot_charge_10, tot_charge_11, ....., tot_charge_1n,
+                .
+                .
+            period_n, tot_charge_n0, tot_charge_n1, ....., tot_charge_nn,
+
+    :param category_id:
+    :return: the table
+    """
+    sql_req = """
+              select l.name,  ifnull(sum(ps.tot_charge), 0) as label_charge
+              from label l
+                  join category_label cl on l.id = cl.label_id
+                  join project_label pl on l.id = pl.label_id
+                  left join ( select c.project_id, sum(c.charge_rate) as tot_charge
+                              from charge c where c.period_id = {} group by c.project_id) ps
+              on ps.project_id = pl.project_id
+              where cl.category_id = {}
+              group by l.id
+              order by l.id;
+              """
+    labels_req = f"""
+        select  l.name from label l
+        join category_label cl on l.id = cl.label_id
+        where cl.category_id = {category_id};
+        """
+
+    labels_names = [name for (name,) in db.session.execute(labels_req)]
+    headers = ['period'] + labels_names
+    all_charges = [headers]
+    for (period_id, period_name) in db.session.execute("select id, name from period order by id"):
+        project_charges = [period_name]
+        charges_for_projects_req = sql_req.format(period_id, category_id)
+        for (project_name, project_charge) in db.session.execute(charges_for_projects_req):
+            project_charge = str(round(project_charge / charge_unit, 2)) if project_charge else '0'
+            project_charges.append(project_charge)
+        all_charges.append(project_charges)
+    return all_charges
+
+
 def charges_for_projects_stacked():
+    """
+    TODO: code is wrong !!!
+    TODO: List of project_names is not like what is returned by the sql_join:
+    TODO: there are more project_names.
+    TODO: Should be a left join on a sub-sql query
+
+    :return:
+    """
     sql_req = """
     select p.name, sum(charge_rate) as tot_charg
     from charge
@@ -58,7 +58,9 @@ def projects():
 @role_required('project')
 def projects_stats():
     num_projects = len(Project.query.all())
-    return render_template('projects_stats.html', subtitle="Statistiques des projets ({})".format(num_projects))
+    this_categories = Category.query.all()
+    return render_template('projects_stats.html', subtitle="Statistiques des projets ({})".format(num_projects),
+                           categories=this_categories)
  
  
 @bp.route('/agents')
@@ -394,6 +396,19 @@ def charge_agent_csv(agent_id):
     return resp
  
  
+@bp.route('/charge/labels/<category_id>')
+@role_required('project')
+def labels_stats_csv(category_id):
+    csv_table = []
+    for line in db_mgr.charges_by_labels_stacked(category_id):
+        line = [cell.replace(",", "-") for cell in line]
+        line_string = ",".join(line)
+        csv_table.append(line_string)
+    resp = make_response("\n".join(csv_table))
+    resp.headers['Content-Type'] = 'text/plain;charset=utf8'
+    return resp
+
+
 @bp.route('/charge/projects')
 @role_required('project')
 def projects_stats_csv():