import pymysql
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from ..models.lottery import Base
from .database import SQLALCHEMY_DATABASE_URL


def init_database():
    # 创建数据库
    conn = pymysql.connect(
        host='127.0.0.1',
        user='root',
        password='911!Dswybs-1024'
    )
    try:
        with conn.cursor() as cursor:
            cursor.execute('CREATE DATABASE IF NOT EXISTS lottery')
        conn.commit()
    finally:
        conn.close()

    # 创建表
    engine = create_engine(SQLALCHEMY_DATABASE_URL)
    Base.metadata.create_all(bind=engine)


if __name__ == "__main__":
    init_database()
    print("数据库初始化完成！")