From ffc42cd18a9e9539193fd1c04580d0bc447d68c5 Mon Sep 17 00:00:00 2001 From: MarkusJoe Date: Sun, 18 Sep 2022 09:19:18 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E4=BA=86=E7=8E=AF?= =?UTF-8?q?=E5=A2=83=E5=8F=98=E9=87=8F=E7=9A=84bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/db/__init__.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/db/__init__.py b/src/db/__init__.py index 703daa6..813821d 100644 --- a/src/db/__init__.py +++ b/src/db/__init__.py @@ -19,7 +19,7 @@ def split_(origin: str, type_: int = 0) -> dict: :param type_: 0 -> Mysql :return: """ - origin = origin.split('|')[-1] + origin = origin.split('://')[-1] host = origin.split('@')[-1].split('/')[0].split(':')[0] port = origin.split('@')[-1].split('/')[0].split(':')[1] if type_ == 0: # Mysql @@ -45,8 +45,9 @@ if database == 'sqlite': else: import pymysql as operator - for k, v in zip(split_(origin=database, type_=0)): - os.environ[k] = v + result = split_(database) + for k, v in zip(result.keys(), result.values()): + os.environ[f"m_{k}"] = v from src.db.db import MySQL as Database __all__ = [Database, operator]