OCP/GCP pod_credit migration (project-koku#3943)

srdeotarse · Oct 18, 2022 · 4bd8fac · 4bd8fac
1 parent 37b73a5
commit 4bd8fac
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/scripts/cji_scripts/migrate_trino.py b/scripts/cji_scripts/migrate_trino.py
@@ -86,18 +86,15 @@ def main():
     logging.info("Running against the following schemas")
     logging.info(schemas)
 
-    add_column_table = [
-        "reporting_ocpgcpcostlineitem_project_daily_summary",
-        "reporting_ocpgcpcostlineitem_project_daily_summary_temp",
-    ]
     # tables_to_drop = ["gcp_openshift_daily"]
     columns_to_add = ["pod_credit"]
     # columns_to_drop = []
 
     for schema in schemas:
         CONNECT_PARAMS["schema"] = schema
         logging.info(f"*** Adding column pod_credit to tables for schema {schema} ***")
-        add_columns_to_table(columns_to_add, add_column_table, CONNECT_PARAMS)
+        add_columns_to_table(columns_to_add, "reporting_ocpgcpcostlineitem_project_daily_summary", CONNECT_PARAMS)
+        add_columns_to_table(columns_to_add, "reporting_ocpgcpcostlineitem_project_daily_summary_temp", CONNECT_PARAMS)
 
 
 if __name__ == "__main__":