diff --git a/lib/workload/stateless/stacks/metadata-manager/handler/sync_tracking_sheet.py b/lib/workload/stateless/stacks/metadata-manager/handler/sync_tracking_sheet.py index f6ad69361..4a32eca47 100644 --- a/lib/workload/stateless/stacks/metadata-manager/handler/sync_tracking_sheet.py +++ b/lib/workload/stateless/stacks/metadata-manager/handler/sync_tracking_sheet.py @@ -18,7 +18,7 @@ def handler(event, context): logger.info("Start processing update from google tracking sheet") logger.info(f'event: {libjson.dumps(event)}') - year = event.get('year', datetime.date.today().year) + year: str = str(event.get('year', datetime.date.today().year)) if isinstance(year, list): raise ValueError("Year cannot be an array") diff --git a/lib/workload/stateless/stacks/metadata-manager/proc/service/tracking_sheet_srv.py b/lib/workload/stateless/stacks/metadata-manager/proc/service/tracking_sheet_srv.py index 0dec34af6..8e5fc9595 100644 --- a/lib/workload/stateless/stacks/metadata-manager/proc/service/tracking_sheet_srv.py +++ b/lib/workload/stateless/stacks/metadata-manager/proc/service/tracking_sheet_srv.py @@ -187,7 +187,7 @@ def persist_lab_metadata(df: pd.DataFrame, sheet_year: str): } -def download_tracking_sheet(year: int) -> pd.DataFrame: +def download_tracking_sheet(year: str) -> pd.DataFrame: """ Download the full original metadata from Google tracking sheet """ @@ -195,9 +195,8 @@ def download_tracking_sheet(year: int) -> pd.DataFrame: account_info = libssm.get_secret(SSM_NAME_GDRIVE_ACCOUNT) frames = [] - year_str = str(year) - logger.info(f"Downloading {year_str} sheet") - sheet_df = libgdrive.download_sheet(account_info, sheet_id, year_str) + logger.info(f"Downloading {year} sheet") + sheet_df = libgdrive.download_sheet(account_info, sheet_id, year) sheet_df = sanitize_lab_metadata_df(sheet_df) frames.append(sheet_df)