File tree Expand file tree Collapse file tree 2 files changed +7
-10
lines changed Expand file tree Collapse file tree 2 files changed +7
-10
lines changed Original file line number Diff line number Diff line change 8
8
from django .apps import apps
9
9
from django .conf import settings
10
10
from django .contrib import admin
11
+ from django .contrib .auth import get_user_model
11
12
from django .db import models , router
12
13
from django .db .models import Q
13
14
from django .db .models .fields .proxy import OrderWrt
25
26
post_create_historical_record ,
26
27
)
27
28
29
+ User = get_user_model ()
28
30
registered_models = {}
29
31
30
32
@@ -216,9 +218,7 @@ def copy_fields(self, model):
216
218
def get_extra_fields (self , model , fields ):
217
219
"""Return dict of extra fields added to the historical record model"""
218
220
219
- user_model = self .user_model or getattr (
220
- settings , 'AUTH_USER_MODEL' , 'auth.User'
221
- )
221
+ user_model = self .user_model or User
222
222
223
223
def revert_url (self ):
224
224
"""URL for this change in the default admin site."""
Original file line number Diff line number Diff line change 1
1
from datetime import datetime , timedelta
2
2
from operator import attrgetter
3
3
4
+ from django .contrib .auth import get_user_model
4
5
from django .db import IntegrityError
5
6
from django .test import TestCase , skipUnlessDBFeature
6
7
7
- try :
8
- from django .contrib .auth import get_user_model
9
- except ImportError :
10
- from django .contrib .auth .models import User
11
- else :
12
- User = get_user_model ()
13
-
14
8
from ..models import Document , Poll
15
9
16
10
11
+ User = get_user_model ()
12
+
13
+
17
14
class AsOfTest (TestCase ):
18
15
model = Document
19
16
You can’t perform that action at this time.
0 commit comments