erikvw/django-collect-offline

from django.contrib import admin
from django.urls import reverse

from rest_framework.authtoken.models import Token
from rest_framework.authtoken.admin import TokenAdmin

from .admin_site import django_collect_offline_admin
from .models import IncomingTransaction, OutgoingTransaction, Client, Server
from django.template.loader import render_to_string

# registering TokenAdmin with model Token fails
# if you have not declared 'rest_framework.authtoken' in INSTALLED_APPS.

# unregister to re-register with our admin class
admin.site.unregister(Token)


class TransactionModelAdminMixin:
    def view(self, obj):
        url = reverse(
            "django_collect_offline:render_url",
            kwargs={"model_name": obj._meta.object_name.lower(), "pk": str(obj.pk)},
        )
        return render_to_string(
            template_name="django_collect_offline/transaction.html",
            context={"url": url},
        )


@admin.register(Token, site=django_collect_offline_admin)
class MyTokenAdmin(TokenAdmin):
    raw_id_fields = ("user",)
    list_display = ("key", "user", "created")
    fields = ("user",)
    ordering = ("-created",)
    autocomplete_fields = ()


@admin.register(IncomingTransaction, site=django_collect_offline_admin)
class IncomingTransactionAdmin(TransactionModelAdminMixin, admin.ModelAdmin):
    ordering = ("-timestamp",)

    list_display = (
        "tx_name",
        "view",
        "producer",
        "action",
        "timestamp",
        "is_consumed",
        "is_error",
        "is_ignored",
        "consumer",
        "consumed_datetime",
        "tx_pk",
        "hostname_modified",
    )

    list_filter = (
        "is_consumed",
        "is_error",
        "is_ignored",
        "consumer",
        "consumed_datetime",
        "producer",
        "action",
        "tx_name",
        "hostname_modified",
    )

    search_fields = ("tx_pk", "tx", "timestamp", "error", "id")


@admin.register(OutgoingTransaction, site=django_collect_offline_admin)
class OutgoingTransactionAdmin(TransactionModelAdminMixin, admin.ModelAdmin):
    ordering = ("-timestamp",)

    list_display = (
        "tx_name",
        "view",
        "producer",
        "action",
        "timestamp",
        "is_consumed_server",
        "is_error",
        "consumer",
        "consumed_datetime",
        "tx_pk",
        "hostname_modified",
    )

    list_filter = (
        "is_error",
        "is_consumed_server",
        "consumer",
        "consumed_datetime",
        "producer",
        "action",
        "tx_name",
        "hostname_modified",
    )

    search_fields = ("tx_pk", "tx", "timestamp", "error", "id")


class HostAdmin(admin.ModelAdmin):
    list_display = (
        "hostname",
        "port",
        "is_active",
        "last_sync_datetime",
        "last_sync_status",
        "comment",
    )

    list_filter = ("is_active", "last_sync_datetime", "last_sync_status")


@admin.register(Client, site=django_collect_offline_admin)
class ClientAdmin(HostAdmin):
    pass


@admin.register(Server, site=django_collect_offline_admin)
class ServerAdmin(HostAdmin):
    pass