From 5bdceaa6d9adf3f662c6eb2b68209a375388d6f1 Mon Sep 17 00:00:00 2001
From: akash <akashthomasjohn@gmail.com>
Date: Wed, 26 Jun 2024 10:18:46 +0530
Subject: [PATCH] migration script

---
 scripts/migrate_v011_to_v020.py | 81 +++++++++++++++++++++++++++++++++
 1 file changed, 81 insertions(+)
 create mode 100644 scripts/migrate_v011_to_v020.py

diff --git a/scripts/migrate_v011_to_v020.py b/scripts/migrate_v011_to_v020.py
new file mode 100644
index 0000000..e8f765a
--- /dev/null
+++ b/scripts/migrate_v011_to_v020.py
@@ -0,0 +1,81 @@
+import psycopg2
+from psycopg2 import sql
+
+# Connection parameters
+dbname = "osvauld"
+user = "postgres"
+password = "password"
+host = "host_url"
+# Connect to the PostgreSQL database
+conn = psycopg2.connect(dbname=dbname, user=user, password=password, host=host)
+
+# Create a cursor object
+cur = conn.cursor()
+
+try:
+    # Fetch data from the 'fields' table
+    cur.execute("SELECT * FROM fields")
+    rows = cur.fetchall()
+
+    # Create Data
+    field_groups = {}
+    for row in rows:
+        field_id, field_name, field_value, field_type, credential_id, user_id, created_at, created_by, updated_at, updated_by = row
+
+        key = f"{field_name}_{credential_id}"
+        
+        if key not in field_groups:
+
+            field_groups[key] = {
+                "field_name": field_name,
+                "field_type": field_type,
+                "credential_id": credential_id,
+                "created_at": created_at,
+                "created_by": created_by,
+                "updated_at": updated_at,
+                "updated_by": updated_by,
+                "user_fields": []
+            }
+
+        field_groups[key]["user_fields"].append({
+            "field_value": field_value,
+            "user_id": user_id,
+        })
+
+    for key, value in field_groups.items():
+
+        # Insert data into 'field_data'
+        cur.execute(
+            sql.SQL("INSERT INTO field_data (field_name, field_type, credential_id, created_at, created_by, updated_at, updated_by) VALUES (%s, %s, %s, %s, %s, %s, %s) RETURNING id"),
+            (value["field_name"], value["field_type"], value["credential_id"], value["created_at"], value["created_by"], value["updated_at"], value["updated_by"])
+        )
+
+        field_id = cur.fetchone()[0]
+
+        for field_value in value["user_fields"]:
+            # Insert data into 'field_values'
+            cur.execute(
+                sql.SQL("INSERT INTO field_values (field_value, user_id, field_id) VALUES (%s, %s, %s)"),
+                (field_value["field_value"], field_value["user_id"], field_id)
+            )
+
+        # Change admin user type to superadmin
+        cur.execute("UPDATE users SET type = 'superadmin' WHERE created_at = (SELECT MIN(created_at) FROM users)")
+
+    # Commit the transaction
+    conn.commit()
+    print("committed")
+    print("Data transferred successfully.")
+
+
+except Exception as e:
+    print(f"An error occurred: {e}")
+    conn.rollback()  # Roll back the transaction on error
+    raise e
+
+finally:
+    # Close the cursor and connection
+    cur.close()
+    conn.close()
+
+