bornhack-website/src/economy/admin.py

from django.contrib import admin

from .models import Chain, Credebtor, Expense, Reimbursement, Revenue

###############################
# chains and credebtors


@admin.register(Chain)
class ChainAdmin(admin.ModelAdmin):
    list_filter = ["name"]
    list_display = ["name", "notes"]
    search_fields = ["name", "notes"]


@admin.register(Credebtor)
class CredebtorAdmin(admin.ModelAdmin):
    list_filter = ["chain", "name"]
    list_display = ["chain", "name", "notes"]
    search_fields = ["chain", "name", "notes"]


###############################
# expenses


def approve_expenses(modeladmin, request, queryset):
    for expense in queryset.all():
        expense.approve(request)


approve_expenses.short_description = "Approve Expenses"


def reject_expenses(modeladmin, request, queryset):
    for expense in queryset.all():
        expense.reject(request)


reject_expenses.short_description = "Reject Expenses"


@admin.register(Expense)
class ExpenseAdmin(admin.ModelAdmin):
    list_filter = [
        "camp",
        "creditor__chain",
        "creditor",
        "responsible_team",
        "approved",
        "user",
    ]
    list_display = [
        "user",
        "description",
        "invoice_date",
        "amount",
        "camp",
        "creditor",
        "responsible_team",
        "approved",
        "reimbursement",
    ]
    search_fields = ["description", "amount", "uuid"]
    actions = [approve_expenses, reject_expenses]


###############################
# revenues


def approve_revenues(modeladmin, request, queryset):
    for revenue in queryset.all():
        revenue.approve(request)


approve_revenues.short_description = "Approve Revenues"


def reject_revenues(modeladmin, request, queryset):
    for revenue in queryset.all():
        revenue.reject(request)


reject_revenues.short_description = "Reject Revenues"


@admin.register(Revenue)
class RevenueAdmin(admin.ModelAdmin):
    list_filter = ["camp", "responsible_team", "approved", "user"]
    list_display = [
        "user",
        "description",
        "invoice_date",
        "amount",
        "camp",
        "responsible_team",
        "approved",
    ]
    search_fields = ["description", "amount", "user"]
    actions = [approve_revenues, reject_revenues]


###############################
# reimbursements


@admin.register(Reimbursement)
class ReimbursementAdmin(admin.ModelAdmin):
    def get_amount(self, obj):
        return obj.amount

    list_filter = ["camp", "user", "reimbursement_user", "paid"]
    list_display = ["camp", "user", "reimbursement_user", "paid", "notes", "get_amount"]
    search_fields = ["user__username", "reimbursement_user__username", "notes"]