lint

Author: max-ostapenko

sql/util/generate_chart_markup.py

@@ -1,16 +1,16 @@
 import re
 import os
-import sys
 from googleapiclient.discovery import build
 import google.auth
 
 # Configuration
 SPREADSHEET_ID = '1Svyw40Th7VbigX6lpR1lb1WXwTUVKZWrK7O2YELrml4'
 PUBCHART_ID = '2PACX-1vRC5wrzy5NEsWNHn9w38RLsMURRScnP4jgjO1mDiVhsfFCY55tujlTUZhUaEWzmPtJza0QA7w8S4uK5'
-SQL_DIR = '../2025/privacy' # Relative to this script's location
+SQL_DIR = '../2025/privacy'  # Relative to this script's location
 
 SCOPES = ['https://www.googleapis.com/auth/spreadsheets']
 
+
 def get_sql_to_sheet_map(sql_dir):
     mapping = {}
     if not os.path.exists(sql_dir):
@@ -24,6 +24,7 @@ def get_sql_to_sheet_map(sql_dir):
             mapping[sheet_name] = filename
     return mapping
 
+
 def generate_figure_markup(spreadsheet_id, sql_dir):
     try:
         credentials, project = google.auth.default(scopes=SCOPES)
@@ -41,7 +42,7 @@ def generate_figure_markup(spreadsheet_id, sql_dir):
         sheet_name = sheet['properties']['title']
         sheet_id = sheet['properties']['sheetId']
         charts = sheet.get('charts', [])
-        
+       
         sql_file = sql_map.get(sheet_name)
         if not sql_file:
             # Try to match case-insensitively or show warning
@@ -50,10 +51,10 @@ def generate_figure_markup(spreadsheet_id, sql_dir):
         for chart in charts:
             title = chart['spec'].get('title', 'Untitled Chart')
             chart_id = chart['chartId']
-            
+
             # Slugify for image name
             image_name = re.sub(r'[^a-z0-9]+', '-', title.lower()).strip('-') + ".png"
-            
+
             # Construct markup
             markup = f"""{{{{ figure_markup(
   image="{image_name}",
@@ -67,12 +68,13 @@ def generate_figure_markup(spreadsheet_id, sql_dir):
             print(markup)
             print()
 
+
 if __name__ == "__main__":
     # Resolve relative SQL_DIR based on script location
     script_dir = os.path.dirname(os.path.abspath(__file__))
     absolute_sql_dir = os.path.normpath(os.path.join(script_dir, SQL_DIR))
-    
+
     print(f"Processing Spreadsheet: {SPREADSHEET_ID}")
     print(f"SQL Directory: {absolute_sql_dir}\n")
-    
+
     generate_figure_markup(SPREADSHEET_ID, absolute_sql_dir)

sql/util/requirements.txt

@@ -1,6 +1,7 @@
 pandas==2.3.3
 google-cloud-bigquery==3.40.0
 google-api-python-client==2.188.0
+google-auth==2.47.0
 requests==2.32.5
 tabulate==0.9.0
 gspread==6.2.1