Initial YakPanel commit

mod/base/database_tool/__init__.py

@@ -0,0 +1,53 @@
+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 "Wrong database type"
+
+    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",
+]

mod/base/database_tool/base.py

@@ -0,0 +1,34 @@
+import sys
+from typing import List, Dict, Optional
+from .util import DB
+
+
+if "/www/server/panel/class" not in sys.path:
+    sys.path.insert(0, "/www/server/panel/class")
+
+from db_mysql import panelMysql
+from database import database
+from databaseModel.mongodbModel import main as mongodb
+from databaseModel.pgsqlModel import main as pgsql
+from databaseModel.sqlserverModel import main as sqlserver
+
+
+class BaseDatabaseTool:
+    _type_name = ""
+
+    def local_server_info(self) -> Optional[Dict]:
+        raise NotImplementedError()
+
+    # 获取所有可以管理的服务器的信息
+    def server_list(self) -> List[Dict]:
+        data = DB('database_servers').where("LOWER(db_type)=LOWER('?')", (self._type_name, )).select()
+        if not isinstance(data, list):
+            data = []
+        local_server = self.local_server_info()
+        if local_server is not None:
+            data.insert(0, local_server)
+        return data
+
+    # 添加一个数据库
+    def add_database(self, server_id: int, database_name: str, **kwargs) -> List[Dict]:
+        raise NotImplementedError()

mod/base/database_tool/mongodb.py

@@ -0,0 +1,51 @@
+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': 'local server',
+            '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, "Successfully added"
+        else:
+            return False, res['msg']

mod/base/database_tool/mysql.py

@@ -0,0 +1,98 @@
+import os
+import re
+
+from typing import Optional, Dict, List, Union, Tuple
+
+from .base import BaseDatabaseTool, panelMysql, database
+from .util import read_file, write_file, DB, GET_CLASS
+
+
+class MysqlTool(BaseDatabaseTool):
+    _type_name = "mysql"
+
+    def local_server_info(self) -> Optional[Dict]:
+        bin_path = "/www/server/mysql/bin/mysql"
+        if not os.path.isfile(bin_path):
+            return None
+
+        conf_file = '/etc/my.cnf'
+        conf = read_file(conf_file)
+        default_port = 3306
+        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': 'local server',
+            'addtime': 0
+        }
+
+    # 检测服务是否可以链接
+    # def server_status(self, server_id: int) -> Union[Dict, str]:
+    #     """
+    #     数据库状态检测
+    #     """
+    #     db_name = None
+    #     if server_id != 0:
+    #         conn_config = DB("database_servers").where("id=? AND LOWER(db_type)=LOWER('mysql')", (server_id,)).find()
+    #         if not conn_config:
+    #             return "远程数据库信息不存在！"
+    #         conn_config["db_name"] = None
+    #         db_user = conn_config["db_user"]
+    #         root_password = conn_config["db_password"]
+    #         db_host = conn_config["db_host"]
+    #         db_port = conn_config["db_port"]
+    #     else:
+    #         db_user = "root"
+    #         root_password = DB("config").where("id=?", (1,)).getField("mysql_root")
+    #         db_host = "localhost"
+    #         try:
+    #             db_port = int(panelMysql().query("show global variables like 'port'")[0][1])
+    #         except:
+    #             db_port = 3306
+    #     mysql_obj = panelMysql()
+    #     flag = mysql_obj.set_host(db_host, db_port, db_name, db_user, root_password)
+    #
+    #     error = ''
+    #     db_status = True
+    #     if flag is False:
+    #         db_status = False
+    #         error = mysql_obj._ex
+    #
+    #     return {
+    #         "status": True,
+    #         'error': str(error),
+    #         "msg": "正常" if db_status is True else "异常",
+    #         "db_status": db_status
+    #     }
+
+    # 添加一个数据库
+    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.db_user = kwargs.get("db_user", "")
+        get_obj.password = kwargs.get("password", "")
+        get_obj.dataAccess = kwargs.get("dataAccess", "")
+        get_obj.address = kwargs.get("address", "")
+        get_obj.codeing = kwargs.get("codeing", "")
+        get_obj.dtype = "MySQL"
+        get_obj.ps = kwargs.get("ps", "")
+        get_obj.host = kwargs.get("host", "")
+        get_obj.pid = str(kwargs.get("pid", '0'))
+        res = database().AddDatabase(get_obj)
+        if res["status"] is True:
+            return True, "Successfully added"
+        else:
+            return False, res['msg']

mod/base/database_tool/pgsql.py

@@ -0,0 +1,54 @@
+import os
+import re
+
+from typing import Optional, Dict, List, Union, Tuple
+
+from .base import BaseDatabaseTool, pgsql
+from .util import read_file, GET_CLASS
+
+
+class PgsqlTool(BaseDatabaseTool):
+    _type_name = "pgsql"
+
+    def local_server_info(self) -> Optional[Dict]:
+        bin_path = "/www/server/pgsql/bin/postgres"
+        if not os.path.isfile(bin_path):
+            return None
+
+        conf_file = '/www/server/pgsql/data/postgresql.conf'
+        conf = read_file(conf_file)
+        default_port = 5432
+        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': 'local server',
+            '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", "")
+        get_obj.db_user = kwargs.get("db_user", "")
+        get_obj.password = kwargs.get("password", "")
+        get_obj.listen_ip = kwargs.get("listen_ip", "")
+        res = pgsql().AddDatabase(get_obj)
+        if res["status"] is True:
+            return True, "Successfully added"
+        else:
+            return False, res['msg']

mod/base/database_tool/sql_server.py

@@ -0,0 +1,28 @@
+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']

mod/base/database_tool/util.py

@@ -0,0 +1,68 @@
+import os
+import sys
+from typing import Optional, Tuple, Callable
+
+if "/www/server/panel/class" not in sys.path:
+    sys.path.insert(0, "/www/server/panel/class")
+
+
+import public
+
+
+def write_file(filename: str, s_body: str, mode='w+') -> bool:
+    """
+    写入文件内容
+    @filename 文件名
+    @s_body 欲写入的内容
+    return bool 若文件不存在则尝试自动创建
+    """
+    try:
+        fp = open(filename, mode=mode)
+        fp.write(s_body)
+        fp.close()
+        return True
+    except:
+        try:
+            fp = open(filename, mode=mode, encoding="utf-8")
+            fp.write(s_body)
+            fp.close()
+            return True
+        except:
+            return False
+
+
+def read_file(filename, mode='r') -> Optional[str]:
+    """
+    读取文件内容
+    @filename 文件名
+    return string(bin) 若文件不存在，则返回None
+    """
+    import os
+    if not os.path.exists(filename):
+        return None
+    fp = None
+    try:
+        fp = open(filename, mode=mode)
+        f_body = fp.read()
+    except:
+        return None
+    finally:
+        if fp and not fp.closed:
+            fp.close()
+    return f_body
+
+
+class _DB:
+
+    def __call__(self, table: str):
+        import db
+        with db.Sql() as t:
+            t.table(table)
+            return t
+
+
+DB = _DB()
+
+GET_CLASS = public.dict_obj
+
+ExecShell: Callable = public.ExecShell