Remove remaining direct references to the User model. Closes GitLab issue #225.

This commit is contained in:
Roberto Rosario
2016-01-28 05:55:15 -04:00
parent 27d8644438
commit 2ebb84f18f
15 changed files with 36 additions and 34 deletions

View File

@@ -3,7 +3,7 @@ from __future__ import unicode_literals
import logging
from django.apps import apps
from django.contrib.auth.models import User
from django.contrib.auth import get_user_model
from django.db import OperationalError
from mayan.celery import app
@@ -100,7 +100,7 @@ def task_upload_new_document(self, document_type_id, shared_uploaded_file_id, de
pk=shared_uploaded_file_id
)
if user_id:
user = User.objects.get(pk=user_id)
user = get_user_model().objects.get(pk=user_id)
else:
user = None
@@ -150,7 +150,7 @@ def task_upload_new_version(self, document_id, shared_uploaded_file_id, user_id,
pk=shared_uploaded_file_id
)
if user_id:
user = User.objects.get(pk=user_id)
user = get_user_model().objects.get(pk=user_id)
else:
user = None