# init_db.py
import os
from app import app, db, User

def init_database():
    with app.app_context():
        db.create_all()
        print("✅ Database tables created!")
        
        # DAFTAR LENGKAP 11 ADMIN
        admins = [
            {'username': 'Hilbraaaam', 'password': 'Ketua Angkatan 24-26'},
            {'username': 'Hudzaifahh', 'password': 'Wakil ketua Angkatan 24-26'},
            {'username': 'Rafasyahh', 'password': 'Humas Kesayangan'},
            {'username': 'Elazzam', 'password': 'Eos 800 D'},
            {'username': 'Azzam Diq', 'password': 'Shidiq'},
            {'username': 'Dzikrii', 'password': 'Bayar woe'},
            {'username': 'Ibrahim', 'password': 'Mboh'},
            {'username': 'Yusupp', 'password': 'Bangun'},
            {'username': 'Azzam JR', 'password': 'Saturn'},
            {'username': 'MK Azzam', 'password': 'Aneka Gold'},
            {'username': 'Sami abd', 'password': 'TamTam'}
        ]
        
        # CEK ADMIN YANG SUDAH ADA
        existing_admins = {admin.username for admin in User.query.filter_by(is_admin=True).all()}
        print(f"📊 Admin yang sudah ada: {len(existing_admins)}")
        
        # TAMBAHKAN ADMIN YANG BELUM ADA
        new_admin_count = 0
        for admin_data in admins:
            if admin_data['username'] not in existing_admins:
                admin = User(
                    username=admin_data['username'],
                    is_admin=True
                )
                admin.set_password(admin_data['password'])
                db.session.add(admin)
                new_admin_count += 1
                print(f"  ✓ Admin baru: {admin_data['username']}")
        
        if new_admin_count > 0:
            db.session.commit()
            print(f"✅ Berhasil menambahkan {new_admin_count} admin baru")
        else:
            print("⚠️ Tidak ada admin baru yang ditambahkan")
        
        # TAMPILKAN STATISTIK FINAL
        total_admins = User.query.filter_by(is_admin=True).count()
        total_users = User.query.count()
        print(f"\n📊 Final stats:")
        print(f"  ✅ Total admin: {total_admins}")
        print(f"  ✅ Total user: {total_users}")

if __name__ == "__main__":
    init_database()