admin.py 827 B

123456789101112131415161718192021222324252627
  1. from django.contrib.admin import AdminSite
  2. from django.contrib.auth.admin import UserAdmin
  3. from django.contrib.auth import get_user_model
  4. User = get_user_model()
  5. from costs.admin import CostAdmin, DocumentAdmin, GoodAdmin, ServiceAdmin
  6. from costs.models import Cost, Document, Good, Service
  7. class MyAdminSite(AdminSite):
  8. # Some models available for new/change but not listed in index pages.
  9. hidden_models = ['service', 'cost', 'good']
  10. def each_context(self, request):
  11. out = super().each_context(request)
  12. out['hidden_models'] = self.hidden_models
  13. return out
  14. admin_site = MyAdminSite()
  15. admin_site.register(User, UserAdmin)
  16. admin_site.register(Cost, CostAdmin)
  17. admin_site.register(Document, DocumentAdmin)
  18. admin_site.register(Good, GoodAdmin)
  19. admin_site.register(Service, ServiceAdmin)