import os
from .database import DatabaseManager

def init_sample_data():
    db = DatabaseManager()
    # Admin user – change password after first login!
    admin = db.create_user('admin@admin.com', 'Admin', 'User', 'Admin', company='Too Many Cables')
    if admin:
        with db.get_connection() as conn:
            conn.execute("UPDATE users SET role = 'admin' WHERE id = ?", (admin,))
            conn.commit()
        print("Admin created with password 'CHANGE_ME_ADMIN_PASSWORD'")
    # Sample customer
    db.create_user('customer@example.com', 'John', 'Customer', 'customer123', company='Example Corp')
    print("Sample data initialized.")

if __name__ == '__main__':
    init_sample_data()