Now add only non-zero charge

hitier
1 parent 28a7e0a2
Showing 1 changed file with 10 additions and 1 deletions Show diff stats
app/commands/commands.py
@@ -81,11 +81,19 @@ def feed_from_irap(csv_file_name):
         db.session.add(n_s)
     db.session.commit()
+    # Feed periods names
+    # Todo: should come from year column name
+    #
     for p in range(2011, 2030):
         n_p = Period(name=f"{p}")
         db.session.add(n_p)
     db.session.commit()
+    # Now feed the charges.
+    #
+    # At least one for each csv row
+    # At most one for each year
+    #
     for r in rows:
         p = Project.query.filter(Project.name == r[project_key]).one()
         a = Agent.query.filter(Agent.firstname == r[firstname_key], Agent.secondname == r[secondname_key]).one()
@@ -96,8 +104,9 @@ def feed_from_irap(csv_file_name):
             try:
                 charge = int(100 * float(charge))
             except ValueError:
-                # print(f"Wrong charge {charge}")
                 charge = 0
+            if charge == 0:
+                continue
             n_c = Charge(agent_id=a.id,
                          project_id=p.id,
                          service_id=s.id,