mamweb/personalni/admin.py

from django.contrib import admin
from django.contrib.auth.models import Group
from django_reverse_admin import ReverseModelAdmin
from django.contrib.messages import WARNING, ERROR, SUCCESS
import seminar.models as m
from datetime import datetime


@admin.register(m.Osoba)
class OsobaAdmin(admin.ModelAdmin):
	actions = ['synchronizuj_maily', 'udelej_orgem']
	search_fields = ['jmeno', 'prijmeni', 'prezdivka']

	def synchronizuj_maily(self, request, queryset):
		for o in queryset:
			if o.user is not None:
				u = o.user
				u.email = o.email
				u.save()
		self.message_user(request, "E-maily synchronizovány.")
	synchronizuj_maily.short_description = "Synchronizuj vybraným osobám e-maily do uživatelů"

	def udelej_orgem(self,request,queryset):
		org_group = Group.objects.get(name='org')
		uspesne_vytvoreni_orgove = 0
		for o in queryset:
			if m.Organizator.objects.filter(osoba=o).exists():
				# Ref: https://docs.djangoproject.com/en/3.2/ref/contrib/admin/#django.contrib.admin.ModelAdmin.message_user
				self.message_user(request, f"Osoba {o} už je org, přeskakuji.", level=WARNING)
				continue
			user = o.user
			if user is None:
				self.message_user(request, f"Osoba {o} nemá uživatele! Přeskakuji.", level=ERROR)
				continue
			user.groups.add(org_group)
			user.is_staff = True
			user.save()
			org = m.Organizator.objects.create(osoba=o, organizuje_od=datetime.now())
			org.save()
			uspesne_vytvoreni_orgove += 1
		self.message_user(request, f'Úspěšně vytvořeno {uspesne_vytvoreni_orgove} orgů.', level=SUCCESS)
	udelej_orgem.short_description = "Udělej z vybraných osob organizátory"

class OsobaInline(admin.TabularInline):
	model = m.Osoba

@admin.register(m.Organizator)
class OrganizatorAdmin(ReverseModelAdmin):
	search_fields = ['osoba__jmeno', 'osoba__prijmeni', 'osoba__prezdivka']
	inline_type = 'stacked'
	inline_reverse = ['osoba']

@admin.register(m.Resitel)
class ResitelAdmin(ReverseModelAdmin):
	search_fields = ['osoba__jmeno', 'osoba__prijmeni', 'osoba__prezdivka']
	ordering = ('osoba__prijmeni', 'osoba__jmeno')
	inline_type = 'stacked'
	inline_reverse = ['osoba']

admin.site.register(m.Skola)
admin.site.register(m.Prijemce)