files-fix

tests/add_admin_role_db.py

@@ -0,0 +1,110 @@
+#!/usr/bin/env python3
+"""
+Добавление роли админа пользователю test_admin@discours.io через базу данных
+"""
+
+import os
+import sys
+
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
+from sqlalchemy import create_engine, text
+
+from settings import DATABASE_URL
+
+
+def add_admin_role_db():
+    """Добавляем роль админа пользователю test_admin@discours.io через базу данных"""
+
+    print("🔧 Подключаемся к базе данных...")
+    engine = create_engine(DATABASE_URL)
+
+    with engine.connect() as conn:
+        # 1. Проверяем, что пользователь test_admin@discours.io существует
+        print("🔍 Проверяем пользователя test_admin@discours.io...")
+        result = conn.execute(text("SELECT id, name, email FROM author WHERE email = 'test_admin@discours.io'"))
+        user = result.fetchone()
+
+        if not user:
+            print("❌ Пользователь test_admin@discours.io не найден в базе данных")
+            return
+
+        user_id = user[0]
+        print(f"✅ Найден пользователь: {user[1]} (ID: {user_id}, email: {user[2]})")
+
+        # 2. Проверяем, что системное сообщество существует
+        print("🔍 Проверяем системное сообщество...")
+        result = conn.execute(text("SELECT id, name, slug FROM community WHERE id = 1"))
+        community = result.fetchone()
+
+        if not community:
+            print("❌ Системное сообщество (ID=1) не найдено в базе данных")
+            return
+
+        print(f"✅ Найдено системное сообщество: {community[1]} (ID: {community[0]}, slug: {community[2]})")
+
+        # 3. Проверяем, есть ли уже роль у пользователя в системном сообществе
+        print("🔍 Проверяем существующие роли...")
+        result = conn.execute(
+            text("""
+            SELECT roles FROM community_author
+            WHERE author_id = :user_id AND community_id = :community_id
+        """),
+            {"user_id": user_id, "community_id": 1},
+        )
+
+        existing_roles_row = result.fetchone()
+        existing_roles = existing_roles_row[0].split(",") if existing_roles_row and existing_roles_row[0] else []
+        print(f"📋 Существующие роли: {existing_roles}")
+
+        # 4. Добавляем роль admin, если её нет
+        if "admin" not in existing_roles:
+            print("👑 Добавляем роль admin...")
+            if existing_roles_row:
+                # Обновляем существующую запись
+                new_roles = ",".join(existing_roles + ["admin"])
+                conn.execute(
+                    text("""
+                    UPDATE community_author
+                    SET roles = :roles
+                    WHERE author_id = :user_id AND community_id = :community_id
+                """),
+                    {"roles": new_roles, "user_id": user_id, "community_id": 1},
+                )
+            else:
+                # Создаем новую запись
+                conn.execute(
+                    text("""
+                    INSERT INTO community_author (community_id, author_id, roles, joined_at)
+                    VALUES (:community_id, :user_id, :roles, :joined_at)
+                """),
+                    {"community_id": 1, "user_id": user_id, "roles": "admin", "joined_at": 0},
+                )
+
+            conn.commit()
+            print("✅ Роль admin успешно добавлена")
+        else:
+            print("ℹ️ Роль admin уже существует")
+
+        # 5. Проверяем результат
+        print("🔍 Проверяем результат...")
+        result = conn.execute(
+            text("""
+            SELECT roles FROM community_author
+            WHERE author_id = :user_id AND community_id = :community_id
+        """),
+            {"user_id": user_id, "community_id": 1},
+        )
+
+        final_roles_row = result.fetchone()
+        final_roles = final_roles_row[0].split(",") if final_roles_row and final_roles_row[0] else []
+        print(f"📋 Финальные роли: {final_roles}")
+
+        if "admin" in final_roles:
+            print("🎉 Пользователь test_admin@discours.io теперь имеет роль admin в системном сообществе!")
+        else:
+            print("❌ Роль admin не была добавлена")
+
+
+if __name__ == "__main__":
+    add_admin_role_db()