114 lines
3.9 KiB
Python
114 lines
3.9 KiB
Python
from django.contrib import admin
|
|
from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
|
|
from django.contrib.auth.models import User
|
|
from django.db import models
|
|
from .models import Worker, Project, Team, WorkLog, UserProfile
|
|
|
|
|
|
# === USER ADMIN ===
|
|
|
|
class UserProfileInline(admin.StackedInline):
|
|
model = UserProfile
|
|
can_delete = False
|
|
verbose_name_plural = 'Profile'
|
|
|
|
|
|
class UserAdmin(BaseUserAdmin):
|
|
inlines = [UserProfileInline]
|
|
list_display = ('username', 'first_name', 'last_name', 'is_staff', 'get_groups')
|
|
list_filter = ('is_staff', 'is_superuser', 'is_active', 'groups')
|
|
|
|
def get_queryset(self, request):
|
|
return super().get_queryset(request).prefetch_related('groups')
|
|
|
|
fieldsets = (
|
|
(None, {'fields': ('username', 'password')}),
|
|
('Personal info', {'fields': ('first_name', 'last_name', 'email')}),
|
|
('Role & Groups', {
|
|
'description': 'Assign user to "Admin" or "Work Logger" group to set permissions automatically.',
|
|
'fields': ('is_active', 'is_staff', 'groups'),
|
|
}),
|
|
('Individual Permissions', {
|
|
'classes': ('collapse',),
|
|
'description': 'Fine-tune permissions beyond what the group provides. Usually not needed.',
|
|
'fields': ('is_superuser', 'user_permissions'),
|
|
}),
|
|
('Important dates', {
|
|
'classes': ('collapse',),
|
|
'fields': ('last_login', 'date_joined'),
|
|
}),
|
|
)
|
|
|
|
def get_groups(self, obj):
|
|
return ', '.join(g.name for g in obj.groups.all()) or '-'
|
|
get_groups.short_description = 'Groups'
|
|
|
|
|
|
admin.site.unregister(User)
|
|
admin.site.register(User, UserAdmin)
|
|
|
|
|
|
# === WORKER ADMIN ===
|
|
|
|
@admin.register(Worker)
|
|
class WorkerAdmin(admin.ModelAdmin):
|
|
list_display = ('name', 'id_no', 'phone_no', 'monthly_salary', 'date_of_employment', 'projects_worked_on_count')
|
|
search_fields = ('name', 'id_no')
|
|
readonly_fields = ('projects_worked_on_count',)
|
|
|
|
|
|
# === PROJECT ADMIN ===
|
|
|
|
@admin.register(Project)
|
|
class ProjectAdmin(admin.ModelAdmin):
|
|
list_display = ('name', 'get_supervisors', 'is_active', 'created_at')
|
|
list_filter = ('is_active',)
|
|
filter_horizontal = ('supervisors',)
|
|
|
|
def get_queryset(self, request):
|
|
return super().get_queryset(request).prefetch_related('supervisors')
|
|
|
|
def get_supervisors(self, obj):
|
|
return ', '.join(u.username for u in obj.supervisors.all()) or '-'
|
|
get_supervisors.short_description = 'Supervisors'
|
|
|
|
def formfield_for_manytomany(self, db_field, request, **kwargs):
|
|
if db_field.name == 'supervisors':
|
|
# Only show users who are staff or in Work Logger/Admin group
|
|
kwargs['queryset'] = User.objects.filter(
|
|
models.Q(is_staff=True) |
|
|
models.Q(groups__name__in=['Admin', 'Work Logger'])
|
|
).distinct().order_by('username')
|
|
return super().formfield_for_manytomany(db_field, request, **kwargs)
|
|
|
|
|
|
# === TEAM ADMIN ===
|
|
|
|
@admin.register(Team)
|
|
class TeamAdmin(admin.ModelAdmin):
|
|
list_display = ('name', 'supervisor', 'worker_count', 'is_active', 'created_at')
|
|
list_filter = ('is_active', 'supervisor')
|
|
filter_horizontal = ('workers',)
|
|
|
|
def worker_count(self, obj):
|
|
return obj.workers.count()
|
|
worker_count.short_description = 'Workers'
|
|
|
|
def formfield_for_foreignkey(self, db_field, request, **kwargs):
|
|
if db_field.name == 'supervisor':
|
|
# Only show users who are staff or in Work Logger/Admin group
|
|
kwargs['queryset'] = User.objects.filter(
|
|
models.Q(is_staff=True) |
|
|
models.Q(groups__name__in=['Admin', 'Work Logger'])
|
|
).distinct().order_by('username')
|
|
return super().formfield_for_foreignkey(db_field, request, **kwargs)
|
|
|
|
|
|
# === WORK LOG ADMIN ===
|
|
|
|
@admin.register(WorkLog)
|
|
class WorkLogAdmin(admin.ModelAdmin):
|
|
list_display = ('date', 'project', 'supervisor')
|
|
list_filter = ('date', 'project', 'supervisor')
|
|
filter_horizontal = ('workers',)
|