mamweb/seminar/admin.py

from django.contrib import admin
from django.contrib.auth.models import Permission
from django.db import models
from django.forms import widgets

from polymorphic.admin import PolymorphicParentModelAdmin, PolymorphicChildModelAdmin, PolymorphicChildModelFilter
from reversion.admin import VersionAdmin
from django_reverse_admin import ReverseModelAdmin
from solo.admin import SingletonModelAdmin


# Todo: reversion

import seminar.models as m

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

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

	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_perm = Permission.objects.filter(codename__exact='org').first()
		print(queryset)
		for o in queryset:
			user = o.user
			user.user_permissions.add(org_perm)
			user.is_staff = True
			user.save()
			org = m.Organizator.objects.create(osoba=o)
			org.save()
	udelej_orgem.short_description = "Udělej vybraných osob organizátory"

@admin.register(m.Organizator)
class OrganizatorAdmin(admin.ModelAdmin):
	search_fields = ['osoba__jmeno', 'osoba__prijmeni', 'prezdivka']
			
@admin.register(m.Resitel)
class ResitelAdmin(admin.ModelAdmin):
	search_fields = ['jmeno', 'prijmeni', 'prezdivka']
	ordering = ('osoba__jmeno','osoba__prijmeni')

@admin.register(m.Problem)
class ProblemAdmin(PolymorphicParentModelAdmin):
	base_model = m.Problem
	child_models = [
		m.Tema,
		m.Clanek,
		m.Uloha,
		m.Konfera,
		]

@admin.register(m.Tema)
class TemaAdmin(PolymorphicChildModelAdmin):
	base_model = m.Tema
	show_in_index = True

@admin.register(m.Clanek)
class ClanekAdmin(PolymorphicChildModelAdmin):
	base_model = m.Clanek
	show_in_index = True

@admin.register(m.Uloha)
class UlohaAdmin(PolymorphicChildModelAdmin):
	base_model = m.Uloha
	show_in_index = True

@admin.register(m.Konfera)
class KonferaAdmin(PolymorphicChildModelAdmin):
	base_model = m.Konfera
	show_in_index = True


class TextAdminInline(admin.TabularInline):
	model = m.Text
	formfield_overrides = {
		models.TextField: {'widget': widgets.TextInput}
	}
	exclude = ['text_zkraceny_set','text_zkraceny']

admin.site.register(m.Text)

class ResitelInline(admin.TabularInline):
	model = m.Resitel
	extra = 1

class SoustredeniUcastniciInline(admin.TabularInline):
	model = m.Soustredeni_Ucastnici
	extra = 1
	fields = ['resitel','poznamka']
	autocomplete_fields = ['resitel']
	ordering = ['resitel__osoba__jmeno', 'resitel__osoba__prijmeni']
	formfield_overrides = {
		models.TextField: {'widget': widgets.TextInput}
	}

	def get_queryset(self,request):
		qs = super().get_queryset(request)
		return qs.select_related('resitel','soustredeni')

class SoustredeniOrganizatoriInline(admin.TabularInline):
	model = m.Soustredeni.organizatori.through
	extra = 1
	fields = ['organizator','poznamka']
	autocomplete_fields = ['organizator']
	ordering = ['organizator__osoba__jmeno','organizator__prijmeni']
	formfield_overrides = {
		models.TextField: {'widget': widgets.TextInput}
	}

	def get_queryset(self,request):
		qs = super().get_queryset(request)
		return qs.select_related('organizator', 'soustredeni')


@admin.register(m.Soustredeni)
class SoustredeniAdmin(admin.ModelAdmin):
	model = m.Soustredeni
	inline_type = 'tabular'
	inlines = [SoustredeniUcastniciInline, SoustredeniOrganizatoriInline]


class PrilohaReseniInline(admin.TabularInline):
	model = m.PrilohaReseni
	extra = 1
admin.site.register(m.PrilohaReseni)

class Reseni_ResiteleInline(admin.TabularInline):
	model = m.Reseni_Resitele


@admin.register(m.Reseni)
class ReseniAdmin(ReverseModelAdmin):
	base_model = m.Reseni
	inline_type = 'tabular'
	inline_reverse = ['text_cely','resitele']
	exclude = ['text_zkraceny', 'text_zkraceny_set']
	inlines = [PrilohaReseniInline]
# FAIL in template
#	inlines = [PrilohaReseniInline,Reseni_ResiteleInline]

admin.site.register(m.Hodnoceni)
admin.site.register(m.Pohadka)
admin.site.register(m.Obrazek)

# Polymorfismus pro stromy
# TODO: Inlines podle https://django-polymorphic.readthedocs.io/en/stable/admin.html

@admin.register(m.TreeNode)
class TreeNodeAdmin(PolymorphicParentModelAdmin):
	base_model = m.TreeNode
	child_models = [
		m.RocnikNode,
		m.CisloNode,
		m.MezicisloNode,
		m.TemaVCisleNode,
		m.UlohaZadaniNode,
		m.PohadkaNode,
		m.UlohaVzorakNode,
		m.TextNode,
		m.CastNode,
		m.OrgTextNode,
		]

	actions = ['aktualizuj_nazvy']

	# XXX: nejspíš je to totální DB HOG, nechcete to použít moc často.
	def aktualizuj_nazvy(self, request, queryset):
		newqs = queryset.get_real_instances()
		for tn in newqs:
			tn.aktualizuj_nazev()
			tn.save()
		self.message_user(request, "Názvy aktualizovány.")
	aktualizuj_nazvy.short_description = "Aktualizuj vybraným TreeNodům názvy"

@admin.register(m.RocnikNode)
class RocnikNodeAdmin(PolymorphicChildModelAdmin):
	base_model = m.RocnikNode
	show_in_index = True

@admin.register(m.CisloNode)
class CisloNodeAdmin(PolymorphicChildModelAdmin):
	base_model = m.CisloNode
	show_in_index = True

@admin.register(m.MezicisloNode)
class MezicisloNodeAdmin(PolymorphicChildModelAdmin):
	base_model = m.MezicisloNode
	show_in_index = True

@admin.register(m.TemaVCisleNode)
class TemaVCisleNodeAdmin(PolymorphicChildModelAdmin):
	base_model = m.TemaVCisleNode
	show_in_index = True

@admin.register(m.UlohaZadaniNode)
class UlohaZadaniNodeAdmin(PolymorphicChildModelAdmin):
	base_model = m.UlohaZadaniNode
	show_in_index = True

@admin.register(m.PohadkaNode)
class PohadkaNodeAdmin(PolymorphicChildModelAdmin):
	base_model = m.PohadkaNode
	show_in_index = True

@admin.register(m.UlohaVzorakNode)
class UlohaVzorakNodeAdmin(PolymorphicChildModelAdmin):
	base_model = m.UlohaVzorakNode
	show_in_index = True

@admin.register(m.TextNode)
class TextNodeAdmin(PolymorphicChildModelAdmin):
	base_model = m.TextNode
	show_in_index = True

@admin.register(m.CastNode)
class TextNodeAdmin(PolymorphicChildModelAdmin):
	base_model = m.CastNode
	show_in_index = True
	fields = ('nadpis',)

@admin.register(m.OrgTextNode)
class TextNodeAdmin(PolymorphicChildModelAdmin):
	base_model = m.OrgTextNode
	show_in_index = True


admin.site.register(m.Nastaveni, SingletonModelAdmin)
admin.site.register(m.Novinky)