Initial Burmddit deployment - AI news aggregator in Burmese

2026-02-19 02:52:58 +00:00
commit dddb86ea94
27 changed files with 5039 additions and 0 deletions
--- a/backend/init_db.py
+++ b/backend/init_db.py
@@ -0,0 +1,142 @@
+#!/usr/bin/env python3
+# Database initialization script
+
+import sys
+import os
+from loguru import logger
+import database
+import config
+
+def init_database():
+    """Initialize database with schema"""
+    logger.info("Initializing Burmddit database...")
+    
+    # Check if DATABASE_URL is set
+    if not config.DATABASE_URL:
+        logger.error("DATABASE_URL not set!")
+        logger.error("Please set it in .env file or environment")
+        return False
+    
+    logger.info(f"Connecting to database: {config.DATABASE_URL[:30]}...")
+    
+    try:
+        # Read and execute schema
+        schema_path = os.path.join(os.path.dirname(__file__), '..', 'database', 'schema.sql')
+        
+        with open(schema_path, 'r') as f:
+            schema_sql = f.read()
+        
+        with database.get_db_connection() as conn:
+            with conn.cursor() as cur:
+                cur.execute(schema_sql)
+        
+        logger.info("✅ Database schema created successfully!")
+        
+        # Verify tables exist
+        with database.get_db_connection() as conn:
+            with conn.cursor() as cur:
+                cur.execute("""
+                    SELECT table_name 
+                    FROM information_schema.tables 
+                    WHERE table_schema = 'public'
+                """)
+                tables = cur.fetchall()
+        
+        logger.info(f"Created {len(tables)} tables:")
+        for table in tables:
+            logger.info(f"  - {table[0]}")
+        
+        # Check categories
+        categories = database.get_all_categories()
+        logger.info(f"\n✅ {len(categories)} categories created:")
+        for cat in categories:
+            logger.info(f"  - {cat['name']} ({cat['name_burmese']})")
+        
+        logger.info("\n🎉 Database initialization complete!")
+        return True
+    
+    except FileNotFoundError:
+        logger.error(f"Schema file not found at: {schema_path}")
+        return False
+    
+    except Exception as e:
+        logger.error(f"Error initializing database: {e}")
+        import traceback
+        logger.error(traceback.format_exc())
+        return False
+
+def reset_database():
+    """Reset database (DANGEROUS - deletes all data!)"""
+    logger.warning("⚠️  RESETTING DATABASE - ALL DATA WILL BE LOST!")
+    
+    confirm = input("Type 'YES DELETE EVERYTHING' to confirm: ")
+    if confirm != 'YES DELETE EVERYTHING':
+        logger.info("Reset cancelled.")
+        return False
+    
+    try:
+        with database.get_db_connection() as conn:
+            with conn.cursor() as cur:
+                # Drop all tables
+                cur.execute("""
+                    DROP SCHEMA public CASCADE;
+                    CREATE SCHEMA public;
+                    GRANT ALL ON SCHEMA public TO postgres;
+                    GRANT ALL ON SCHEMA public TO public;
+                """)
+        
+        logger.info("✅ Database reset complete")
+        
+        # Reinitialize
+        return init_database()
+    
+    except Exception as e:
+        logger.error(f"Error resetting database: {e}")
+        return False
+
+def show_stats():
+    """Show database statistics"""
+    try:
+        stats = database.get_site_stats()
+        
+        logger.info("\n📊 DATABASE STATISTICS")
+        logger.info("=" * 40)
+        logger.info(f"Total articles: {stats['total_articles']}")
+        logger.info(f"Total views: {stats['total_views']}")
+        logger.info(f"Active subscribers: {stats['subscribers']}")
+        logger.info(f"Articles today: {stats['articles_today']}")
+        logger.info("=" * 40)
+        
+        # Get recent articles
+        recent = database.get_recent_articles(5)
+        logger.info(f"\n📰 RECENT ARTICLES ({len(recent)}):")
+        for article in recent:
+            logger.info(f"  - {article['title_burmese'][:50]}...")
+        
+        return True
+    
+    except Exception as e:
+        logger.error(f"Error fetching stats: {e}")
+        return False
+
+def main():
+    """Main CLI"""
+    import argparse
+    
+    parser = argparse.ArgumentParser(description='Burmddit Database Management')
+    parser.add_argument('command', choices=['init', 'reset', 'stats'], 
+                       help='Command to execute')
+    
+    args = parser.parse_args()
+    
+    if args.command == 'init':
+        success = init_database()
+    elif args.command == 'reset':
+        success = reset_database()
+    elif args.command == 'stats':
+        success = show_stats()
+    
+    sys.exit(0 if success else 1)
+
+if __name__ == '__main__':
+    main()