Добавлена ветка prod_prep

app/db/migrate_json_to_pg.py

@@ -0,0 +1,49 @@
+"""Migrate data from JSON files to PostgreSQL."""
+import asyncio
+import json
+from pathlib import Path
+
+from app.db.postgres import PostgresDB
+
+DB_DIR = Path(__file__).resolve().parent.parent.parent / "data"
+USERS_FILE = DB_DIR / "users.json"
+BOOKINGS_FILE = DB_DIR / "bookings.json"
+
+
+async def migrate():
+    print("Starting migration from JSON to PostgreSQL...")
+    await PostgresDB.init_pool()
+
+    # Migrate users
+    if USERS_FILE.exists():
+        with open(USERS_FILE) as f:
+            users = json.load(f)
+        for uid, data in users.items():
+            name = data.get("name", "")
+            phone = data.get("phone", "")
+            await PostgresDB.set_user(int(uid), name, phone)
+        print(f"Migrated {len(users)} users")
+
+    # Migrate bookings
+    if BOOKINGS_FILE.exists():
+        with open(BOOKINGS_FILE) as f:
+            bookings = json.load(f)
+        for b in bookings:
+            # Re-create via raw INSERT to preserve IDs
+            pool = await PostgresDB.get_pool()
+            async with pool.acquire() as conn:
+                await conn.execute("""
+                    INSERT INTO bookings (id, user_id, user_name, service, master, date, time, status)
+                    VALUES ($1, $2, $3, $4, $5, $6, $7, $8)
+                    ON CONFLICT (id) DO NOTHING
+                """, b.get("id"), b.get("user_id"), b.get("user_name", ""),
+                    b.get("service"), b.get("master"), b.get("date"),
+                    b.get("time"), b.get("status", "active"))
+        print(f"Migrated {len(bookings)} bookings")
+
+    print("Migration complete!")
+    await PostgresDB.close()
+
+
+if __name__ == "__main__":
+    asyncio.run(migrate())