Now ingest status history for Irap

hitier
1 parent b27d1dd1
Showing 1 changed file with 15 additions and 7 deletions Show diff stats
app/commands/commands.py
@@ -304,7 +304,7 @@ def feed_from_irap(csv_file_name):
         db.session.add(n_ap)
     db.session.commit()
  
-    # Now feed the charges.
+    # Now feed both charges and status history
     #
     # At least one for each csv row
     # At most one for each year
@@ -317,22 +317,30 @@ def feed_from_irap(csv_file_name):
         # TODO: period names should come from db request
         for period_name in range(2011, 2030):
             t = Period.query.filter(Period.name == period_name).one()
-            charge = r[f"{period_name}"]
+            charge_value = r[f"{period_name}"]
             # Charge are stored as percent in db, but as fraction of ETP in irap csv
             # we make the conversion here.
             try:
-                charge = int(100 * float(charge))
+                charge_value = int(100 * float(charge_value))
             except ValueError:
-                charge = 0
-            if charge == 0:
+                charge_value = 0
+            if charge_value == 0:
                 continue
             n_c = Charge(agent_id=a.id,
                          project_id=p.id,
                          service_id=s.id,
                          capacity_id=c.id,
                          period_id=t.id,
-                         charge_rate=charge)
-            db.session.add(n_c)
+                         charge_rate=charge_value)
+            status_value = r[status_key].strip().upper()
+            if not status_value:
+                continue
+            st = AgentStatus.query.filter(AgentStatus.name == status_value).one_or_none()
+            if st is None:
+                continue
+            print(f"Adding history p: {t.name}, a: {a.firstname}, s: {st.name}")
+            n_ap = AgentHistory(period_id=t.id, agent_id=a.id, status_id=st.id)
+            db.session.add_all([n_c, n_ap])
     db.session.commit()