Initial commit: Ebook Translation System with Docker setup

2025-11-11 16:01:34 +01:00
commit e1b95c613d
43 changed files with 13922 additions and 0 deletions
--- a/ebook_backend&admin_panel/admin-backend/manage_test_db.py
+++ b/ebook_backend&admin_panel/admin-backend/manage_test_db.py
@@ -0,0 +1,134 @@
+"""
+Test Database Management Script
+This script helps create and manage the test database for unit tests.
+"""
+
+import psycopg2
+from psycopg2.extensions import ISOLATION_LEVEL_AUTOCOMMIT
+import sys
+
+# Test database configuration
+TEST_DB_NAME = "test_ebook_db"
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+TEST_DB_URL = os.getenv("TEST_DATABASE_URL", "postgresql://postgres:postgres@localhost:5432/test_ebook_db")
+
+def create_test_database():
+    """Create test database if it doesn't exist"""
+    try:
+        # Connect to default postgres database to create test database
+        conn = psycopg2.connect(
+            host="localhost",
+            port="5432",
+            user="postgres",
+            password="postgres",
+            database="postgres"
+        )
+        conn.set_isolation_level(ISOLATION_LEVEL_AUTOCOMMIT)
+        cursor = conn.cursor()
+        
+        # Check if test database exists
+        cursor.execute("SELECT 1 FROM pg_database WHERE datname = %s", (TEST_DB_NAME,))
+        exists = cursor.fetchone()
+        
+        if not exists:
+            cursor.execute(f"CREATE DATABASE {TEST_DB_NAME}")
+            print(f"✅ Created test database: {TEST_DB_NAME}")
+        else:
+            print(f"ℹ️  Test database {TEST_DB_NAME} already exists")
+            
+        cursor.close()
+        conn.close()
+        return True
+        
+    except Exception as e:
+        print(f"❌ Error creating test database: {e}")
+        return False
+
+def drop_test_database():
+    """Drop test database"""
+    try:
+        # Connect to default postgres database to drop test database
+        conn = psycopg2.connect(
+            host="localhost",
+            port="5432",
+            user="postgres",
+            password="postgres",
+            database="postgres"
+        )
+        conn.set_isolation_level(ISOLATION_LEVEL_AUTOCOMMIT)
+        cursor = conn.cursor()
+        
+        # Terminate all connections to test database
+        cursor.execute(f"""
+            SELECT pg_terminate_backend(pid)
+            FROM pg_stat_activity
+            WHERE datname = '{TEST_DB_NAME}' AND pid <> pg_backend_pid()
+        """)
+        
+        cursor.execute(f"DROP DATABASE IF EXISTS {TEST_DB_NAME}")
+        print(f"🗑️  Dropped test database: {TEST_DB_NAME}")
+        
+        cursor.close()
+        conn.close()
+        return True
+        
+    except Exception as e:
+        print(f"❌ Error dropping test database: {e}")
+        return False
+
+def check_test_database():
+    """Check if test database exists"""
+    try:
+        conn = psycopg2.connect(
+            host="localhost",
+            port="5432",
+            user="postgres",
+            password="postgres",
+            database="postgres"
+        )
+        cursor = conn.cursor()
+        
+        cursor.execute("SELECT 1 FROM pg_database WHERE datname = %s", (TEST_DB_NAME,))
+        exists = cursor.fetchone()
+        
+        cursor.close()
+        conn.close()
+        
+        if exists:
+            print(f"✅ Test database {TEST_DB_NAME} exists")
+            return True
+        else:
+            print(f"❌ Test database {TEST_DB_NAME} does not exist")
+            return False
+            
+    except Exception as e:
+        print(f"❌ Error checking test database: {e}")
+        return False
+
+def main():
+    """Main function to handle command line arguments"""
+    if len(sys.argv) < 2:
+        print("Usage: python manage_test_db.py [create|drop|check]")
+        print("  create - Create test database")
+        print("  drop   - Drop test database")
+        print("  check  - Check if test database exists")
+        return
+    
+    command = sys.argv[1].lower()
+    
+    if command == "create":
+        create_test_database()
+    elif command == "drop":
+        drop_test_database()
+    elif command == "check":
+        check_test_database()
+    else:
+        print(f"Unknown command: {command}")
+        print("Available commands: create, drop, check")
+
+if __name__ == "__main__":
+    main()