style: improve or temprorarily ignore mypy warnings

Author: Lee-W

dags/airflow-log-cleanup.py

@@ -85,7 +85,8 @@
     schedule_interval=SCHEDULE_INTERVAL,
     start_date=START_DATE,
     tags=["teamclairvoyant", "airflow-maintenance-dags"],
-    template_undefined=jinja2.Undefined,
+    # TODO: the whole dag will be re-written at a later point. ignore the type for now
+    template_undefined=jinja2.Undefined,  # type: ignore[arg-type]
 )
 if hasattr(dag, "doc_md"):
     dag.doc_md = __doc__

dags/app/discord.py

@@ -10,7 +10,7 @@
 )
 
 
-@tenacity.retry(**RETRY_ARGS)
+@tenacity.retry(**RETRY_ARGS)  # type: ignore[call-overload]
 def send_webhook_message(webhook_url: str, username: str, msg: str) -> None:
     session.post(
         webhook_url,

dags/app/team_registration_bot/udf.py

@@ -2,6 +2,7 @@
 from datetime import datetime
 
 from google.cloud import bigquery
+from google.cloud.bigquery.table import RowIterator
 
 YEAR = datetime.now().year
 
@@ -10,7 +11,7 @@
 CLIENT = bigquery.Client(project=os.getenv("BIGQUERY_PROJECT"))
 
 
-def _get_statistics_from_bigquery() -> dict:
+def _get_statistics_from_bigquery() -> RowIterator:
     query_job = CLIENT.query(
         f"""
         WITH UNIQUE_RECORDS AS (

dags/ods/kktix_ticket_orders/udfs/bigquery_loader.py

@@ -13,11 +13,11 @@
 FROM
   `{TABLE}`
 """  # nosec
-AIRFLOW_HOME = os.getenv("AIRFLOW_HOME")
+AIRFLOW_HOME = os.getenv("AIRFLOW_HOME", "")
 
 
 def create_table_if_needed() -> None:
-    client = bigquery.Client(project=os.getenv("BIGQUERY_PROJECT"))
+    client = bigquery.Client(project=os.getenv("BIGQUERY_PROJECT", ""))
     sql_filepath = (
         Path(AIRFLOW_HOME) / "dags/ods/kktix_ticket_orders/sqls/create_table.sql"
     )