| 16 |
16 |
|
| 17 |
17 |
from django.contrib import admin
|
| 18 |
18 |
from django.contrib.auth import get_user_model
|
|
19 |
from django.shortcuts import redirect
|
|
20 |
from django.urls import reverse
|
|
21 |
from django.urls.exceptions import NoReverseMatch
|
| 19 |
22 |
from django.utils.translation import ugettext_lazy as _
|
|
23 |
from django.views.decorators.cache import never_cache
|
| 20 |
24 |
|
| 21 |
25 |
from . import models
|
| 22 |
26 |
|
| 23 |
27 |
|
|
28 |
@never_cache
|
|
29 |
def login(request, *args, **kwargs):
|
|
30 |
try:
|
|
31 |
auth_login_url = reverse('auth_login')
|
|
32 |
except NoReverseMatch:
|
|
33 |
return admin.site.orig_login(request, *args, **kwargs)
|
|
34 |
auth_login_url += '?%s' % request.GET.urlencode()
|
|
35 |
return redirect(auth_login_url)
|
|
36 |
|
|
37 |
|
|
38 |
@never_cache
|
|
39 |
def logout(request, *args, **kwargs):
|
|
40 |
try:
|
|
41 |
return redirect(reverse('auth_logout'))
|
|
42 |
except NoReverseMatch:
|
|
43 |
return admin.site.orig_logout(request, *args, **kwargs)
|
|
44 |
|
|
45 |
|
|
46 |
if admin.site.login != login:
|
|
47 |
admin.site.orig_login = admin.site.login
|
|
48 |
admin.site.login = login
|
|
49 |
|
|
50 |
if admin.site.logout != logout:
|
|
51 |
admin.site.orig_logout = admin.site.logout
|
|
52 |
admin.site.logout = logout
|
|
53 |
|
|
54 |
|
| 24 |
55 |
class UserDocumentAdmin(admin.ModelAdmin):
|
| 25 |
56 |
list_display = ['user', 'filename', 'thumbnail', 'created', 'origin']
|
| 26 |
57 |
fields = ['id', 'user', 'filename', 'thumbnail', 'created', 'origin']
|
| 27 |
|
-
|