58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
import sys
|
|
sys.path.append('../')
|
|
import sqlparse
|
|
import pymysql
|
|
from applications.config.database import HOST, USERNAME, PASSWORD, DATABASE,PORT
|
|
|
|
|
|
def is_exist_database():
|
|
db = pymysql.connect(host=HOST,port = int(PORT), user=USERNAME, password=PASSWORD, charset='utf8mb4')
|
|
cursor1 = db.cursor()
|
|
sql = "select * from information_schema.SCHEMATA WHERE SCHEMA_NAME = '%s' ; " % DATABASE
|
|
res = cursor1.execute(sql)
|
|
db.close()
|
|
return res
|
|
|
|
|
|
def init_database():
|
|
db = pymysql.connect(host=HOST,port = int(PORT), user=USERNAME, password=PASSWORD, charset='utf8mb4')
|
|
cursor1 = db.cursor()
|
|
sql = "CREATE DATABASE IF NOT EXISTS %s" % DATABASE
|
|
res = cursor1.execute(sql)
|
|
db.close()
|
|
return res
|
|
|
|
|
|
def execute_fromfile(filename):
|
|
db = pymysql.connect(host=HOST, user=USERNAME, password=PASSWORD, database=DATABASE, charset='utf8mb4')
|
|
fd = open(filename, 'r', encoding='utf-8')
|
|
cursor = db.cursor()
|
|
sqlfile = fd.read()
|
|
sqlfile = sqlparse.format(sqlfile, strip_comments=True).strip()
|
|
|
|
sqlcommamds = sqlfile.split(';')
|
|
|
|
for command in sqlcommamds:
|
|
try:
|
|
cursor.execute(command)
|
|
db.commit()
|
|
|
|
except Exception as msg:
|
|
|
|
db.rollback()
|
|
db.close()
|
|
|
|
|
|
def init_db():
|
|
if is_exist_database():
|
|
print('数据库已经存在,为防止误初始化，请收动删除 %s 数据库' % str(DATABASE))
|
|
return
|
|
if init_database():
|
|
print('数据库%s创建成功' % str(DATABASE))
|
|
execute_fromfile('pear.sql')
|
|
print('表创建成功')
|
|
print('欢迎使用pear-admin-flask,请使用 flask run 命令启动程序')
|
|
|
|
if __name__ == '__main__':
|
|
init_db()
|