fix: 修复了环境变量的bug

This commit is contained in:
MarkusJoe
2022-09-18 09:19:18 +08:00
parent 95c7e3a8fa
commit ffc42cd18a
1 file changed
+4 -3
+4 -3
View File
@@ -19,7 +19,7 @@ def split_(origin: str, type_: int = 0) -> dict:
:param type_: 0 -> Mysql :param type_: 0 -> Mysql
:return: :return:
""" """
origin = origin.split('|')[-1] origin = origin.split('://')[-1]
host = origin.split('@')[-1].split('/')[0].split(':')[0] host = origin.split('@')[-1].split('/')[0].split(':')[0]
port = origin.split('@')[-1].split('/')[0].split(':')[1] port = origin.split('@')[-1].split('/')[0].split(':')[1]
if type_ == 0: # Mysql if type_ == 0: # Mysql
@@ -45,8 +45,9 @@ if database == 'sqlite':
else: else:
import pymysql as operator import pymysql as operator
for k, v in zip(split_(origin=database, type_=0)): result = split_(database)
os.environ[k] = v for k, v in zip(result.keys(), result.values()):
os.environ[f"m_{k}"] = v
from src.db.db import MySQL as Database from src.db.db import MySQL as Database
__all__ = [Database, operator] __all__ = [Database, operator]