diff --git a/hr/admin.py b/hr/admin.py index 0832a2b..c83ec1e 100644 --- a/hr/admin.py +++ b/hr/admin.py @@ -4,16 +4,21 @@ from django.contrib.auth.admin import UserAdmin from hr.models import Application, Recommendation, Audit class ApplicationAdmin(admin.ModelAdmin): - list_display = ('user', 'character', 'status') + list_display = ('user', 'character', 'status', 'recommendations') search_fields = ['user', 'character', 'status'] + def recommendations(self, obj): + return len(obj.recommendation_set.all()) + + recommendations.short_description = '# of Recommendations' + def save_model(self, request, obj, form, change): obj.save(user=request.user) admin.site.register(Application, ApplicationAdmin) class RecommendationAdmin(admin.ModelAdmin): - list_display = ('user', 'user_character') + list_display = ('user', 'user_character', 'application') search_fields = ['user_character'] admin.site.register(Recommendation, RecommendationAdmin)