1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
from django.contrib import admin
from utils.widgets import LGEntryForm, OLAdminBase
from .models import CheckIn, Daily, Weather
class WeatherAdmin(OLAdminBase):
pass
class CheckInAdmin(OLAdminBase):
list_display = ('date', 'location')
pass
class DailyAdmin(admin.ModelAdmin):
form = LGEntryForm
list_display = ('date', 'user', 'location')
list_filter = (
'date',
('location', admin.RelatedOnlyFieldListFilter),
)
fieldsets = (
(None, {
'fields': (
'user',
'body_markdown',
'weather_human',
)
}),
('Details', {
'fields': (
'location',
'weather',
'date'
),
'classes': ('collapse',),
}),
)
def get_form(self, request, obj=None, **kwargs):
form = super(DailyAdmin, self).get_form(request, **kwargs)
form.current_user = request.user
return form
def get_queryset(self, request):
qs = super(DailyAdmin, self).get_queryset(request)
if request.user.is_superuser:
return qs
return qs.filter(user=request.user)
admin.site.register(CheckIn, CheckInAdmin)
admin.site.register(Weather, WeatherAdmin)
admin.site.register(Daily, DailyAdmin)
|
