Blame view

src/core/pyros_django/common/admin.py 7.23 KB
eefbbbd2   Etienne Pallier   Model splitting g...
1
2

# Django imports
5ce2836f   Alexis Koralewski   update models (ad...
3
4
5
from django import forms
from django.contrib import admin
from django.contrib.auth.models import User
5ce2836f   Alexis Koralewski   update models (ad...
6
7
8
# EP
from django.conf import settings

eefbbbd2   Etienne Pallier   Model splitting g...
9
10
# Project imports
from user_manager.models import Country, Institute, PyrosUser, UserLevel, ScientificProgram
5ce2836f   Alexis Koralewski   update models (ad...
11
from common.models import *
bfe3fba1   Etienne Pallier   started common/mo...
12
from devices.models import Detector, Filter, AgentDeviceStatus, FilterWheel, Telescope, PlcDevice, PlcDeviceStatus
eefbbbd2   Etienne Pallier   Model splitting g...
13
from routine_manager.models import Image, Schedule, Sequence, Album, Plan, ScheduleHasSequences #, StrategyObs, NrtAnalysis
3f6876e5   Alexis Koralewski   fix issues with t...
14
#from routine_manager.models import Image, StrategyObs, Schedule, Request, Alert, Sequence, Album, Plan, NrtAnalysis, ScheduleHasSequences
bfe3fba1   Etienne Pallier   started common/mo...
15

5ce2836f   Alexis Koralewski   update models (ad...
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

# EP added
class ReadOnlyModelAdmin(admin.ModelAdmin):
    """ModelAdmin class that prevents modifications through the admin.
    The changelist and the detail view work, but a 403 is returned
    if one actually tries to edit an object.
    Source: https://gist.github.com/aaugustin/1388243
    """

    actions = None

    def get_readonly_fields(self, request, obj=None):
        return self.fields or [f.name for f in self.model._meta.fields]

    def has_add_permission(self, request):
        return False

    # Allow viewing objects but not actually changing them
    def has_change_permission(self, request, obj=None):
        if request.method not in ('GET', 'HEAD'):
            return False
        return super(ReadOnlyModelAdmin, self).has_change_permission(request, obj)

    def has_delete_permission(self, request, obj=None):
        return False


# EP added

# Edit mode
# DEBUG = False
# View only mode
# DEBUG = True

''' Uncomment for production '''

# if settings.DEBUG:
# class PyrosModelAdmin(ReadOnlyModelAdmin):
#     pass
# else:
#     class PyrosModelAdmin(admin.ModelAdmin):
#         pass


class PyrosModelAdmin(admin.ModelAdmin):
    pass

# Many To Many interface adapter


class PyrosUserAndSPInline(admin.TabularInline):
    #model = ScientificProgram.pyros_users.through
    pass


class SequenceAndScheduleInline(admin.TabularInline):
    model = Schedule.sequences.through

class PyrosUserAndUserLevelInline(admin.TabularInline):
    # add admin representation for m2m relation between PyrosUser and UserLevel
    model = UserLevel.pyros_users.through

class ScheduleAdmin(admin.ModelAdmin):
    inlines = [
        SequenceAndScheduleInline,
    ]
    exclude = ('sequences',)


# One To Many interface adapters

class SequenceInline(admin.TabularInline):
    model = Sequence
    readonly_fields = ("name",)
    fields = ("name",)
    show_change_link = True


3f6876e5   Alexis Koralewski   fix issues with t...
94
95
96
97
98
# class RequestInline(admin.TabularInline):
#     model = Request
#     readonly_fields = ("name",)
#     fields = ("name",)
#     show_change_link = True
5ce2836f   Alexis Koralewski   update models (ad...
99
100
101
102
103
104
105
106
107
108
109


class AlbumInline(admin.TabularInline):
    model = Album
    readonly_fields = ("name",)
    fields = ("name",)
    show_change_link = True


class PlanInline(admin.TabularInline):
    model = Plan
3b81a22b   Alexis Koralewski   Rework on Request...
110
111
    #readonly_fields = ("name",)
    #fields = ("name",)
5ce2836f   Alexis Koralewski   update models (ad...
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
    show_change_link = True


class ImageInline(admin.TabularInline):
    model = Image
    readonly_fields = ("name",)
    fields = ("name",)
    show_change_link = True


class DetectorInline(admin.TabularInline):
    model = Detector
    readonly_fields = ("device_name",)
    fields = ("device_name",)
    show_change_link = True


class PyrosUserInline(admin.TabularInline):
    model = PyrosUser
    readonly_fields = ("user_username",)
    fields = ("user_username",)
    show_change_link = True


class FilterInline(admin.TabularInline):
    model = Filter
    readonly_fields = ("device_name",)
    fields = ("device_name",)
    show_change_link = True


3f6876e5   Alexis Koralewski   fix issues with t...
143
144
145
146
147
# class AlertInline(admin.TabularInline):
#     model = Alert
#     readonly_fields = ("request_name",)
#     fields = ("request_name",)
#     show_change_link = True
5ce2836f   Alexis Koralewski   update models (ad...
148
149
150
151


# Admin model classes

3f6876e5   Alexis Koralewski   fix issues with t...
152
153
# class RequestAdmin(PyrosModelAdmin):
#    pass
3b81a22b   Alexis Koralewski   Rework on Request...
154
155
156
    # inlines = [
    #     SequenceInline,
    # ]
5ce2836f   Alexis Koralewski   update models (ad...
157
158
159
160
161
162
163
164
165
166
167
168
169
170


class SequenceAdmin(PyrosModelAdmin):
    inlines = [
        AlbumInline,
        SequenceAndScheduleInline,  # for M2M interface
    ]


class PyrosUserAdmin(PyrosModelAdmin):
    list_display = ("user_username","is_active","laboratory")
    list_filter = ("is_active",)
    list_editable = ("is_active",)
    inlines = [
3f6876e5   Alexis Koralewski   fix issues with t...
171
        #RequestInline,
5ce2836f   Alexis Koralewski   update models (ad...
172
173
174
175
176
        # A user has many SPs
#        PyrosUserAndSPInline,  # for M2M interface   
    ]


eefbbbd2   Etienne Pallier   Model splitting g...
177
'''
5ce2836f   Alexis Koralewski   update models (ad...
178
179
class StrategyObsAdmin(PyrosModelAdmin):
    inlines = [
3f6876e5   Alexis Koralewski   fix issues with t...
180
        #AlertInline,
5ce2836f   Alexis Koralewski   update models (ad...
181
    ]
eefbbbd2   Etienne Pallier   Model splitting g...
182
'''
5ce2836f   Alexis Koralewski   update models (ad...
183
184
185
186


class ScientificProgramAdmin(PyrosModelAdmin):
    inlines = [
3f6876e5   Alexis Koralewski   fix issues with t...
187
        #RequestInline,
5ce2836f   Alexis Koralewski   update models (ad...
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
        # A SP has many users:
    #    PyrosUserAndSPInline,  # for M2M interface
    ]
    exclude = ('pyros_users',)  # for M2M interface


class CountryAdmin(PyrosModelAdmin):
    inlines = [
        PyrosUserInline,
    ]


class UserLevelAdmin(PyrosModelAdmin):
    inlines = [
        #PyrosUserInline,
        PyrosUserAndUserLevelInline,
    ]
    list_display = ("name","priority",)
    # we need to exclude pyros_users which represents the m2m relation between UserLevel and PyrosUser
    exclude = ("pyros_users",)


class FilterAdmin(PyrosModelAdmin):
3b81a22b   Alexis Koralewski   Rework on Request...
211
212
213
214
    # inlines = [
    #     PlanInline,
    # ]
    pass
5ce2836f   Alexis Koralewski   update models (ad...
215
216
217
218
219
220
221
222


class FilterWheelAdmin(PyrosModelAdmin):
    inlines = [
        FilterInline,
    ]


eefbbbd2   Etienne Pallier   Model splitting g...
223
'''
5ce2836f   Alexis Koralewski   update models (ad...
224
225
226
227
class NrtAnalysisAdmin(PyrosModelAdmin):
    inlines = [
        ImageInline,
    ]
eefbbbd2   Etienne Pallier   Model splitting g...
228
'''
5ce2836f   Alexis Koralewski   update models (ad...
229
230
231


class DetectorAdmin(PyrosModelAdmin):
3b81a22b   Alexis Koralewski   Rework on Request...
232
233
234
235
    pass
    # inlines = [
    #     AlbumInline,
    # ]
5ce2836f   Alexis Koralewski   update models (ad...
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266


class TelescopeAdmin(PyrosModelAdmin):
    inlines = [
        DetectorInline,
    ]


class PlanAdmin(PyrosModelAdmin):
    inlines = [
        ImageInline,
    ]


# class AlbumAdmin(admin.ModelAdmin):
class AlbumAdmin(PyrosModelAdmin):
    inlines = [
        PlanInline,
    ]


# Link the models to the admin interface

# (EP added 10/7/19)
admin.site.register(AgentCmd)
admin.site.register(AgentLogs)
admin.site.register(AgentSurvey)
admin.site.register(AgentDeviceStatus)


admin.site.register(Album, AlbumAdmin)
3f6876e5   Alexis Koralewski   fix issues with t...
267
#admin.site.register(Alert)
5ce2836f   Alexis Koralewski   update models (ad...
268
269
270
271
272
273
admin.site.register(Country, CountryAdmin)
admin.site.register(Detector, DetectorAdmin)
admin.site.register(Filter, FilterAdmin)
admin.site.register(FilterWheel, FilterWheelAdmin)
admin.site.register(Image)
admin.site.register(Log)
eefbbbd2   Etienne Pallier   Model splitting g...
274
#admin.site.register(NrtAnalysis, NrtAnalysisAdmin)
5ce2836f   Alexis Koralewski   update models (ad...
275
admin.site.register(Plan, PlanAdmin)
3f6876e5   Alexis Koralewski   fix issues with t...
276
#admin.site.register(Request, RequestAdmin)
5ce2836f   Alexis Koralewski   update models (ad...
277
278
279
280
281
282
admin.site.register(Schedule, ScheduleAdmin)
admin.site.register(ScheduleHasSequences)
admin.site.register(ScientificProgram, ScientificProgramAdmin)
admin.site.register(Sequence, SequenceAdmin)
admin.site.register(SiteWatch)
admin.site.register(SiteWatchHistory)
eefbbbd2   Etienne Pallier   Model splitting g...
283
#admin.site.register(StrategyObs, StrategyObsAdmin)
b512920a   Etienne Pallier   models cleanup
284
##admin.site.register(TaskId)
5ce2836f   Alexis Koralewski   update models (ad...
285
286
287
288
289
290
291
292
293
294
admin.site.register(Telescope, TelescopeAdmin)
admin.site.register(PyrosUser, PyrosUserAdmin)
admin.site.register(UserLevel, UserLevelAdmin)
admin.site.register(Version)
admin.site.register(WeatherWatch)
admin.site.register(WeatherWatchHistory)
admin.site.register(PlcDeviceStatus)
admin.site.register(PlcDevice)
admin.site.register(Config)
admin.site.register(Institute)