File tree Expand file tree Collapse file tree 1 file changed +20
-11
lines changed Expand file tree Collapse file tree 1 file changed +20
-11
lines changed Original file line number Diff line number Diff line change 2
2
import os
3
3
4
4
from django .conf import settings
5
+ from django .core .exceptions import ImproperlyConfigured
5
6
from django .db import models
6
7
from django .utils .translation import gettext_lazy as _
7
8
@@ -40,14 +41,22 @@ def __str__(self):
40
41
return self .key
41
42
42
43
43
- class TokenProxy (Token ):
44
- """
45
- Proxy mapping pk to user pk for use in admin.
46
- """
47
- @property
48
- def pk (self ):
49
- return self .user .pk
50
-
51
- class Meta :
52
- proxy = True
53
- verbose_name = "token"
44
+ if 'rest_framework.authtoken' in settings .INSTALLED_APPS :
45
+ class TokenProxy (Token ):
46
+ """
47
+ Proxy mapping pk to user pk for use in admin.
48
+ """
49
+ @property
50
+ def pk (self ):
51
+ return self .user .pk
52
+
53
+ class Meta :
54
+ proxy = True
55
+ verbose_name = "token"
56
+ else :
57
+ class TokenProxy :
58
+ def __init__ (self , * args , ** kwargs ):
59
+ raise ImproperlyConfigured (
60
+ '"rest_framework.authtoken" must be in your '
61
+ 'settings.INSTALLED_APPS to use the Token or TokenProxy '
62
+ 'model.' )
You can’t perform that action at this time.
0 commit comments