from account.models import SignupCodeResult
from avatar.models import Avatar, avatar_storage
from django.contrib import admin
from django.contrib import messages
from django.contrib.admin import helpers
from django.contrib.admin.utils import model_ngettext
from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
from django.contrib.auth.models import User
from django.contrib.auth.models import Permission
from django.template.response import TemplateResponse
import os
from shutil import rmtree
from random import randint
from .models import UserConfig
from voty.initproc.models import Vote

class UserConfigInline(admin.StackedInline):
    model = UserConfig
    can_delete = False
    verbose_name_plural = 'config'

# Define a new User admin
class UserAdmin(BaseUserAdmin):
    inlines = (UserConfigInline, )

    actions = ['anonymize']

    def anonymize(self, request, queryset):
        if not request.POST.get('post'):
            # not confirmed yet; display the confirmation page
            return TemplateResponse(request,"initadmin/anonymize_users_confirmation.html", {
                'opts': self.model._meta,
                'users': queryset,
                'action_checkbox_name': helpers.ACTION_CHECKBOX_NAME,
            })

        # confirmed; anonymize the users
        count = queryset.count()
        if count:
            for user in queryset:
                Vote.objects.filter(user=user,initiative__state='v').delete()
                Avatar.objects.filter(user=user).delete()
                avatar_dir = avatar_storage.path(os.path.join("avatars", str(user.pk)))
                if os.path.exists(avatar_dir):
                    rmtree(avatar_dir)
                user.first_name = ''
                user.last_name = ''
                user.email = ''
                user.is_staff = False  # for good measure
                user.is_active = False
                UserConfig.objects.filter(user=user).delete()
                for result in user.signupcoderesult_set.all():
                    result.signup_code.delete() # also deletes the result, by cascading
                while True:
                    user.username = 'anon' + str(randint(1, 9999999999))
                    if not User.objects.filter(username=user.username).exists():
                        break
                user.save()

            self.message_user(request, "%d %s erfolgreich anonymisiert." % (count, model_ngettext(self.opts, count)), messages.SUCCESS)

        return None

    anonymize.short_description = 'Ausgewählte Benutzer*innen anonymisieren'

# gender-neutral verbose model name
User._meta.verbose_name = 'Benutzer*in'
User._meta.verbose_name_plural = 'Benutzer*innen'

# Re-register UserAdmin
admin.site.unregister(User)
admin.site.register(User, UserAdmin)

admin.site.register(SignupCodeResult)

admin.site.register(Permission)
@@ -36,16 +36,30 @@
 deResult
+, EmailAddress
 %0Afrom av
@@ -1518,32 +1518,96 @@
 =user).delete()%0A
+                EmailAddress.objects.filter(user=user).delete()%0A
                 

remove email address when anonymizing users