File size: 1,221 Bytes
3a5cf48 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | from .pgsql import PgsqlTool
from .mongodb import MongodbTool
from .mysql import MysqlTool
from .sql_server import SQLServerTool
from typing import Optional
DB_TYPE = (
"pgsql",
"mongodb",
"mysql",
"sqlserver"
)
def add_database(db_type: str, data: dict) -> Optional[str]:
"""
data: 中包含的有效参数为
database_name：数据库名称
server_id：数据库 id
db_user：数据库用户名
password：数据库用户的密码
dataAccess ：链接限制方式 如：ip
address：可允许使用的ip， 配合上一个参数使用
codeing： 编码
ps：备注
listen_ip: pgsql 有效，可设置访问地址
"""
if db_type not in DB_TYPE:
return "错误的数据库类型"
if db_type == "pgsql":
tool = PgsqlTool()
elif db_type == "mongodb":
tool = MongodbTool()
elif db_type == "mysql":
tool = MysqlTool()
else:
tool = SQLServerTool()
f, msg = tool.add_database(data.pop("server_id"), data.pop("database_name"), **data)
if not f:
return msg
return None
__all__ = [
"PgsqlTool",
"MongodbTool",
"MysqlTool",
"SQLServerTool",
"add_database",
]
|