Add docstrings to new functions

airtable_to_bq.py

@@ -33,6 +33,13 @@
 
 
 def update_staging(dag: DAG, start_task, config: dict):
+    """
+    Update staging tables
+    :param dag: Dag to add tasks to
+    :param start_task: Task that should happen before the first task generated by this function
+    :param config: Task configuration
+    :return: Final task defined in this function
+    """
     bucket = DEV_DATA_BUCKET
     name = config["name"]
     sql_dir = f"sql/{DATASET}/{config.get('parent_name', name)}"
@@ -142,6 +149,14 @@ def update_staging(dag: DAG, start_task, config: dict):
 
 
 def update_production(dag: DAG, start_task, end_task, config: dict) -> None:
+    """
+    Run production table update and backup tasks
+    :param dag: Dag to run tasks within
+    :param start_task: Task that should happen before production update
+    :param end_task: Task that should happen after backup
+    :param config: Task configuration
+    :return: None
+    """
     STAGING_DATASET = f"staging_{DATASET}"
     name = config["name"]
     with dag:

bq_to_airtable.py

@@ -31,6 +31,14 @@
 
 
 def update_airtable(dag: DAG, start_task, end_task, config: dict):
+    """
+    Generates tasks needed to move data from BQ to airtable
+    :param dag: Dag we are adding tasks to
+    :param start_task: Task that should run before any generated by this function
+    :param end_task: Task that should run after any generated by this function except the `add_to_airtable` task
+    :param config: Task configuration
+    :return: The `add_to_airtable`, to be executed in series downstream
+    """
     bucket = DEV_DATA_BUCKET
     name = config["name"]
     sql_dir = f"sql/{DATASET}/{config.get('parent_name', name)}"