Merge pull request #66 from Doist/proxi/num-of-rows

fix: Fix a crash in table_data_csv with non-default num_of_rows.
Doist · Aug 20, 2024 · f521248 · f521248
2 parents 64bc512 + fb7b8e2
commit f521248
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bitmapist/cohort/tmpl/table_data_csv.mako b/bitmapist/cohort/tmpl/table_data_csv.mako
@@ -1,4 +1,4 @@
-Date,Total,${ ",".join(str(i) for i in range(0, 13)) }
+Date,Total,${ ",".join(str(i) for i in range(0, num_of_rows+1)) }
 %for row_data in dates_data:
 <%
 if time_group == 'months':
@@ -13,7 +13,7 @@ else:
 total_count = row_data[1]
 
 day_results = []
-for i in range(2, 15):
+for i in range(2, num_of_rows+3):
     prct = row_data[i]
     if prct == '':
         day_results.append('')