feat: 支持自定义Database名称

configure/__init__.py

@@ -11,9 +11,10 @@ conf = {
     "INTRODUCTION": "",
     "INTRODUCTION_LINK": "",
     "MYSQL_URL": "localhost",
-    "MYSQL_NAME": "local",
+    "MYSQL_NAME": "localhost",
     "MYSQL_PASSWD": "123456",
     "MYSQL_PORT": 3306,
+    "MYSQL_DATABASE": "HBlog",
     "MAIL_SERVER": "",
     "MAIL_PORT": "",
     "MAIL_USE_TLS": False,

sql/__init__.py

@@ -3,4 +3,8 @@ from configure import conf
 
 
 DB = MysqlDB
-db = DB(host=conf["MYSQL_URL"], name=conf["MYSQL_NAME"], passwd=conf["MYSQL_PASSWD"], port=conf["MYSQL_PORT"])
+db = DB(host=conf["MYSQL_URL"],
+        name=conf["MYSQL_NAME"],
+        passwd=conf["MYSQL_PASSWD"],
+        port=conf["MYSQL_PORT"],
+        database=conf["MYSQL_DATABASE"])

sql/mysql.py

@@ -15,16 +15,20 @@ class MysqlDB(Database):
                  host: Optional[str],
                  name: Optional[str],
                  passwd: Optional[str],
-                 port: Optional[str]):
+                 port: Optional[str],
+                 database: str = "HBlog"):
         if host is None or name is None:
             raise DBException
+
         super(MysqlDB, self).__init__(host=host, name=name, passwd=passwd, port=port)
+        self.database = database
+
         try:
             self._db = pymysql.connect(user=self._name,
                                        password=self._passwd,
                                        host=self._host,
                                        port=self._port,
-                                       database="HBlog")
+                                       database=self.database)
         except pymysql.err.OperationalError:
             raise
         self._cursor = self._db.cursor()