Fix linting

Author: pboers1988

orchestrator/migrations/helpers.py

@@ -1,5 +1,7 @@
+from uuid import uuid4
+
 import sqlalchemy as sa
-from uuid import UUID, uuid4
+
 
 def get_resource_type_id_by_name(conn, name):
     result = conn.execute(
@@ -78,12 +80,12 @@ def create_missing_modify_note_workflows(conn):
 
 def create_workflows(conn, new):
     """
-    Create a new workflow
+    Create a new workflow.
 
     Args:
         conn: DB connection as available in migration main file
         new: an dict of your workflow data
-    
+
     Example:
         >>> new_workflows = {
                 "workflow_name": {
@@ -121,13 +123,13 @@ def create_workflows(conn, new):
 
 def create_fixed_inputs(conn, product_id, new):
     """
-    Created fixed inputs for a product
+    Create a fixed inputs for a product.
 
     Args:
         conn: DB connection as available in migration main file
         product_id: UUID of the product to link to
         new: an dict of your workflow data
-        
+
     Example:
         >>> new = {
                 "fixed_input_1": "value",
@@ -138,23 +140,26 @@ def create_fixed_inputs(conn, product_id, new):
     for key, value in new.items():
         uuids[key] = uuid4()
         conn.execute(
-            sa.text("""
+            sa.text(
+                """
                 INSERT INTO fixed_inputs (fixed_input_id, name, value, created_at, product_id)
                 VALUES (:fixed_input_id, :key, :value, now(), :product_id)
                 ON CONFLICT DO NOTHING;
-            """),
+            """
+            ),
             {"fixed_input_id": uuids[key], "key": key, "value": value, "product_id": product_id},
         )
     return uuids
 
+
 def create_products(conn, new):
     """
-    Create a new workflow
+    Create a new workflow.
 
     Args:
         conn: DB connection as available in migration main file
         new: an dict of your workflow data
-    
+
     Example:
         >>> new = {
                 "Example Product": {
@@ -186,37 +191,39 @@ def create_products(conn, new):
         current_uuid = product["product_id"]
         uuids[name] = current_uuid
         conn.execute(
-            sa.text("""
+            sa.text(
+                """
                 INSERT INTO products (product_id, name, description, product_type, tag, status, created_at)
                 VALUES (:product_id, :name, :description, :product_type, :tag, :status, now())
                 ON CONFLICT DO NOTHING;
-            """),
+            """
+            ),
             product,
         )
         if "product_block_ids" in product:
             for product_block_uuid in product["product_block_ids"]:
                 # Link many-to-many if product blocks are given.
-                    conn.execute(
-                        sa.text("INSERT INTO product_product_blocks VALUES (:product_id, :product_block_id)"),
-                        {
-                            "product_id": current_uuid,
-                            "product_block_id": product_block_uuid,
-                        },
-                    )
+                conn.execute(
+                    sa.text("INSERT INTO product_product_blocks VALUES (:product_id, :product_block_id)"),
+                    {
+                        "product_id": current_uuid,
+                        "product_block_id": product_block_uuid,
+                    },
+                )
         if "fixed_inputs" in product:
             create_fixed_inputs(conn, current_uuid, product["fixed_inputs"])
     return uuids
 
 
 def create_product_blocks(conn, new):
     """
-    Create a new workflow
+    Create a new workflow.
 
     Args:
         conn: DB connection as available in migration main file
         new: an dict of your workflow data
         products: list of product block ids to link these product blocks to
-    
+
     Example:
         >>> new = {
                 "Example Product Block": {
@@ -239,14 +246,16 @@ def create_product_blocks(conn, new):
         product_block["product_block_id"] = str(product_block.get("product_block_id", uuid4()))
         uuids[name] = product_block["product_block_id"]
         conn.execute(
-            sa.text("""
+            sa.text(
+                """
                 INSERT INTO product_blocks (product_block_id, name, description, tag, status, created_at)
                 VALUES (:product_block_id, :name, :description, :tag, :status, now())
                 ON CONFLICT DO NOTHING;
-            """),
+            """
+            ),
             product_block,
         )
-            
+
     return uuids
 
 
@@ -336,10 +345,9 @@ def delete_resource_types(conn, delete):
     conn.execute(sa.text("DELETE FROM resource_types WHERE resource_type in :obsolete"), obsolete=tuple(delete))
 
 
-
 def create(conn, new):
     """
-    Call other functions in this file based on the schema
+    Call other functions in this file based on the schema.
 
     Args:
         conn: DB connection as available in migration main file
@@ -381,7 +389,7 @@ def create(conn, new):
                         "status": "active",
                         "resources": {
                             "resource_type1": "Resource description",
-                            "resource_type2": "Resource description"  
+                            "resource_type2": "Resource description"
                         }
                     },
                     "Generated UUID Product Block": {
@@ -391,14 +399,14 @@ def create(conn, new):
                         "status": "active",
                         "resources": {
                             "resource_type1": "Resource description",
-                            "resource_type3": "Resource description"  
+                            "resource_type3": "Resource description"
                         }
                     }
                 },
                 "resources": {
                     "Existing Product": {
                         "resource_type4": "Resource description",
-                        "resource_type5": "Resource description"  
+                        "resource_type5": "Resource description"
                     }
                 },
                 "workflows": {
@@ -413,7 +421,6 @@ def create(conn, new):
     """
     resources = new.get("resources", {})
     product_block_uuids = {}
-    product_uuids = {}
 
     if "product_blocks" in new:
         for product_block_name, product_block in new["product_blocks"].items():
@@ -423,7 +430,7 @@ def create(conn, new):
                 resources.update(res_dict)
                 del product_block["resources"]
         product_block_uuids = create_product_blocks(conn, new["product_blocks"])
-    
+
     if "products" in new:
         for product in new.get("product_blocks", {}).values():
             if "product_blocks" in product:
@@ -435,13 +442,13 @@ def create(conn, new):
                     except KeyError:
                         try:
                             product["product_block_ids"].append(get_product_block_id_by_name(conn, product_block_name))
-                        except:
+                        except Exception:
                             raise ValueError(f"{product_block_name} is not a valid product block.")
                 del product["product_blocks"]
-        product_uuids = create_products(conn, new["products"])
-    
+        create_products(conn, new["products"])
+
     if resources:
         create_resource_types_for_product_blocks(conn, resources)
-    
+
     if "workflows" in new:
         create_workflows(conn, new["workflows"])