New json route for agent charge

hitier
1 parent 1aeda847
Showing 2 changed files with 16 additions and 5 deletions Show diff stats
app/db_mgr.py
app/main/routes.py
@@ -9,7 +9,8 @@ def agents():
 def charges_by_agent(agent_id):
     # all_charges = db.session.query(Charge).all()
     sql_txt = "select sum(charge_rate), p.name from charge inner join period p on charge.period_id = p.id\
-                where agent_id={} group by period_id order by p.name"\
-                .format(agent_id)
+                where agent_id={} group by period_id order by p.name" \
+        .format(agent_id)
     request = db.session.execute(sql_txt)
-    return request.fetchall()
+    all_charges = [[period, rate] for (rate, period) in request.fetchall()]
+    return all_charges
-from flask import render_template
+import json
+
+from flask import render_template, make_response
  
 from . import bp
  
 from app.models import Agent, Project, Service, Capacity, Period
+from .. import db_mgr
  
  
 @bp.route('/')
@@ -59,4 +62,11 @@ def periods():
  
 @bp.route('/charge/add')
 def charge_add():
-    return render_template('charge.html', subtitle="Affecter un agent")
 \ No newline at end of file
+    return render_template('charge.html', subtitle="Affecter un agent")
+
+
+@bp.route('/charge/agent/<agent_id>')
+def charge_user(agent_id):
+    resp = make_response(json.dumps(db_mgr.charges_by_agent(agent_id)))
+    resp.headers['Content-Type'] = 'application/json'
+    return resp