foerderbarometer/input/admin.py

import csv

from django.contrib import admin
from django.db import models, transaction
from django.http import HttpResponse

from input.utils.mail import send_decision_mails

from .forms import BaseProjectForm
from .models import (
    Account,
    Project,
    ProjectCategory,
    ProjectRequest,
    ProjectDeclined,
    WikimediaProject,
    HonoraryCertificate,
    Library,
    ELiterature,
    Software,
    IFG,
    Travel,
    Email,
    BusinessCard,
    List,
    Literature,
    TYPE_PROJ,
)


class RequestURLBeforeInternNotesMixin:
    """
    Ensures that 'request_url' appears directly before 'intern_notes'.
    Works whether 'fields' is explicitly defined or derived from the Model/Form.
    """

    def get_fields(self, request, obj=None):
        fields = [*super().get_fields(request, obj)]

        fields.remove('request_url')

        index = fields.index('intern_notes')

        fields.insert(index, 'request_url')

        return fields


def export_as_csv(self, request, queryset):
    meta = self.model._meta
    field_names = [field.name for field in meta.fields]

    response = HttpResponse(content_type='text/csv')
    response['Content-Disposition'] = 'attachment; filename={}.csv'.format(meta)
    writer = csv.writer(response)

    writer.writerow(field_names)
    for obj in queryset:
        row = writer.writerow([getattr(obj, field) for field in field_names])

    return response


export_as_csv.short_description = "Ausgewähltes zu CSV exportieren"

admin.site.add_action(export_as_csv)


@admin.register(ProjectCategory, WikimediaProject)
class ProjectCategoryAdmin(admin.ModelAdmin):
    list_display = ['name', 'order', 'project_count']

    def get_queryset(self, request):
        return super().get_queryset(request).annotate(
            project_count=models.Count('projects'),
        )

    @admin.display(description='# Projekte', ordering='project_count')
    def project_count(self, obj):
        return obj.project_count


class ProjectAdminForm(BaseProjectForm):

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        for field, model in self.categories.items():
            if self.initial.get(f'{field}_other'):
                self.initial[field] = [*self.initial[field], model.other]

    def clean(self):
        cleaned_data = BaseProjectForm.clean(self)

        if self.errors:
            return cleaned_data

        if cleaned_data['granted']:
            for field in 'granted_date', 'granted_from', 'account':
                if not cleaned_data[field]:
                    self.add_error(field, 'Dieses Feld ist erforderlich, um dieses Projekt zu bewilligen.')

        return cleaned_data


class BaseProjectAdmin(admin.ModelAdmin):
    save_as = True
    form = ProjectAdminForm
    search_fields = ('name', 'pid','finance_id', 'realname', 'start', 'end', 'participants_estimated', 'participants_real', 'cost', 'status', 'end_quartal')
    list_display = ('name', 'pid','finance_id', 'realname', 'start', 'end', 'participants_estimated', 'participants_real', 'cost', 'status', 'end_quartal')
    date_hierarchy = 'end'
    readonly_fields = ('end_quartal', 'project_of_year', 'pid', 'finance_id')
    fieldsets = [
        ('Kontakt', {'fields': (
            'realname',
            'email',
        )}),
        ('Projekt', {'fields': (
            'name',
            'description',
            'start',
            'end',
            'otrs',
            'plan',
            'page',
            'urls',
            'group',
            'location',
            'participants_estimated',
            'participants_real',
            'insurance',
            'insurance_technic',
            'support',
            'cost',
            'categories',
            'categories_other',
            'wikimedia_projects',
            'wikimedia_projects_other',
            'notes',
        )}),
        ('Mailing', {'fields': (
            'mail_state',
            'end_mail_send',
            'survey_mail_send',
            'survey_mail_date',
        )}),
        ('Bewilligung', {'fields': (
            'granted',
            'granted_date',
            'granted_from',
            'intern_notes',
        )}),
        ('Accounting', {'fields': (
            'account',
            'status',
            *readonly_fields,
        )}),
    ]

    class Media:
        js = ('dropdown/js/otrs_link.js',)

    granted: bool

    def get_queryset(self, request):
        return super().get_queryset(request).filter(granted=self.granted)


@admin.register(Project)
class ProjectAdmin(BaseProjectAdmin):
    granted = True


@admin.register(ProjectRequest)
class ProjectRequestAdmin(BaseProjectAdmin):
    granted = None

    def save_model(self, request, obj: ProjectRequest, form: ProjectAdminForm, change: bool):
        super().save_model(request, obj, form, change)

        if obj.granted is None:
            return None

        transaction.on_commit(lambda: send_decision_mails(obj))

        return obj.granted


@admin.register(ProjectDeclined)
class ProjectDeclinedAdmin(BaseProjectAdmin):
    granted = False

    def has_add_permission(self, request):
        return False

    def has_change_permission(self, request, obj=None):
        return False


@admin.register(BusinessCard)
class BusinessCardAdmin(RequestURLBeforeInternNotesMixin, admin.ModelAdmin):
    save_as = True
    search_fields = ('realname', 'service_id', 'granted', 'granted_date', 'project')
    list_display = ('realname', 'service_id', 'granted', 'granted_date', 'project', 'terms_accepted')
    list_display_links = ('realname', 'service_id')
    #    action = ['export_as_csv']
    date_hierarchy = 'granted_date'
    readonly_fields = ['service_id']

    class Media:
        js = ('dropdown/js/base.js', 'dropdown/js/otrs_link.js')


@admin.register(Literature)
class LiteratureAdmin(RequestURLBeforeInternNotesMixin, admin.ModelAdmin):
    save_as = True
    search_fields = ('realname', 'service_id', 'granted', 'granted_date')
    list_display = ('realname', 'service_id', 'granted', 'granted_date', 'terms_accepted')
    list_display_links = ('realname', 'service_id')
    date_hierarchy = 'granted_date'
    readonly_fields = ['service_id']

    class Media:
        js = ('dropdown/js/otrs_link.js',)


@admin.register(Account)
class AccountAdmin(admin.ModelAdmin):
    save_as = True


@admin.register(HonoraryCertificate)
class HonoraryCertificateAdmin(admin.ModelAdmin):
    save_as = True
    search_fields = ['realname', 'granted', 'project__name', 'project__pid']
    list_display = ('realname', 'granted', 'project')
    date_hierarchy = 'granted_date'
    autocomplete_fields = ['project']

    class Media:
        js = ('dropdown/js/otrs_link.js',)


@admin.register(Library, ELiterature, Software)
class LibraryAdmin(RequestURLBeforeInternNotesMixin, admin.ModelAdmin):
    save_as = True
    search_fields = ('realname', 'service_id', 'granted', 'granted_date')
    list_display = ('realname', 'service_id', 'granted', 'granted_date')
    list_display_links = ('realname', 'service_id')
    date_hierarchy = 'granted_date'
    readonly_fields = ['service_id']
    exclude = ['type']

    class Media:
        js = ('dropdown/js/otrs_link.js',)

    def get_queryset(self, request):
        return super().get_queryset(request).filter(type=self.model.TYPE)

    def formfield_for_dbfield(self, db_field, request, **kwargs):
        if db_field.name == 'library':
            kwargs['label'] = self.model.LIBRARY_LABEL
            kwargs['help_text'] = self.model.LIBRARY_HELP_TEXT

        elif db_field.name == 'duration':
            kwargs['help_text'] = self.model.DURATION_HELP_TEXT

        return super().formfield_for_dbfield(db_field, request, **kwargs)


@admin.register(IFG)
class IFGAdmin(RequestURLBeforeInternNotesMixin, admin.ModelAdmin):
    save_as = True
    search_fields = ('realname', 'service_id', 'granted', 'granted_date')
    list_display = ('realname', 'service_id', 'granted', 'granted_date')
    list_display_links = ('realname', 'service_id')
    date_hierarchy = 'granted_date'
    readonly_fields = ['service_id']

    class Media:
        js = ('dropdown/js/otrs_link.js',)


@admin.register(Travel)
class TravelAdmin(admin.ModelAdmin):
    save_as = True
    search_fields = ['realname', 'service_id', 'granted_date', 'project__name', 'project__pid']
    list_display = ('realname', 'service_id', 'granted', 'granted_date', 'project_end', 'project',
                    'project_end_quartal')
    list_display_links = ('realname', 'project')
    date_hierarchy = 'project_end'
    autocomplete_fields = ['project']
    readonly_fields = ['service_id', 'project_end', 'project_end_quartal']

    class Media:
        js = ('dropdown/js/otrs_link.js',)


@admin.register(Email)
class EmailAdmin(RequestURLBeforeInternNotesMixin, admin.ModelAdmin):
    save_as = True
    search_fields = ('realname', 'service_id', 'granted', 'granted_date')
    list_display = ('realname', 'service_id', 'granted', 'granted_date', 'terms_accepted')
    list_display_links = ('realname', 'service_id')
    date_hierarchy = 'granted_date'
    radio_fields = {'adult': admin.VERTICAL}
    readonly_fields = ['service_id']

    class Media:
        js = ('dropdown/js/base.js', 'dropdown/js/otrs_link.js')


@admin.register(List)
class ListAdmin(RequestURLBeforeInternNotesMixin, admin.ModelAdmin):
    save_as = True
    search_fields = ('realname', 'service_id', 'granted', 'granted_date')
    list_display = ('realname', 'service_id', 'granted', 'granted_date', 'terms_accepted')
    list_display_links = ('realname', 'service_id')
    date_hierarchy = 'granted_date'
    readonly_fields = ['service_id']