|
| 1 | +from cms.admin_mixins import RelatedReadonlyFieldsMixin |
1 | 2 | from cms.models import Page, Section, Sitemap |
2 | 3 | from django.contrib import admin |
3 | 4 |
|
4 | 5 |
|
5 | | -class SitemapAdmin(admin.ModelAdmin): |
6 | | - pass |
| 6 | +class SitemapAdmin(RelatedReadonlyFieldsMixin, admin.ModelAdmin): |
| 7 | + fields = ["id", "parent_sitemap", "page", "name", "order", "display_start_at", "display_end_at"] |
| 8 | + readonly_fields = ["id"] |
| 9 | + related_readonly_config = { |
| 10 | + "page": ["id", "is_active", "css", "title", "subtitle"], |
| 11 | + "parent_sitemap": ["id", "name", "order", "display_start_at", "display_end_at"], |
| 12 | + } |
| 13 | + |
| 14 | + def get_fieldsets(self, request, obj=...): |
| 15 | + original_fieldsets = super().get_fieldsets(request, obj) |
| 16 | + if obj and obj.parent_sitemap: |
| 17 | + original_fieldsets.append( |
| 18 | + ( |
| 19 | + "Parent Sitemap 정보", |
| 20 | + { |
| 21 | + "fields": [f"get_parent_sitemap_{f}" for f in self.related_readonly_config["parent_sitemap"]], |
| 22 | + "classes": ("collapse",), |
| 23 | + }, |
| 24 | + ) |
| 25 | + ) |
| 26 | + if obj and obj.page: |
| 27 | + original_fieldsets.append( |
| 28 | + ( |
| 29 | + "Page 정보", |
| 30 | + { |
| 31 | + "fields": [f"get_page_{f}" for f in self.related_readonly_config["page"]], |
| 32 | + "classes": ("collapse",), |
| 33 | + }, |
| 34 | + ) |
| 35 | + ) |
| 36 | + return original_fieldsets |
| 37 | + |
| 38 | + def get_queryset(self, request): |
| 39 | + return super().get_queryset(request).select_related("page").select_related("parent_sitemap") |
7 | 40 |
|
8 | 41 |
|
9 | 42 | class PageAdmin(admin.ModelAdmin): |
10 | 43 | pass |
11 | 44 |
|
12 | 45 |
|
13 | | -class SectionAdmin(admin.ModelAdmin): |
14 | | - pass |
| 46 | +class SectionAdmin(RelatedReadonlyFieldsMixin, admin.ModelAdmin): |
| 47 | + fields = ["id", "page", "order", "css", "body"] |
| 48 | + readonly_fields = ["id"] |
| 49 | + related_readonly_config = {"page": ["id", "is_active", "css", "title", "subtitle"]} |
| 50 | + |
| 51 | + def get_fieldsets(self, request, obj=...): |
| 52 | + original_fieldsets = super().get_fieldsets(request, obj) |
| 53 | + if obj and obj.page: |
| 54 | + original_fieldsets.append( |
| 55 | + ( |
| 56 | + "Page 정보", |
| 57 | + { |
| 58 | + "fields": [f"get_page_{f}" for f in self.related_readonly_config["page"]], |
| 59 | + "classes": ("collapse",), |
| 60 | + }, |
| 61 | + ) |
| 62 | + ) |
| 63 | + return original_fieldsets |
| 64 | + |
| 65 | + def get_queryset(self, request): |
| 66 | + return super().get_queryset(request).select_related("page") |
15 | 67 |
|
16 | 68 |
|
17 | 69 | admin.site.register(Sitemap, SitemapAdmin) |
|
0 commit comments