Django admin注冊model究竟要怎麼寫才優雅

Django admin 注冊 model、擷取子產品下的類名

比如在Django admin 注冊models時，會用到。

對于APP裡自帶的models，可以使用這種方式注冊。

from django.contrib import admin

# Register your models here.

from django.apps import apps
from django.contrib.admin.sites import AlreadyRegistered

app_models = apps.get_app_config("crm").get_models()  # 擷取app:crm下所有的model,得到一個生成器
# 周遊注冊model
for model in app_models:
    try:
        admin.site.register(model)
    except AlreadyRegistered:
        pass      

但是如果想要擴充models在Django admin頁面的顯示效果時，就沒法用上面的方法了，但是可以使用裝飾器來簡化書寫的代碼量。

from django.contrib import admin

# Register your models here.

from bbs import models


@admin.register(models.Article)
class ArticleAdmin(admin.ModelAdmin):
    list_display = ('title', 'category', 'author', 'pub_date', 'last_modify', 'status', 'priority')


@admin.register(models.Comment)
class CommentAdmin(admin.ModelAdmin):
    list_display = ('article', 'parent_comment', 'comment_type', 'comment', 'user', 'date')


@admin.register(models.Category)
class CategoryAdmin(admin.ModelAdmin):
    list_display = ('name', 'set_as_top_menu', 'position_index')


admin.site.register(models.UserProfile)      

補充一個擷取子產品中類名的方法。

import inspect

# 找出子產品裡所有的類名
def get_classes(arg):
    classes = []
    clsmembers = inspect.getmembers(arg, inspect.isclass)
    for (name, _) in clsmembers:
        classes.append(name)
    return classes