增加命令行新建空白视图功能

applications/common/script/initdb.py

@@ -0,0 +1,60 @@
+from dotenv import dotenv_values
+import sqlparse
+import pymysql
+config = dotenv_values('.env')
+# MySql配置信息
+HOST = config.get('MYSQL_HOST') or '127.0.0.1'
+PORT = config.get('MYSQL_PORT') or 3306
+DATABASE = config.get('MYSQL_DATABASE') or 'PearAdminFlask'
+USERNAME = config.get('MYSQL_USERNAME') or 'root'
+PASSWORD = config.get('MYSQL_PASSWORD') or '123456'
+
+
+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, port=int(PORT), 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('test/pear.sql')
+    print('表创建成功')
+    print('欢迎使用pear-admin-flask,请使用 flask run 命令启动程序')