diff --git a/core/admin.py b/core/admin.py index b54eadf892ec46490d97dc3debac5ce39cd2748f..c61de65ec0e7ec1e0ea13da75004b0047f62db0a 100755 --- a/core/admin.py +++ b/core/admin.py @@ -4,7 +4,19 @@ from core.models.accounts import Company, Supervisor, Student from core.models.feedbacks import Feedback from core.models.vacancies import Vacancy -admin.site.register(Company) +class CompanyAdmin(admin.ModelAdmin): + list_display = ('name', 'status') + actions = ['set_as_verified', 'set_as_unverified'] + + def set_as_verified(self, request, queryset): + queryset.update(status=Company.VERIFIED) + set_as_verified.short_description = 'Set selected as verified' + + def set_as_unverified(self, request, queryset): + queryset.update(status=Company.UNVERIFIED) + set_as_unverified.short_description = 'Set selected as unverified' + +admin.site.register(Company, CompanyAdmin) admin.site.register(Student) admin.site.register(Supervisor) admin.site.register(Vacancy)