troggle-unchained/core/admin.py

175 lines
5.3 KiB
Python
Raw Normal View History

2011-07-11 02:10:22 +01:00
from django.contrib import admin
2023-01-19 18:35:56 +00:00
from django.core import serializers
2011-07-11 02:10:22 +01:00
from django.http import HttpResponse
from troggle.core.models.caves import Area, Cave, CaveAndEntrance, Entrance
2023-07-08 23:43:57 +01:00
from troggle.core.models.logbooks import QM, LogbookEntry, PersonLogEntry, CaveSlug
from troggle.core.models.survex import (
DrawingFile,
SingleScan,
SurvexBlock,
SurvexDirectory,
SurvexFile,
SurvexPersonRole,
SurvexStation,
)
from troggle.core.models.wallets import Wallet
from troggle.core.models.troggle import DataIssue, Expedition, Person, PersonExpedition
2011-07-11 02:10:22 +01:00
"""This code significantly adds to the capabilities of the Django Management control panel for Troggle data.
In particular, it enables JSON export of any data with 'export_as_json'
and configures the search fields to be used within the control panel.
What is the search path for the css and js inclusions in the Media subclasses though ?!
The page looks for /static/jquery/jquery.min.js
"""
2011-07-11 02:10:22 +01:00
class TroggleModelAdmin(admin.ModelAdmin):
def save_model(self, request, obj, form, change):
2022-03-23 22:55:43 +00:00
"""overriding admin save to fill the new_since parsing_field
2022-03-23 22:55:43 +00:00
new_since_parsing is not currently used in troggle. It is a fossil."""
obj.new_since_parsing = True
obj.save()
2011-07-11 02:10:22 +01:00
class Media:
js = ("jquery/jquery.min.js", "js/QM_helper.js") # not currently available to troggle, see media/js/README
2011-07-11 02:10:22 +01:00
2011-07-11 02:10:22 +01:00
class RoleInline(admin.TabularInline):
model = SurvexPersonRole
extra = 4
2011-07-11 02:10:22 +01:00
class SurvexBlockAdmin(TroggleModelAdmin):
inlines = (RoleInline,)
2023-03-17 20:01:52 +00:00
# class QMsFoundInline(admin.TabularInline):
# model = QM
# fk_name = "found_by"
# fields = ("number", "grade", "location_description", "comment") # need to add foreignkey to cave part
# extra = 1
2023-01-30 16:18:19 +00:00
class PersonLogEntryInline(admin.TabularInline):
model = PersonLogEntry
raw_id_fields = ("personexpedition",)
2011-07-11 02:10:22 +01:00
extra = 1
2011-07-11 02:10:22 +01:00
class LogbookEntryAdmin(TroggleModelAdmin):
prepopulated_fields = {"slug": ("title",)}
search_fields = ("title", "expedition__year")
date_heirarchy = "date"
2023-03-17 20:01:52 +00:00
# inlines = (PersonLogEntryInline, QMsFoundInline)
2011-07-11 02:10:22 +01:00
class Media:
css = {"all": ("css/troggleadmin.css",)} # this does not exist
actions = ("export_logbook_entries_as_html", "export_logbook_entries_as_txt")
def export_logbook_entries_as_html(self, modeladmin, request, queryset):
response = downloadLogbook(request=request, queryset=queryset, extension="html") # fails, no queryset
2011-07-11 02:10:22 +01:00
return response
def export_logbook_entries_as_txt(self, modeladmin, request, queryset):
response = downloadLogbook(request=request, queryset=queryset, extension="txt") # fails, no queryset
2011-07-11 02:10:22 +01:00
return response
2011-07-11 02:10:22 +01:00
class PersonExpeditionInline(admin.TabularInline):
model = PersonExpedition
extra = 1
2011-07-11 02:10:22 +01:00
class PersonAdmin(TroggleModelAdmin):
search_fields = ("first_name", "last_name")
2011-07-11 02:10:22 +01:00
inlines = (PersonExpeditionInline,)
2011-07-11 02:10:22 +01:00
class QMAdmin(TroggleModelAdmin):
2023-03-17 20:01:52 +00:00
search_fields = ("number", "expoyear")
list_display = ("__str__", "grade")
list_display_links = ("__str__",)
2023-03-17 20:01:52 +00:00
# list_editable = ("comment", "page_ref", "grade")
# list_per_page = 20
# raw_id_fields = ("found_by", "ticked_off_by")
2011-07-11 02:10:22 +01:00
2011-07-11 02:10:22 +01:00
class PersonExpeditionAdmin(TroggleModelAdmin):
search_fields = ("person__first_name", "expedition__year")
2011-07-11 02:10:22 +01:00
2011-07-11 02:10:22 +01:00
class CaveAdmin(TroggleModelAdmin):
search_fields = ("official_name", "kataster_number", "unofficial_number")
2011-07-11 02:10:22 +01:00
extra = 4
2011-07-11 02:10:22 +01:00
class EntranceAdmin(TroggleModelAdmin):
search_fields = ("caveandentrance__cave__kataster_number",)
2011-07-11 02:10:22 +01:00
class SurvexStationAdmin(TroggleModelAdmin):
search_fields = ("name",)
class SurvexFileAdmin(TroggleModelAdmin):
search_fields = ("path",)
class SurvexDirectoryAdmin(TroggleModelAdmin):
search_fields = (
"path",
"survexdirectory",
)
class DrawingFileAdmin(TroggleModelAdmin):
search_fields = ("dwgname",)
class WalletAdmin(TroggleModelAdmin):
search_fields = ("fpath",)
2011-07-11 02:10:22 +01:00
admin.site.register(Cave, CaveAdmin)
admin.site.register(Area)
admin.site.register(CaveAndEntrance)
admin.site.register(Entrance, EntranceAdmin)
2023-07-08 23:43:57 +01:00
admin.site.register(CaveSlug)
2019-06-26 20:57:24 +01:00
admin.site.register(SurvexBlock, SurvexBlockAdmin)
admin.site.register(DrawingFile, DrawingFileAdmin)
2011-07-11 02:10:22 +01:00
admin.site.register(Expedition)
admin.site.register(Person, PersonAdmin)
2019-06-26 20:57:24 +01:00
admin.site.register(SurvexPersonRole)
admin.site.register(SurvexDirectory, SurvexDirectoryAdmin)
admin.site.register(SurvexFile, SurvexFileAdmin)
admin.site.register(SurvexStation, SurvexStationAdmin)
admin.site.register(PersonExpedition, PersonExpeditionAdmin)
2011-07-11 02:10:22 +01:00
admin.site.register(LogbookEntry, LogbookEntryAdmin)
admin.site.register(QM, QMAdmin)
admin.site.register(Wallet, WalletAdmin)
2020-06-24 00:18:01 +01:00
admin.site.register(SingleScan)
admin.site.register(DataIssue)
2011-07-11 02:10:22 +01:00
def export_as_json(modeladmin, request, queryset):
response = HttpResponse(content_type="text/json")
response["Content-Disposition"] = "attachment; filename=troggle_output.json"
2011-07-11 02:10:22 +01:00
serializers.serialize("json", queryset, stream=response)
return response
2011-07-11 02:10:22 +01:00
def export_as_xml(modeladmin, request, queryset):
response = HttpResponse(content_type="text/xml")
response["Content-Disposition"] = "attachment; filename=troggle_output.xml"
2011-07-11 02:10:22 +01:00
serializers.serialize("xml", queryset, stream=response)
return response
admin.site.add_action(export_as_xml)
admin.site.add_action(export_as_json)