from django.contrib import admin from rangefilter.filters import DateRangeFilter from django.contrib.auth.models import User from collabs.models import * from django import forms class Collabs_hour_Form_admin(forms.ModelForm): model = Collabs_hour fields = ["user", "sRemarques", "bNoticed"] readonly_fields = ["user", "nHour", "nMinutes"] class Collabs_hour_Admin(admin.ModelAdmin): always_show_username = True class Meta: verbose_name = 'Heure supplémentaire' verbose_name_plural = 'Heures supplémentaires' list_display = ('dtDate','nHour', 'nMinutes', 'user', 'sBases', 'get_total_hour_by_user','has_remarques', 'bNoticed') list_editable = ['bNoticed'] list_filter = [('dtDate', DateRangeFilter), ('user', admin.RelatedOnlyFieldListFilter),'sBases','type'] search_fields = ['userName'] readonly_fields = ["userName"] fields = ["sBases", "dtDate",'type', "user","nHour", "nMinutes", "sRemarques"] def get_form(self, request, obj=None, **kwargs): #if request.user.has_perm("collabs.can_notice"): if request.user.is_superuser: #kwargs['form'] = Collabs_hour_Form_admin # ModelForm print("PASS SUPERUSER") return super().get_form(request, obj, **kwargs) def formfield_for_foreignkey(self, db_field, request=None, **kwargs): field = super(Collabs_hour_Admin, self).formfield_for_foreignkey( db_field, request, **kwargs) if db_field.remote_field.model == User: field.label_from_instance = self.get_user_label return field def formfield_for_manytomany(self, db_field, request=None, **kwargs): field = super(Collabs_hour_Admin, self).formfield_for_manytomany( db_field, request, **kwargs) if db_field.remote_field.model == User: field.label_from_instance = self.get_user_label return field def get_user_label(self, user): name = user.get_full_name() username = user.username if not self.always_show_username: return name or username return (name and name != username and '%s (%s)' % (name, username) or username) def save_model(self, request, obj, form, change): obj.userName = obj.user.username super().save_model(request, obj, form, change) admin.site.register(Collabs_hour, Collabs_hour_Admin) admin.site.register(Collabs_hour_types) #admin.site.register(Collabs_hour)