Initial commit

froide_govplan/admin.py

@@ -0,0 +1,82 @@
+from django import forms
+from django.contrib import admin
+from django.urls import reverse_lazy
+
+from tinymce.widgets import TinyMCE
+
+from froide.helper.widgets import TagAutocompleteWidget
+
+from .models import Government, GovernmentPlan, GovernmentPlanUpdate
+
+
+class GovernmentPlanAdminForm(forms.ModelForm):
+    class Meta:
+        model = GovernmentPlan
+        fields = "__all__"
+        widgets = {
+            "categories": TagAutocompleteWidget(
+                autocomplete_url=reverse_lazy("api:category-autocomplete")
+            ),
+        }
+
+
+class GovernmentPlanUpdateAdminForm(forms.ModelForm):
+    class Meta:
+        model = GovernmentPlanUpdate
+        fields = "__all__"
+        widgets = {
+            "content": TinyMCE(attrs={'cols': 80, 'rows': 30})
+        }
+
+
+class GovernmentAdmin(admin.ModelAdmin):
+    prepopulated_fields = {"slug": ("name",)}
+    list_display = ("name", "public", "start_date", "end_date")
+    list_filter = ("public",)
+
+
+class GovernmentPlanAdmin(admin.ModelAdmin):
+    form = GovernmentPlanAdminForm
+
+    save_on_top = True
+    prepopulated_fields = {"slug": ("title",)}
+    raw_id_fields = ("responsible_publicbody",)
+    list_display = (
+        "title",
+        "public",
+        "status",
+        "rating",
+    )
+    list_filter = ("status", "rating", "public", "government")
+
+
+class GovernmentPlanUpdateAdmin(admin.ModelAdmin):
+    form = GovernmentPlanUpdateAdminForm
+    raw_id_fields = ("user", "foirequest")
+    list_display = (
+        "plan",
+        "user",
+        "timestamp",
+        "status",
+        "rating",
+        "public",
+    )
+    list_filter = (
+        "status",
+        "public",
+    )
+    search_fields = ("title", "plan__title",)
+    date_hierarchy = "timestamp"
+
+    def get_queryset(self, request):
+        qs = super().get_queryset(request)
+        qs = qs.prefetch_related(
+            "plan",
+            "user",
+        )
+        return qs
+
+
+admin.site.register(Government, GovernmentAdmin)
+admin.site.register(GovernmentPlan, GovernmentPlanAdmin)
+admin.site.register(GovernmentPlanUpdate, GovernmentPlanUpdateAdmin)