30 lines
705 B
Python
30 lines
705 B
Python
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("数据库初始化完成！")
|