ocr api

2025-07-17 16:00:42 +02:00
parent 7860aa3579
commit fed33b7623
9 changed files with 893 additions and 2 deletions
@@ -0,0 +1,96 @@
+#!/usr/bin/env python3
+"""
+Test script to verify PostgreSQL database creation and setup
+"""
+
+import os
+import sys
+
+# Add the app directory to the Python path
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'app'))
+
+def test_postgres_setup():
+    """Test PostgreSQL database creation and setup"""
+    print("=== Testing PostgreSQL Database Setup ===")
+
+    # Check if PostgreSQL credentials are available
+    required_vars = ['POSTGRES_HOST', 'POSTGRES_DB', 'POSTGRES_USER', 'POSTGRES_PASSWORD']
+    missing_vars = [var for var in required_vars if not os.getenv(var)]
+
+    if missing_vars:
+        print(f"⚠ PostgreSQL test skipped - missing environment variables: {', '.join(missing_vars)}")
+        print("Set these variables to test PostgreSQL:")
+        print("  POSTGRES_HOST=your-host")
+        print("  POSTGRES_DB=your-database")
+        print("  POSTGRES_USER=your-username")
+        print("  POSTGRES_PASSWORD=your-password")
+        return False
+
+    try:
+        from app.database import db_manager, init_db
+
+        print(f"Database type: {db_manager.db_type}")
+        print(f"PostgreSQL host: {os.getenv('POSTGRES_HOST')}")
+        print(f"PostgreSQL database: {os.getenv('POSTGRES_DB')}")
+        print(f"PostgreSQL user: {os.getenv('POSTGRES_USER')}")
+
+        # Initialize database (this will create the database if it doesn't exist)
+        init_db()
+        print("✓ Database initialized successfully")
+
+        # Test a simple cache operation
+        from app.database import save_to_cache, get_cached_data
+
+        test_url = "https://example.com"
+        test_route = "test"
+        test_data = {"title": "Test Page", "content": "Test content"}
+
+        # Save to cache
+        save_to_cache(test_url, test_route, test_data)
+        print("✓ Data saved to cache")
+
+        # Retrieve from cache
+        cached_data = get_cached_data(test_url, test_route)
+        if cached_data and cached_data == test_data:
+            print("✓ Data retrieved from cache successfully")
+        else:
+            print("✗ Failed to retrieve data from cache")
+            return False
+
+        # Test cache stats
+        from app.services.cache import get_cache_stats
+        stats = get_cache_stats()
+        if stats['status'] == 'success':
+            print("✓ Cache stats retrieved successfully")
+            print(f"  Database type: {stats['stats']['database_type']}")
+            print(f"  Total entries: {stats['stats']['total_entries']}")
+        else:
+            print("✗ Failed to get cache stats")
+            return False
+
+        print("✓ PostgreSQL database setup test passed!")
+        return True
+
+    except Exception as e:
+        print(f"✗ PostgreSQL setup test failed: {e}")
+        import traceback
+        traceback.print_exc()
+        return False
+
+def main():
+    """Run the PostgreSQL setup test"""
+    print("PostgreSQL Database Setup Test")
+    print("=" * 50)
+
+    success = test_postgres_setup()
+
+    print("\n" + "=" * 50)
+    if success:
+        print("🎉 PostgreSQL setup test passed! Database creation and connection working correctly.")
+        return 0
+    else:
+        print("❌ PostgreSQL setup test failed. Please check the configuration and PostgreSQL server.")
+        return 1
+
+if __name__ == "__main__":
+    sys.exit(main())