Files
Reskreen/collabs/admin.py
2022-05-02 14:40:10 +02:00

117 lines
3.9 KiB
Python

from django.contrib import admin
from rangefilter.filters import DateRangeFilter
from django.contrib.auth.models import User
from django.db.models import Q
from django.shortcuts import render
from collabs.models import *
from django import forms
from datetime import datetime
def is_member(user, group):
return user.groups.filter(name=group).exists()
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', 'type', '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_rangefilter_dtDate_default(self, request):
return (datetime.today().replace(day=1), datetime.today())
def get_queryset(self, request):
queryset = super(Collabs_hour_Admin, self).get_queryset(request)
if not is_member(request.user, "Administration"):
queryset = queryset.filter(Q(userName=request.user.username) | Q(user=request.user))
if not "dtDate__range__gte" in request.GET.keys():
start, end = self.get_rangefilter_dtDate_default(request)
queryset = queryset.filter(dtDate__range=[start,end])
print("PASS if QUERY")
print(request.GET.keys())
return queryset
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 get_list_display(self, request):
default_list_display = super(Collabs_hour_Admin, self).get_list_display(request)
if is_member(request.user,"Administration"):
#default_list_display = ('dtDate', 'nHour', 'nMinutes', 'user', 'sBases', 'type', 'get_total_hour_by_user', 'has_remarques', 'bNoticed')
pass
else:
default_list_display = ('dtDate', 'nHour', 'nMinutes', 'user', 'sBases', 'type', 'has_remarques')
return default_list_display
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 = request.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)