From 52333874c9e07735cf0d38aa42de92fb13098940 Mon Sep 17 00:00:00 2001
From: msramalho <19508417+msramalho@users.noreply.github.com>
Date: Wed, 9 Mar 2022 12:38:04 +0100
Subject: [PATCH] making column names configurable through the command line

---
 auto_archive.py | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/auto_archive.py b/auto_archive.py
index 2891ea0..0c0a033 100644
--- a/auto_archive.py
+++ b/auto_archive.py
@@ -61,7 +61,7 @@ def expand_url(url):
     return url
 
 
-def process_sheet(sheet, header=1):
+def process_sheet(sheet, header=1, columns=GWorksheet.COLUMN_NAMES):
     gc = gspread.service_account(filename='service_account.json')
     sh = gc.open(sheet)
 
@@ -80,16 +80,16 @@ def process_sheet(sheet, header=1):
     # loop through worksheets to check
     for ii, wks in enumerate(sh.worksheets()):
         logger.info(f'Opening worksheet {ii}: "{wks.title}"')
-        gw = GWorksheet(wks, header_row=header)
+        gw = GWorksheet(wks, header_row=header, columns=columns)
 
         if not gw.col_exists('url'):
             logger.warning(
-                f'No "Media URL" column found, skipping worksheet {wks.title}')
+                f'No "{columns["url"]}" column found, skipping worksheet {wks.title}')
             continue
 
         if not gw.col_exists('status'):
             logger.warning(
-                f'No "Archive status" column found, skipping worksheet {wks.title}')
+                f'No "{columns["status"]}" column found, skipping worksheet {wks.title}')
             continue
 
         # archives will be in a folder 'doc_name/worksheet_name'
@@ -139,7 +139,7 @@ def process_sheet(sheet, header=1):
                     update_sheet(gw, row, result)
                 else:
                     gw.set_cell(row, 'status', 'failed: no archiver')
-
+        logger.success(f'Finshed worksheet {wks.title}')
     driver.quit()
 
 
@@ -147,13 +147,17 @@ def main():
     parser = argparse.ArgumentParser(
         description='Automatically archive social media videos from a Google Sheets document')
     parser.add_argument('--sheet', action='store', dest='sheet')
-    parser.add_argument('--header', action='store', dest='header', default=1, type=int)
+    parser.add_argument('--header', action='store', dest='header', default=1, type=int, help='1-based index for the header row')
+    for k, v in GWorksheet.COLUMN_NAMES.items():
+        parser.add_argument(f'--col-{k}', action='store', dest=k, default=v, help=f'the name of the column to fill with {k} (defaults={v})')
+
     args = parser.parse_args()
+    config_columns = {k: getattr(args, k) for k in GWorksheet.COLUMN_NAMES.keys()}
 
     logger.info(f'Opening document {args.sheet}')
 
     mkdir_if_not_exists('tmp')
-    process_sheet(args.sheet, header=args.header)
+    process_sheet(args.sheet, header=args.header, columns=config_columns)
     shutil.rmtree('tmp')