mirror of https://github.com/snachodog/mybuddy.git
140 lines
4.6 KiB
Python
140 lines
4.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
from django.contrib import admin
|
|
from django.conf import settings
|
|
|
|
from import_export import fields, resources
|
|
from import_export.admin import ImportExportMixin, ExportActionMixin
|
|
|
|
from core import models
|
|
|
|
|
|
class ImportExportResourceBase(resources.ModelResource):
|
|
id = fields.Field(attribute='id')
|
|
child = fields.Field(attribute='child_id', column_name='child_id')
|
|
child_first_name = fields.Field(
|
|
attribute='child__first_name', readonly=True)
|
|
child_last_name = fields.Field(attribute='child__last_name', readonly=True)
|
|
|
|
class Meta:
|
|
clean_model_instances = True
|
|
exclude = ('duration',)
|
|
|
|
|
|
class ChildImportExportResource(resources.ModelResource):
|
|
class Meta:
|
|
model = models.Child
|
|
exclude = ('picture', 'slug')
|
|
|
|
|
|
@admin.register(models.Child)
|
|
class ChildAdmin(ImportExportMixin, ExportActionMixin, admin.ModelAdmin):
|
|
list_display = ('first_name', 'last_name', 'birth_date', 'slug')
|
|
list_filter = ('last_name',)
|
|
search_fields = ('first_name', 'last_name', 'birth_date')
|
|
fields = ['first_name', 'last_name', 'birth_date']
|
|
if settings.BABY_BUDDY['ALLOW_UPLOADS']:
|
|
fields.append('picture')
|
|
resource_class = ChildImportExportResource
|
|
|
|
|
|
class DiaperChangeImportExportResource(ImportExportResourceBase):
|
|
class Meta:
|
|
model = models.DiaperChange
|
|
|
|
|
|
@admin.register(models.DiaperChange)
|
|
class DiaperChangeAdmin(ImportExportMixin, ExportActionMixin,
|
|
admin.ModelAdmin):
|
|
list_display = ('child', 'time', 'wet', 'solid', 'color')
|
|
list_filter = ('child', 'wet', 'solid', 'color')
|
|
search_fields = ('child__first_name', 'child__last_name',)
|
|
resource_class = DiaperChangeImportExportResource
|
|
|
|
|
|
class FeedingImportExportResource(ImportExportResourceBase):
|
|
class Meta:
|
|
model = models.Feeding
|
|
|
|
|
|
@admin.register(models.Feeding)
|
|
class FeedingAdmin(ImportExportMixin, ExportActionMixin, admin.ModelAdmin):
|
|
list_display = ('start', 'end', 'duration', 'child', 'type', 'method',
|
|
'amount')
|
|
list_filter = ('child', 'type', 'method',)
|
|
search_fields = ('child__first_name', 'child__last_name', 'type',
|
|
'method',)
|
|
resource_class = FeedingImportExportResource
|
|
|
|
|
|
class NoteImportExportResource(ImportExportResourceBase):
|
|
class Meta:
|
|
model = models.Note
|
|
|
|
|
|
@admin.register(models.Note)
|
|
class NoteAdmin(ImportExportMixin, ExportActionMixin, admin.ModelAdmin):
|
|
list_display = ('time', 'child', 'note',)
|
|
list_filter = ('child',)
|
|
search_fields = ('child__last_name',)
|
|
resource_class = NoteImportExportResource
|
|
|
|
|
|
class SleepImportExportResource(ImportExportResourceBase):
|
|
class Meta:
|
|
model = models.Sleep
|
|
|
|
|
|
@admin.register(models.Sleep)
|
|
class SleepAdmin(ImportExportMixin, ExportActionMixin, admin.ModelAdmin):
|
|
list_display = ('start', 'end', 'duration', 'child', 'nap')
|
|
list_filter = ('child',)
|
|
search_fields = ('child__first_name', 'child__last_name',)
|
|
resource_class = SleepImportExportResource
|
|
|
|
|
|
class TemperatureImportExportResource(ImportExportResourceBase):
|
|
class Meta:
|
|
model = models.Temperature
|
|
|
|
|
|
@admin.register(models.Temperature)
|
|
class TemperatureAdmin(ImportExportMixin, ExportActionMixin, admin.ModelAdmin):
|
|
list_display = ('child', 'temperature', 'time',)
|
|
list_filter = ('child',)
|
|
search_fields = ('child__first_name', 'child__last_name', 'temperature',)
|
|
resource_class = TemperatureImportExportResource
|
|
|
|
|
|
@admin.register(models.Timer)
|
|
class TimerAdmin(admin.ModelAdmin):
|
|
list_display = ('name', 'child', 'start', 'end', 'duration', 'active',
|
|
'user')
|
|
list_filter = ('child', 'active', 'user')
|
|
search_fields = ('child__first_name', 'child__last_name', 'name', 'user')
|
|
|
|
|
|
class TummyTimeImportExportResource(ImportExportResourceBase):
|
|
class Meta:
|
|
model = models.TummyTime
|
|
|
|
|
|
@admin.register(models.TummyTime)
|
|
class TummyTimeAdmin(ImportExportMixin, ExportActionMixin, admin.ModelAdmin):
|
|
list_display = ('start', 'end', 'duration', 'child', 'milestone',)
|
|
list_filter = ('child',)
|
|
search_fields = ('child__first_name', 'child__last_name', 'milestone',)
|
|
resource_class = TummyTimeImportExportResource
|
|
|
|
|
|
class WeightImportExportResource(ImportExportResourceBase):
|
|
class Meta:
|
|
model = models.Weight
|
|
|
|
|
|
@admin.register(models.Weight)
|
|
class WeightAdmin(ImportExportMixin, ExportActionMixin, admin.ModelAdmin):
|
|
list_display = ('child', 'weight', 'date',)
|
|
list_filter = ('child',)
|
|
search_fields = ('child__first_name', 'child__last_name', 'weight',)
|
|
resource_class = WeightImportExportResource
|