Merge pull request #28 from ppfeufer/development

added some filter to the admin backend

afat/admin.py

@@ -9,6 +9,17 @@
 from .models import AFat, AFatLink, AFatLinkType
 
 
+def custom_filter_title(title):
+    class Wrapper(admin.FieldListFilter):
+        def __new__(cls, *args, **kwargs):
+            instance = admin.FieldListFilter.create(*args, **kwargs)
+            instance.title = title
+
+            return instance
+
+    return Wrapper
+
+
 # Register your models here.
 @admin.register(AFatLink)
 class AFatLinkAdmin(admin.ModelAdmin):
@@ -20,13 +31,28 @@ class AFatLinkAdmin(admin.ModelAdmin):
         "afattime",
         "creator",
         "fleet",
-        "link_type",
+        "_link_type",
         "is_esilink",
         "hash",
         "deleted_at",
     )
+
+    list_filter = (
+        "is_esilink",
+        ("link_type__name", custom_filter_title("fleet type")),
+    )
+
     ordering = ("-afattime",)
 
+    def _link_type(self, obj):
+        if obj.link_type:
+            return obj.link_type.name
+        else:
+            return "-"
+
+    _link_type.short_description = "Fleet Type"
+    _link_type.admin_order_field = "link_type__name"
+
 
 @admin.register(AFat)
 class AFatAdmin(admin.ModelAdmin):
@@ -35,6 +61,9 @@ class AFatAdmin(admin.ModelAdmin):
     """
 
     list_display = ("character", "system", "shiptype", "afatlink", "deleted_at")
+
+    list_filter = ("character", "system", "shiptype")
+
     ordering = ("-character",)
 
 
@@ -44,5 +73,74 @@ class AFatLinkTypeAdmin(admin.ModelAdmin):
     config for fatlinktype model
     """
 
-    list_display = ("id", "name")
+    # list_select_related = True
+
+    list_display = (
+        "id",
+        "_name",
+        "_is_enabled",
+    )
+
+    list_filter = ("is_enabled",)
+
     ordering = ("name",)
+
+    def _name(self, obj):
+        return obj.name
+
+    _name.short_description = "Fleet Type"
+    _name.admin_order_field = "name"
+
+    def _is_enabled(self, obj):
+        return obj.is_enabled
+
+    _is_enabled.boolean = True
+    _is_enabled.short_description = "Is Enabled"
+    _is_enabled.admin_order_field = "is_enabled"
+
+    actions = (
+        "mark_as_active",
+        "mark_as_inactive",
+    )
+
+    def mark_as_active(self, request, queryset):
+        """
+        Mark fleet type as active
+        :param request:
+        :param queryset:
+        """
+
+        notifications_count = 0
+
+        for obj in queryset:
+            obj.is_enabled = True
+            obj.save()
+
+            notifications_count += 1
+
+        self.message_user(
+            request, "{} fleet types marked as active".format(notifications_count)
+        )
+
+    mark_as_active.short_description = "Activate selected fleet type(s)"
+
+    def mark_as_inactive(self, request, queryset):
+        """
+        Mark fleet type as inactive
+        :param request:
+        :param queryset:
+        """
+
+        notifications_count = 0
+
+        for obj in queryset:
+            obj.is_enabled = False
+            obj.save()
+
+            notifications_count += 1
+
+        self.message_user(
+            request, "{} fleet types marked as inactive".format(notifications_count)
+        )
+
+    mark_as_inactive.short_description = "Deactivate selected fleet type(s)"

afat/views.py

@@ -699,7 +699,9 @@ def link_add(request):
     if "msg" in request.session:
         msg = request.session.pop("msg")
 
-    link_types = AFatLinkType.objects.all().order_by("name")
+    link_types = AFatLinkType.objects.filter(
+        is_enabled=True,
+    ).order_by("name")
 
     context = {"link_types": link_types, "msg": msg, "permissions": permissions}