File size: 1,521 Bytes
17e971c | 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 | import os
import re
from typing import Optional, Dict, List, Union, Tuple
from .base import BaseDatabaseTool, mongodb
from .util import read_file, GET_CLASS
class MongodbTool(BaseDatabaseTool):
_type_name = "mongodb"
def local_server_info(self) -> Optional[Dict]:
bin_path = "/www/server/mongodb/bin/mongod"
if not os.path.isfile(bin_path):
return None
conf_file = '/www/server/mongodb/config.conf'
conf = read_file(conf_file)
default_port = 27017
if not isinstance(conf, str):
port = default_port
else:
rep_port = re.compile(r"\s*port\s*:\s*(?P<port>\d+)", re.M)
port_res = rep_port.search(conf)
if not port_res:
port = default_port
else:
port = int(port_res.group("port"))
return {
'id': 0,
'db_host': '127.0.0.1',
'db_port': port,
'db_user': 'root',
'db_password': '',
'ps': '本地服务器',
'addtime': 0
}
# 添加一个数据库
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", "")
res = mongodb().AddDatabase(get_obj)
if res["status"] is True:
return True, "添加成功"
else:
return False, res['msg']
|