118 add youtube videos as source to histogram (#122)

* typo * no need to specify unit for ISO 8601 format
ncsa · May 22, 2024 · e7acf92 · e7acf92
1 parent 34567de
commit e7acf92
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/containerized_analytics/smile/histogram/histogram.py b/containerized_analytics/smile/histogram/histogram.py
@@ -67,9 +67,9 @@ def plot_freq(index, counts, interval, localPath, remotePath):
     return div_url
 
 
-def count_freq(df, time_col_name, time_freq, time_unit):
+def count_freq(df, time_col_name, time_freq, time_unit=None):
     # convert time column to datetime
-    df[time_col_name] = pd.to_datetime(df[time_col_name],unit=time_unit)
+    df[time_col_name] = pd.to_datetime(df[time_col_name], unit=time_unit)
     # set index to datetime
     df.set_index(df[time_col_name],inplace=True)
 

diff --git a/containerized_analytics/smile/histogram/rabbitmq_handler.py b/containerized_analytics/smile/histogram/rabbitmq_handler.py
@@ -111,15 +111,16 @@ def rabbitmq_handler(ch, method, properties, body):
             freq = count_freq(df, 'info.dateuploaded', interval, 's')
 
         # youtube videos
-        elif 'snnipet.publishedAt' in df.columns:
+        elif 'snippet.publishedAt' in df.columns:
             # default at 1 month
             if 'interval' in event:
                 interval = event['interval']
             else:
                 interval = '1M'
-            freq = count_freq(df, 'snnipet.publishedAt', interval, 's')
+            freq = count_freq(df, 'snippet.publishedAt', interval)
 
         else:
+            print("No valid time column found")
             return {'url': 'null'}
 
         index = freq.index.tolist()