New charges_for_projects() REST/CSV exposed

ce commit là je le change.

New charges_for_projects() REST/CSV exposed
ce commit là je le change.
hitier
1 parent 55d58635
Showing 3 changed files with 41 additions and 0 deletions Show diff stats
app/db_mgr.py
app/main/routes.py
tests/backend_tests.py
@@ -202,6 +202,28 @@ def charges_by_project_stacked(project_id, category=&quot;service&quot;):
     return all_charges
  
  
+def charges_for_projects_stacked():
+    sql_req = """
+    select p.name, sum(charge_rate) as tot_charg
+    from charge
+             join project p on p.id = charge.project_id
+    where period_id = {}
+    group by project_id
+    """
+    projects_req = "select name from project order by id"
+    projects_names = [name for (name,) in db.session.execute(projects_req)]
+    headers = ['period'] + projects_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)
+        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_by_agent_stacked(agent_id):
     """
     Build the list of charges for all projects of one agent, period by period
@@ -284,3 +284,16 @@ def charge_agent_csv(agent_id):
     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():
+    csv_table = []
+    for line in db_mgr.charges_for_projects_stacked():
+        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
@@ -75,6 +75,12 @@ class DbMgrTestCase(BaseTestCase):
         # Waiting for 17 periods + headers line
         self.assertEqual(18, len(stacked_charges))
  
+    def test_charges_for_projects_stacked(self):
+        stacked_charges = db_mgr.charges_for_projects_stacked()
+        # Waiting for 17 periods + headers line
+        self.assertEqual(18, len(stacked_charges))
+        self.assertEqual(102, len(stacked_charges[0]))
+
     def test_category_labels(self):
         category_labels = db_mgr.category_labels()
         categories = [_cl['name'] for _cl in category_labels]