29 lines
851 B
Python
29 lines
851 B
Python
import os
|
|
import re
|
|
|
|
from typing import Optional, Dict, List, Union, Tuple
|
|
|
|
from .base import BaseDatabaseTool, sqlserver
|
|
from .util import read_file, GET_CLASS
|
|
|
|
|
|
class SQLServerTool(BaseDatabaseTool):
|
|
_type_name = "sqlserver"
|
|
|
|
def local_server_info(self) -> Optional[Dict]:
|
|
return None
|
|
|
|
# 添加一个数据库
|
|
def add_database(self, server_id: int, database_name: str, **kwargs) -> Tuple[bool, str]:
|
|
get_obj = GET_CLASS()
|
|
get_obj.name = database_name
|
|
get_obj.sid = server_id
|
|
get_obj.ps = kwargs.get("ps", "")
|
|
get_obj.db_user = kwargs.get("db_user", "")
|
|
get_obj.password = kwargs.get("password", "")
|
|
res = sqlserver().AddDatabase(get_obj)
|
|
if res["status"] is True:
|
|
return True, "Successfully added"
|
|
else:
|
|
return False, res['msg']
|