admin.py

import csv

from django.contrib import admin
from django.contrib import messages
from django.utils import timezone
from django.conf import settings

from plugins.consortial_billing import models
from core import files


class AgentContactInline(admin.TabularInline):
    model = models.AgentContact
    extra = 0
    raw_id_fields = ('account', 'agent')


class SupporterContactInline(admin.TabularInline):
    model = models.SupporterContact
    extra = 0
    raw_id_fields = ('account', 'supporter')


class SupporterInline(admin.TabularInline):
    fk_name = 'band'
    model = models.Supporter
    fields = ('name', '_country')
    readonly_fields = ('name', '_country')
    show_change_link = True
    can_delete = False

    def has_add_permission(self, request, obj):
        return False

    def _country(self, obj):
        return obj.band.country.name if obj and obj.band else ''


class OldBandInline(admin.TabularInline):
    model = models.OldBand
    raw_id_fields = ('band', 'supporter')
    readonly_fields = ('band', 'supporter')
    can_delete = False

    def has_add_permission(self, request, obj):
        return False


class BaseBandInline(admin.TabularInline):
    model = models.Band
    extra = 0
    verbose_name_plural = 'Base Bands (including Old Bands)'
    exclude = ('warnings',)
    readonly_fields = (
        'size',
        'country',
        'currency',
        'level',
        'datetime',
        'fee',
        'category',
    )
    can_delete = False

    def get_queryset(self, request):
        queryset = super().get_queryset(request)
        return queryset.filter(category='base').order_by('-datetime')


class BillingAgentAdmin(admin.ModelAdmin):
    list_display = (
        'name',
        '_contacts',
        'country',
        'redirect_url',
        'default',
    )
    list_editable = (
        'country',
        'default',
    )
    inlines = [
        AgentContactInline,
        BaseBandInline,
    ]

    def _contacts(self, obj):
        if obj and obj.agentcontact_set.exists():
            contacts = obj.agentcontact_set.all()
            return ', '.join([str(contact) for contact in contacts])
        else:
            return ''


class SupporterSizeAdmin(admin.ModelAdmin):
    list_display = (
        'pk',
        'name',
        'description',
        'multiplier',
        'internal_notes',
    )
    list_editable = (
        'name',
        'description',
        'multiplier',
        'internal_notes',
    )
    inlines = [
        BaseBandInline,
    ]


class SupportLevelAdmin(admin.ModelAdmin):
    list_display = (
        'pk',
        'name',
        'description',
        'order',
        'internal_notes',
        'default',
    )
    list_editable = (
        'name',
        'description',
        'order',
        'internal_notes',
        'default',
    )
    inlines = [
        BaseBandInline,
    ]


class CurrencyAdmin(admin.ModelAdmin):
    list_display = (
        'pk',
        'code',
        'symbol',
        'region',
        '_exchange_rate',
        'internal_notes',
    )
    list_editable = (
        'code',
        'symbol',
        'region',
        'internal_notes',
    )
    readonly_fields = (
        '_exchange_rate',
    )
    inlines = [
        BaseBandInline,
    ]


    def _exchange_rate(self, obj):
        if obj:
            rate, warnings = obj.exchange_rate()
            return rate
        else:
            return ''


class BandAdmin(admin.ModelAdmin):
    list_display = (
        'pk',
        'size',
        'country',
        'currency',
        'level',
        'fee',
        'datetime',
        'billing_agent',
        'category',
    )
    list_filter = (
        'category',
        'datetime',
        'size',
        'currency',
        'level',
        'fee',
        'billing_agent',
        'country',
    )
    date_hierarchy = 'datetime'
    inlines = [
        SupporterInline,
        OldBandInline,
    ]


class SupporterAdmin(admin.ModelAdmin):

    # For performance
    list_per_page = 25

    list_display = (
        'name',
        '_contacts',
        'address',
        'postal_code',
        '_country',
        'size',
        'level',
        'fee',
        'currency',
        'billing_agent',
        'active',
        'display',
    )
    search_fields = (
        'name',
        'id',
        'ror',
        'address',
        'postal_code',
        'supportercontact__account__first_name',
        'supportercontact__account__last_name',
        'supportercontact__account__email',
        'internal_notes',
    )
    list_filter = (
        'active',
        'band__size',
        'band__level',
        'band__currency',
        'band__country',
    )
    raw_id_fields = (
        'band',
        'prospective_band',
    )

    inlines = [
        SupporterContactInline,
        OldBandInline,
    ]

    actions = [
        'export_supporters',
    ]

    def _contacts(self, obj):
        if obj and obj.supportercontact_set.exists():
            contacts = obj.supportercontact_set.all()
            return ', '.join([str(contact) for contact in contacts])
        else:
            return ''

    def _country(self, obj):
        return obj.band.country.name if obj and obj.band else ''

    @admin.action(description="Export selected supporters")
    def export_supporters(self, request, queryset):
        """
        Basic way of exporting supporters.
        It is better to use django's dumpdata if you have access to the command
        line.
        """
        fieldnames = [
            'name',
            'ror',
            'address',
            'postal_code',
            'display',
            'active',
            'size',
            'currency',
            'level',
            'fee',
            'warnings',
            'datetime',
            'billing_agent',
        ]

        timestamp = timezone.now().strftime('%Y-%m-%dT%H-%M-%S')
        filename = f'supporters_export_{timestamp}.csv'
        filepath = files.create_temp_file('', filename)
        with open(filepath, 'w') as file:
            writer = csv.DictWriter(file, fieldnames=fieldnames)
            writer.writeheader()

            for supporter in queryset:
                row = {}
                for fieldname in fieldnames:
                    row[fieldname] = getattr(supporter, fieldname)

                writer.writerow(row)

        count = queryset.count()
        self.message_user(
            request,
            f"Downloaded {count} records",
            messages.SUCCESS,
        )
        return files.serve_temp_file(filepath, filename)


admin_list = [
    (models.BillingAgent, BillingAgentAdmin),
    (models.SupporterSize, SupporterSizeAdmin),
    (models.SupportLevel, SupportLevelAdmin),
    (models.Currency, CurrencyAdmin),
    (models.Band, BandAdmin),
    (models.Supporter, SupporterAdmin),
]


[admin.site.register(*t) for t in admin_list]