admin.py 907 B

123456789101112131415161718192021222324252627282930
  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. site_title = 'Administration de Transparency'
  9. site_header = site_title
  10. # Some models available for new/change but not listed in index pages.
  11. hidden_models = ['service', 'cost', 'good']
  12. def each_context(self, request):
  13. out = super().each_context(request)
  14. out['hidden_models'] = self.hidden_models
  15. return out
  16. admin_site = MyAdminSite()
  17. admin_site.register(User, UserAdmin)
  18. admin_site.register(Cost, CostAdmin)
  19. admin_site.register(Document, DocumentAdmin)
  20. admin_site.register(Good, GoodAdmin)
  21. admin_site.register(Service, ServiceAdmin)