HGSSystem/sql/db.py

from conf import Config
from tool.typing import List

if Config.database.upper() == 'MYSQL':
    try:
        from .mysql_db import MysqlDB
    except ImportError:
        print("无法导入MysqlDB程序")
        raise
    else:
        DB = MysqlDB
else:
    print(f"不支持的数据库类型: {Config.database}")
    raise Exception


def search_from_garbage_checker_user(columns: List[str], where, db: DB):
    cur = db.search(columns=columns, table='garbage_checker_user', where=where)
    if cur is None:
        return None
    res = cur.fetchall()
    return res