forked from beba/foerderbarometer
125 lines
4.3 KiB
Python
125 lines
4.3 KiB
Python
import csv
|
|
|
|
from django.contrib import admin
|
|
from django.http import HttpResponse
|
|
|
|
from .models import Account, Project, HonoraryCertificate, Library, IFG, Travel,\
|
|
Email, BusinessCard, List, Literature
|
|
|
|
|
|
def export_as_csv(self, request, queryset):
|
|
|
|
meta = self.model._meta
|
|
field_names = [field.name for field in meta.fields]
|
|
|
|
response = HttpResponse(content_type='text/csv')
|
|
response['Content-Disposition'] = 'attachment; filename={}.csv'.format(meta)
|
|
writer = csv.writer(response)
|
|
|
|
writer.writerow(field_names)
|
|
for obj in queryset:
|
|
row = writer.writerow([getattr(obj, field) for field in field_names])
|
|
|
|
return response
|
|
|
|
export_as_csv.short_description = "Export Selected"
|
|
|
|
admin.site.add_action(export_as_csv)
|
|
|
|
@admin.register(Project)
|
|
class ProjectAdmin(admin.ModelAdmin):
|
|
save_as = True
|
|
search_fields = ('name', 'pid','finance_id', 'realname', 'start', 'end', 'participants_estimated', 'participants_real', 'cost', 'status')
|
|
list_display = ('name', 'pid','finance_id', 'realname', 'start', 'end', 'participants_estimated', 'participants_real', 'cost', 'status')
|
|
# action = ['export_as_csv']
|
|
date_hierarchy = 'start'
|
|
|
|
|
|
@admin.register(BusinessCard)
|
|
class BusinessCardAdmin(admin.ModelAdmin):
|
|
save_as = True
|
|
search_fields = (('realname', 'service_id'), 'granted', 'granted_date', 'project')
|
|
list_display = ('realname', 'service_id', 'granted', 'granted_date', 'project')
|
|
list_display_links = ('realname', 'service_id')
|
|
# action = ['export_as_csv']
|
|
date_hierarchy = 'granted_date'
|
|
class Media:
|
|
js = ('dropdown/js/base.js',)
|
|
|
|
@admin.register(Literature)
|
|
class LiteratureAdmin(admin.ModelAdmin):
|
|
save_as = True
|
|
search_fields = (('realname', 'service_id'), 'granted', 'granted_date')
|
|
list_display = ('realname', 'service_id', 'granted', 'granted_date')
|
|
list_display_links = ('realname', 'service_id')
|
|
date_hierarchy = 'granted_date'
|
|
class Media:
|
|
js = ('dropdown/js/base.js',)
|
|
|
|
|
|
@admin.register(Account)
|
|
class AccountAdmin(admin.ModelAdmin):
|
|
save_as = True
|
|
|
|
|
|
@admin.register(HonoraryCertificate)
|
|
class HonoraryCertificateAdmin(admin.ModelAdmin):
|
|
save_as = True
|
|
search_fields = ('realname', 'granted', 'granted_date', 'project')
|
|
list_display = ('realname', 'granted', 'granted_date', 'project')
|
|
list_display_links = ('realname', 'project')
|
|
date_hierarchy = 'granted_date'
|
|
|
|
@admin.register(Library)
|
|
class LibraryAdmin(admin.ModelAdmin):
|
|
save_as = True
|
|
search_fields = (('realname', 'service_id'), 'granted', 'granted_date')
|
|
list_display = ('realname', 'service_id', 'granted', 'granted_date')
|
|
list_display_links = ('realname', 'service_id')
|
|
date_hierarchy = 'granted_date'
|
|
|
|
@admin.register(IFG)
|
|
class IFGAdmin(admin.ModelAdmin):
|
|
save_as = True
|
|
search_fields = (('realname', 'service_id'), 'granted', 'granted_date')
|
|
list_display = ('realname', 'service_id', 'granted', 'granted_date')
|
|
list_display_links = ('realname', 'service_id')
|
|
date_hierarchy = 'granted_date'
|
|
|
|
@admin.register(Travel)
|
|
class TravelAdmin(admin.ModelAdmin):
|
|
save_as = True
|
|
search_fields = ('realname', 'granted', 'granted_date', 'project')
|
|
list_display = ('realname', 'granted', 'granted_date', 'project')
|
|
list_display_links = ('realname', 'project')
|
|
date_hierarchy = 'granted_date'
|
|
|
|
@admin.register(Email)
|
|
class EmailAdmin(admin.ModelAdmin):
|
|
save_as = True
|
|
search_fields = (('realname', 'service_id'), 'granted', 'granted_date')
|
|
list_display = ('realname', 'service_id', 'granted', 'granted_date')
|
|
list_display_links = ('realname', 'service_id')
|
|
date_hierarchy = 'granted_date'
|
|
|
|
|
|
@admin.register(List)
|
|
class ListAdmin(admin.ModelAdmin):
|
|
save_as = True
|
|
search_fields = (('realname', 'service_id'), 'granted', 'granted_date')
|
|
list_display = ('realname', 'service_id', 'granted', 'granted_date')
|
|
list_display_links = ('realname', 'service_id')
|
|
date_hierarchy = 'granted_date'
|
|
|
|
# commented out because of the individual registering to control displays in admin panel
|
|
|
|
#admin.site.register([
|
|
# Account,
|
|
# HonoraryCertificate,
|
|
# Library,
|
|
# IFG,
|
|
# Travel,
|
|
# Email,
|
|
# List,
|
|
# ])
|