Archived
fix: 修复了环境变量的bug
This commit is contained in:
1 file changed
+4
-3
+4
-3
@@ -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]
|
||||||
Reference in New Issue
Block a user