From 2836fc1139a000f23c46c70467fd0768e6f8d523 Mon Sep 17 00:00:00 2001
From: mkg <1650473152@qq.com>
Date: Thu, 13 May 2021 11:35:45 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E8=A1=A5=E6=95=B0=E6=8D=AE=E5=BA=93?=
 =?UTF-8?q?=E5=AE=89=E8=A3=85=E8=84=9A=E6=9C=AC=E6=9C=AA=E6=8C=87=E5=AE=9A?=
 =?UTF-8?q?=E7=AB=AF=E5=8F=A3=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dev/initDb.py | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/dev/initDb.py b/dev/initDb.py
index 0f55553..1746370 100644
--- a/dev/initDb.py
+++ b/dev/initDb.py
@@ -1,12 +1,13 @@
 import sys
+
 sys.path.append('../')
 import sqlparse
 import pymysql
-from applications.config.database import HOST, USERNAME, PASSWORD, DATABASE,PORT
+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')
+    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)
@@ -15,7 +16,7 @@ def is_exist_database():
 
 
 def init_database():
-    db = pymysql.connect(host=HOST,port = int(PORT), user=USERNAME, password=PASSWORD, charset='utf8mb4')
+    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)
@@ -24,7 +25,8 @@ def init_database():
 
 
 def execute_fromfile(filename):
-    db = pymysql.connect(host=HOST, user=USERNAME, password=PASSWORD, database=DATABASE, charset='utf8mb4')
+    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()
@@ -53,5 +55,6 @@ def init_db():
     print('表创建成功')
     print('欢迎使用pear-admin-flask,请使用 flask run 命令启动程序')
 
+
 if __name__ == '__main__':
     init_db()