如何在Django REST项目中仅获取自定义模型及其方法？

嘿，我来帮你搞定这个难题！要实现只获取你自己编写的模型及其方法，还要解决JSON序列化的问题，咱们可以分三步来做：

1. 过滤出自定义应用的模型 #

Django的django.apps.get_models()会返回所有注册的模型，包括系统自带的（比如User、Group这些）。咱们可以通过模型的_meta.app_label属性判断它所属的应用，只保留你自定义的那5个应用的模型。

先定义你的自定义应用列表：

CUSTOM_APPS = ['Authentication', 'crc', 'project', 'scenarios', 'tasks']

然后筛选模型：

from django.apps import apps

def get_custom_models():
    all_models = apps.get_models()
    # 只保留属于自定义应用的模型
    custom_models = [model for model in all_models if model._meta.app_label in CUSTOM_APPS]
    return custom_models

2. 提取模型的自定义方法 #

接下来要从每个模型里提取你自己编写的方法，排除Python内置的魔法方法（比如__str__、__init__）。这里可以用inspect模块来帮咱们区分：

import inspect

def get_custom_methods(model):
    # 获取模型中所有可调用的成员（包括实例方法、类方法、静态方法）
    all_methods = inspect.getmembers(model, predicate=inspect.isroutine)
    # 过滤掉双下划线开头的内置方法
    custom_methods = [name for name, _ in all_methods if not name.startswith('__')]
    return custom_methods

如果想只保留你自己新增/重写的方法（排除从models.Model继承来的save()、delete()这类方法），可以调整判断逻辑：

def get_custom_methods(model):
    custom_methods = []
    for name, method in inspect.getmembers(model, predicate=inspect.isroutine):
        if not name.startswith('__'):
            # 仅保留定义在当前模型模块中的方法
            if method.__module__ == model.__module__:
                custom_methods.append(name)
    return custom_methods

3. 编写API视图并解决序列化问题 #

之前你遇到的“class not JSON serializable”错误，是因为直接返回了模型类对象——JSON无法序列化类本身。咱们需要把模型信息转换成字典格式，再通过DRF的Response返回。

用DRF的APIView写视图：

from rest_framework.views import APIView
from rest_framework.response import Response

class CustomModelsView(APIView):
    def get(self, request):
        custom_models = get_custom_models()
        result = []
        for model in custom_models:
            model_info = {
                'model_name': model.__name__,
                'app_label': model._meta.app_label,
                'custom_methods': get_custom_methods(model)
            }
            result.append(model_info)
        return Response(result)

最后把视图注册到URL配置里：

# 在项目或应用的urls.py中
from django.urls import path
from .views import CustomModelsView

urlpatterns = [
    # 其他URL配置
    path('custom-models/', CustomModelsView.as_view(), name='custom-models'),
]

访问这个URL，就能得到所有自定义模型的名称、所属应用以及对应的自定义方法列表，完全不会有序列化错误。

额外小提示 #

• 如果还想获取模型的字段信息，可以加上[field.name for field in model._meta.fields]来提取字段名称列表，加到model_info里就行。
• 要是需要区分实例方法、类方法和静态方法，可以在get_custom_methods里用inspect.ismethod、inspect.isclassmethod等细分判断。

内容的提问来源于stack exchange，提问作者Thinker