33 lines
900 B
Python
33 lines
900 B
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
@Author ：mengying
|
||
@Date ：2024/5/31 17:31
|
||
@Email : 652044581@qq.com
|
||
@Desc : 创建数据库
|
||
"""
|
||
import pymysql
|
||
|
||
from config import config
|
||
|
||
|
||
def create_database():
|
||
"""项目启动自动创建数据表"""
|
||
connection = pymysql.connect(host=config.MYSQL_HOST, port=config.MYSQL_PORT, user=config.MYSQL_USER, passwd=config.MYSQL_PASSWORD)
|
||
create_database_sql = f"CREATE DATABASE IF NOT EXISTS {config.MYSQL_DB} CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;"
|
||
try:
|
||
# 创建游标对象
|
||
with connection.cursor() as cursor:
|
||
# 执行SQL语句
|
||
cursor.execute(create_database_sql)
|
||
# 提交事务
|
||
connection.commit()
|
||
except pymysql.MySQLError as e:
|
||
print(f"Error: {e}")
|
||
finally:
|
||
# 关闭数据库连接
|
||
connection.close()
|
||
|
||
|
||
# 初始化创建数据库
|
||
create_database()
|