Merge pull request #288 from coloraven/master

mysql数据库增加支持设置charset

Author: Boris-code

feapder/db/mysqldb.py

@@ -41,7 +41,7 @@ def wapper(*args, **kwargs):
 
 class MysqlDB:
     def __init__(
-        self, ip=None, port=None, db=None, user_name=None, user_pass=None, **kwargs
+        self, ip=None, port=None, db=None, user_name=None, user_pass=None, charset="utf8mb4", **kwargs
     ):
         # 可能会改setting中的值，所以此处不能直接赋值为默认值，需要后加载赋值
         if not ip:
@@ -68,7 +68,7 @@ def __init__(
                 user=user_name,
                 passwd=user_pass,
                 db=db,
-                charset="utf8mb4",
+                charset=charset,
                 cursorclass=cursors.SSCursor,
             )  # cursorclass 使用服务的游标，默认的在多线程下大批量插入数据会使内存递增
 
@@ -83,7 +83,7 @@ def __init__(
             user_pass: {}
             exception: {}
             """.format(
-                    ip, port, db, user_name, user_pass, e
+                    ip, port, db, user_name, user_pass, charset, e
                 )
             )
         else:
@@ -117,7 +117,9 @@ def from_url(cls, url, **kwargs):
             "user_pass": url_parsed.password.strip(),
             "db": url_parsed.path.strip("/").strip(),
         }
-
+        # 解析 query 字符串参数，比如 ?charset=utf8
+        query_params = dict(parse.parse_qsl(url_parsed.query))
+        connect_params.update(query_params)
         connect_params.update(kwargs)
 
         return cls(**connect_params)