Add time logging to csv generator

hitier
1 parent 8e79f472
Showing 1 changed file with 6 additions and 0 deletions Show diff stats
web/run.py
@@ -903,6 +903,8 @@ def generate_csv_contents(target_slug, input_slug, started_at, stopped_at):
  
  
 def generate_csv_contents_spz(target_slug, input_slug, started_at, stopped_at):
+    now = datetime.datetime.now()
+    hp_logger.debug(f"Generating csv content with Speasy for {target_slug}/{input_slug} between {stopped_at} {stopped_at}")
     target_config = get_target_config(target_slug)
     plasma_params = {}
     for plasma_struct in target_config['models'][input_slug]:
@@ -957,6 +959,10 @@ def generate_csv_contents_spz(target_slug, input_slug, started_at, stopped_at):
     cols_ordered = [_c for _c in cols_ordered if _c in final_df.columns]
     final_df = final_df[cols_ordered]
     final_df.index.name = 'time'
+
+    then = datetime.datetime.now()
+    hp_logger.debug(f"Took {then-now} to generate")
+
     return final_df.to_csv(date_format='%Y-%m-%dT%H:%M:%S+00:00', float_format="%.5f",
                            header=True,
                            sep=",")