45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
# coding=utf-8
|
||
"""
|
||
@project: MaxKB
|
||
@Author：虎虎
|
||
@file： database_model_manage.py
|
||
@date：2025/4/15 11:06
|
||
@desc:
|
||
"""
|
||
from importlib import import_module
|
||
from django.conf import settings
|
||
|
||
|
||
def new_instance_by_class_path(class_path: str):
|
||
"""
|
||
根据class_path 创建实例
|
||
"""
|
||
parts = class_path.rpartition('.')
|
||
package_path = parts[0]
|
||
class_name = parts[2]
|
||
module = import_module(package_path)
|
||
HandlerClass = getattr(module, class_name)
|
||
return HandlerClass()
|
||
|
||
|
||
class DatabaseModelManage:
|
||
"""
|
||
模型字典
|
||
"""
|
||
model_dict = {}
|
||
|
||
@staticmethod
|
||
def get_model(model_name):
|
||
"""
|
||
根据模型
|
||
"""
|
||
return DatabaseModelManage.model_dict.get(model_name)
|
||
|
||
@staticmethod
|
||
def init():
|
||
handles = [new_instance_by_class_path(class_path) for class_path in
|
||
(settings.MODEL_HANDLES if hasattr(settings, 'MODEL_HANDLES') else [])]
|
||
for h in handles:
|
||
model_dict = h.get_model_dict()
|
||
DatabaseModelManage.model_dict = {**DatabaseModelManage.model_dict, **model_dict}
|