mirror of
https://github.com/tencentmusic/cube-studio.git
synced 2024-12-21 06:19:31 +08:00
34 lines
1.1 KiB
Python
Executable File
34 lines
1.1 KiB
Python
Executable File
#!/usr/bin/env python
|
||
import os,time
|
||
import pysnooper
|
||
|
||
# @pysnooper.snoop()
|
||
def init_db():
|
||
SQLALCHEMY_DATABASE_URI = os.getenv('MYSQL_SERVICE', '')
|
||
if SQLALCHEMY_DATABASE_URI:
|
||
import sqlalchemy.engine.url as url
|
||
uri = url.make_url(SQLALCHEMY_DATABASE_URI)
|
||
"""Inits the Myapp application"""
|
||
import pymysql
|
||
# 创建连接
|
||
while True:
|
||
try:
|
||
# 创建连接
|
||
conn = pymysql.connect(host=uri.host, port=uri.port, user=uri.username, password=uri.password, charset='utf8')
|
||
break
|
||
except:
|
||
print('链接数据库失败，5s后重连', flush=True)
|
||
time.sleep(5)
|
||
# 创建游标
|
||
cursor = conn.cursor()
|
||
|
||
# 创建数据库的sql(如果数据库存在就不创建，防止异常)
|
||
sql = "CREATE DATABASE IF NOT EXISTS %s DEFAULT CHARACTER SET utf8 DEFAULT COLLATE utf8_general_ci;" % uri.database
|
||
print(sql)
|
||
# 执行创建数据库的sql
|
||
cursor.execute(sql)
|
||
conn.commit()
|
||
|
||
|
||
init_db()
|