# +----------------------------------------------------------------------
# | ChatWork智能聊天办公系统
# +----------------------------------------------------------------------
# | 软件声明: 本系统并非自由软件,未经授权任何形式的商业使用均属非法。
# | 版权保护: 任何企业和个人不允许对程序代码以任何形式任何目的复制/分发。
# | 授权要求: 如有商业使用需求,请务必先与版权所有者取得联系并获得正式授权。
# +----------------------------------------------------------------------
# | Author: ChatWork Team <2474369941@qq.com>
# +----------------------------------------------------------------------
import os
import importlib
from fastapi import FastAPI
from tortoise.contrib.fastapi import register_tortoise

__all__ = ["register_db"]


def __loading_db_configs():
 """ Load Db configuration """
 configs = {}
 try:
 package = importlib.import_module("config")
 clz = getattr(package, "GlobalSetting", None)
 if not clz:
 return configs

 obj = clz().dict()
 db_config = obj.get("DATABASES", {})

 if db_config.get("apps"):
 for k, v in db_config.get("apps").items():
 if v and v.get("models") and isinstance(v.get("models"), str):
 db_config["apps"][k]["models"] = __loading_model_files(v.get("models"))

 return db_config
 except ModuleNotFoundError:
 return configs


def __loading_model_files(path: str):
 """ Load Db models """
 root_path: str = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) + os.sep
 mode_path: str = root_path + path.replace(".", os.sep)

 all_files = []
 for root, dirs, filenames in os.walk(mode_path):
 dirs[:] = [d for d in dirs if not d.startswith("__")]
 path_parts = os.path.relpath(root, mode_path).split(os.sep)
 path_parts = [p for p in path_parts if not p.startswith("__")]
 file_base_path = ".".join(path_parts) if path_parts else ""
 for filename in filenames:
 if not filename.startswith("__"):
 filename_without_ext = os.path.splitext(filename)[0]
 if file_base_path and file_base_path.endswith("."):
 file_base_path = file_base_path[:-1]
 file_path = f"{file_base_path}.{filename_without_ext}" if file_base_path else filename_without_ext
 if not (file_path.startswith(".") and file_path != "."):
 all_files.append(path + "." + file_path)

 return all_files


def register_db(app: FastAPI):
 """ Connect Databases """
 register_tortoise(
 app,
 config=__loading_db_configs(),
 generate_schemas=False,
 add_exception_handlers=False,
 )