Trino migration to remove aws_openshift_daily (project-koku#4408)

It will get recreated with new columns.
srdeotarse · Jun 1, 2023 · 7c94c8d · 7c94c8d
1 parent 43eacc9
commit 7c94c8d
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/scripts/cji_scripts/migrate_trino.py b/scripts/cji_scripts/migrate_trino.py
@@ -119,21 +119,21 @@ def main():
     logging.info("Running against the following schemas")
     logging.info(schemas)
 
-    # tables_to_drop = [
-    #     "aws_line_items_daily",
-    # ]
+    tables_to_drop = [
+        "aws_openshift_daily",
+    ]
     # columns_to_drop = ["ocp_matched"]
-    columns_to_add = {
-        "calculated_amortized_cost": "double",
-        "markup_cost_amortized": "double",
-    }
+    # columns_to_add = {
+    #     "calculated_amortized_cost": "double",
+    #     "markup_cost_amortized": "double",
+    # }
 
     for schema in schemas:
         CONNECT_PARAMS["schema"] = schema
-        logging.info(f"*** Adding column to tables for schema {schema} ***")
-        add_columns_to_table(columns_to_add, "reporting_ocpawscostlineitem_project_daily_summary", CONNECT_PARAMS)
-        # logging.info(f"*** Dropping tables {tables_to_drop} for schema {schema} ***")
-        # drop_tables(tables_to_drop, CONNECT_PARAMS)
+        # logging.info(f"*** Adding column to tables for schema {schema} ***")
+        # add_columns_to_table(columns_to_add, "reporting_ocpawscostlineitem_project_daily_summary", CONNECT_PARAMS)
+        logging.info(f"*** Dropping tables {tables_to_drop} for schema {schema} ***")
+        drop_tables(tables_to_drop, CONNECT_PARAMS)
 
 
 if __name__ == "__main__":