Change django.db.models.loading.get_model (deprecated in django 1.7) to django.apps.apps.get_model

taiga/auth/backends.py

@@ -36,7 +36,7 @@ import base64
 import re
 
 from django.core import signing
-from django.db.models import get_model
+from django.apps import apps
 from rest_framework.authentication import BaseAuthentication
 from taiga.base import exceptions as exc
 
@@ -85,7 +85,7 @@ def get_user_for_token(token):
     except signing.BadSignature:
         raise exc.NotAuthenticated("Invalid token")
 
-    model_cls = get_model("users", "User")
+    model_cls = apps.get_model("users", "User")
 
     try:
         user = model_cls.objects.get(pk=data["user_id"])

taiga/auth/services.py

@@ -23,7 +23,7 @@ should be contained in a class". Because of that, it
 not uses clasess and uses simple functions.
 """
 
-from django.db.models.loading import get_model
+from django.apps import apps
 from django.db.models import Q
 from django.db import transaction as tx
 from django.db import IntegrityError
@@ -68,7 +68,7 @@ def is_user_already_registred(*, username:str, email:str, github_id:int=None) ->
     Checks if a specified user is already registred.
     """
 
-    user_model = get_model("users", "User")
+    user_model = apps.get_model("users", "User")
 
     or_expr = Q(username=username) | Q(email=email)
     if github_id:
@@ -86,7 +86,7 @@ def get_membership_by_token(token:str):
     If not matches with any membership NotFound exception
     is raised.
     """
-    membership_model = get_model("projects", "Membership")
+    membership_model = apps.get_model("projects", "Membership")
     qs = membership_model.objects.filter(token=token)
     if len(qs) == 0:
         raise exc.NotFound("Token not matches any valid invitation.")
@@ -108,7 +108,7 @@ def public_register(username:str, password:str, email:str, full_name:str):
     if is_user_already_registred(username=username, email=email):
         raise exc.IntegrityError("User is already registred.")
 
-    user_model = get_model("users", "User")
+    user_model = apps.get_model("users", "User")
     user = user_model(username=username,
                       email=email,
                       full_name=full_name)
@@ -149,7 +149,7 @@ def private_register_for_new_user(token:str, username:str, email:str,
     if is_user_already_registred(username=username, email=email):
         raise exc.WrongArguments(_("Username or Email is already in use."))
 
-    user_model = get_model("users", "User")
+    user_model = apps.get_model("users", "User")
     user = user_model(username=username,
                       email=email,
                       full_name=full_name)
@@ -177,7 +177,7 @@ def github_register(username:str, email:str, full_name:str, github_id:int, bio:s
 
     :returns: User
     """
-    user_model = get_model("users", "User")
+    user_model = apps.get_model("users", "User")
     user, created = user_model.objects.get_or_create(github_id=github_id,
                                                      defaults={"username": username,
                                                                "email": email,

taiga/base/api/permissions.py

@@ -18,7 +18,7 @@ import abc
 
 from taiga.base.utils import sequence as sq
 from taiga.permissions.service import user_has_perm, is_project_owner
-from django.db.models.loading import get_model
+from django.apps import apps
 
 
 ######################################################################
@@ -181,7 +181,7 @@ class HasProjectParamAndPerm(PermissionComponent):
         super().__init__(*components)
 
     def check_permissions(self, request, view, obj=None):
-        Project = get_model('projects', 'Project')
+        Project = apps.get_model('projects', 'Project')
         project_id = request.QUERY_PARAMS.get("project", None)
         try:
             project = Project.objects.get(pk=project_id)

taiga/projects/history/freeze_impl.py

@@ -15,7 +15,7 @@
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
 from functools import partial
-from django.db.models.loading import get_model
+from django.apps import apps
 from django.contrib.contenttypes.models import ContentType
 from taiga.base.utils.iterators import as_tuple
 from taiga.base.utils.iterators import as_dict
@@ -41,7 +41,7 @@ def _get_generic_values(ids:tuple, *, typename=None, attr:str="name") -> tuple:
 
 @as_dict
 def _get_users_values(ids:set) -> dict:
-    user_model = get_model("users", "User")
+    user_model = apps.get_model("users", "User")
     ids = filter(lambda x: x is not None, ids)
     qs = user_model.objects.filter(pk__in=tuple(ids))
 
@@ -199,7 +199,7 @@ def milestone_freezer(milestone) -> dict:
 
 
 def userstory_freezer(us) -> dict:
-    rp_cls = get_model("userstories", "RolePoints")
+    rp_cls = apps.get_model("userstories", "RolePoints")
     rpqsd = rp_cls.objects.filter(user_story=us)
 
     points = {}

taiga/projects/history/models.py

@@ -17,7 +17,7 @@ import uuid
 from django.utils.translation import ugettext_lazy as _
 from django.utils import timezone
 from django.db import models
-from django.db.models.loading import get_model
+from django.apps import apps
 from django.utils.functional import cached_property
 from django.conf import settings
 from django_pgjson.fields import JsonField
@@ -79,7 +79,7 @@ class HistoryEntry(models.Model):
     @cached_property
     def owner(self):
         pk = self.user["pk"]
-        model = get_model("users", "User")
+        model = apps.get_model("users", "User")
         return model.objects.get(pk=pk)
 
     @cached_property

taiga/projects/history/services.py

@@ -35,7 +35,7 @@ from functools import lru_cache
 from django.conf import settings
 from django.contrib.contenttypes.models import ContentType
 from django.core.paginator import Paginator, InvalidPage
-from django.db.models.loading import get_model
+from django.apps import apps
 from django.db import transaction as tx
 
 from taiga.mdrender.service import render as mdrender
@@ -207,7 +207,7 @@ def _rebuild_snapshot_from_diffs(keysnapshot, partials):
 
 
 def get_last_snapshot_for_key(key:str) -> FrozenObj:
-    entry_model = get_model("history", "HistoryEntry")
+    entry_model = apps.get_model("history", "HistoryEntry")
 
     # Search last snapshot
     qs = (entry_model.objects
@@ -251,7 +251,7 @@ def take_snapshot(obj:object, *, comment:str="", user=None, delete:bool=False):
     new_fobj = freeze_model_instance(obj)
     old_fobj, need_real_snapshot = get_last_snapshot_for_key(key)
 
-    entry_model = get_model("history", "HistoryEntry")
+    entry_model = apps.get_model("history", "HistoryEntry")
     user_id = None if user is None else user.id
     user_name = "" if user is None else user.get_full_name()
 
@@ -306,7 +306,7 @@ def get_history_queryset_by_model_instance(obj:object, types=(HistoryType.change
     Get one page of history for specified object.
     """
     key = make_key_from_model_object(obj)
-    history_entry_model = get_model("history", "HistoryEntry")
+    history_entry_model = apps.get_model("history", "HistoryEntry")
 
     qs = history_entry_model.objects.filter(key=key, type__in=types)
     if not include_hidden:

taiga/projects/models.py

@@ -19,7 +19,7 @@ import itertools
 from django.core.exceptions import ValidationError
 from django.db import models
 from django.db.models import signals
-from django.db.models.loading import get_model
+from django.apps import apps
 from django.conf import settings
 from django.dispatch import receiver
 from django.contrib.auth import get_user_model
@@ -211,8 +211,8 @@ class Project(ProjectDefaults, TaggedMixin, models.Model):
         return user_model.objects.filter(id__in=list(members))
 
     def update_role_points(self, user_stories=None):
-        RolePoints = get_model("userstories", "RolePoints")
-        Role = get_model("users", "Role")
+        RolePoints = apps.get_model("userstories", "RolePoints")
+        Role = apps.get_model("users", "Role")
 
         # Get all available roles on this project
         roles = self.get_roles().filter(computable=True)
@@ -251,7 +251,7 @@ class Project(ProjectDefaults, TaggedMixin, models.Model):
         return dict_sum(*flat_role_dicts)
 
     def _get_points_increment(self, client_requirement, team_requirement):
-        userstory_model = get_model("userstories", "UserStory")
+        userstory_model = apps.get_model("userstories", "UserStory")
         user_stories = userstory_model.objects.none()
         last_milestones = self.milestones.order_by('-estimated_finish')
         last_milestone = last_milestones[0] if last_milestones else None
@@ -742,9 +742,9 @@ def membership_post_delete(sender, instance, using, **kwargs):
 # On membership object is deleted, update watchers of all objects relation.
 @receiver(signals.post_delete, sender=Membership, dispatch_uid='update_watchers_on_membership_post_delete')
 def update_watchers_on_membership_post_delete(sender, instance, using, **kwargs):
-    models = [get_model("userstories", "UserStory"),
-              get_model("tasks", "Task"),
-              get_model("issues", "Issue")]
+    models = [apps.get_model("userstories", "UserStory"),
+              apps.get_model("tasks", "Task"),
+              apps.get_model("issues", "Issue")]
 
     # `user_id` is used beacuse in some momments
     # instance.user can contain pointer to now

taiga/projects/notifications/mixins.py

@@ -18,12 +18,9 @@ from functools import partial
 from operator import is_not
 
 from django.conf import settings
-from django.db.models.loading import get_model
 from django.db import models
 from django.utils.translation import ugettext_lazy as _
 
-from rest_framework import serializers
-from taiga.projects.history.models import HistoryType
 from taiga.projects.notifications import services
 
 
@@ -144,18 +141,3 @@ class WatchedModelMixin(models.Model):
                         self.get_owner(),)
         is_not_none = partial(is_not, None)
         return frozenset(filter(is_not_none, participants))
-
-
-# class WatcherValidationSerializerMixin(object):
-#     def validate_watchers(self, attrs, source):
-#         values =  set(attrs.get(source, []))
-#         if values:
-#             project = None
-#             if "project" in attrs and attrs["project"]:
-#                 project = attrs["project"]
-#             elif self.object:
-#                 project = self.object.project
-#             model_cls = get_model("projects", "Membership")
-#             if len(values) != model_cls.objects.filter(project=project, user__in=values).count():
-#                 raise serializers.ValidationError("Error, some watcher user is not a member of the project")
-#         return attrs

taiga/projects/notifications/services.py

@@ -16,7 +16,7 @@
 
 from functools import partial
 
-from django.db.models.loading import get_model
+from django.apps import apps
 from django.db import IntegrityError
 from django.contrib.contenttypes.models import ContentType
 
@@ -33,7 +33,7 @@ def notify_policy_exists(project, user) -> bool:
     Check if policy exists for specified project
     and user.
     """
-    model_cls = get_model("notifications", "NotifyPolicy")
+    model_cls = apps.get_model("notifications", "NotifyPolicy")
     qs = model_cls.objects.filter(project=project,
                                   user=user)
     return qs.exists()
@@ -43,7 +43,7 @@ def create_notify_policy(project, user, level=NotifyLevel.notwatch):
     """
     Given a project and user, create notification policy for it.
     """
-    model_cls = get_model("notifications", "NotifyPolicy")
+    model_cls = apps.get_model("notifications", "NotifyPolicy")
     try:
         return model_cls.objects.create(project=project,
                                         user=user,
@@ -56,7 +56,7 @@ def create_notify_policy_if_not_exists(project, user, level=NotifyLevel.notwatch
     """
     Given a project and user, create notification policy for it.
     """
-    model_cls = get_model("notifications", "NotifyPolicy")
+    model_cls = apps.get_model("notifications", "NotifyPolicy")
     try:
         result = model_cls.objects.get_or_create(project=project,
                                                user=user,
@@ -70,7 +70,7 @@ def get_notify_policy(project, user):
     """
     Get notification level for specified project and user.
     """
-    model_cls = get_model("notifications", "NotifyPolicy")
+    model_cls = apps.get_model("notifications", "NotifyPolicy")
     instance, _ = model_cls.objects.get_or_create(project=project, user=user,
                                                   defaults={"notify_level": NotifyLevel.notwatch})
     return instance

taiga/projects/references/api.py

@@ -14,7 +14,7 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-from django.db.models.loading import get_model
+from django.apps import apps
 from django.shortcuts import get_object_or_404
 
 from rest_framework.response import Response
@@ -37,7 +37,7 @@ class ResolverViewSet(viewsets.ViewSet):
 
         data = serializer.data
 
-        project_model = get_model("projects", "Project")
+        project_model = apps.get_model("projects", "Project")
         project = get_object_or_404(project_model, slug=data["project"])
 
         self.check_permissions(request, "list", project)

taiga/projects/references/services.py

@@ -14,11 +14,11 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-from django.db.models.loading import get_model
+from django.apps import apps
 
 
 def get_instance_by_ref(project_id, obj_ref):
-    model_cls = get_model("references", "Reference")
+    model_cls = apps.get_model("references", "Reference")
     try:
         instance = model_cls.objects.get(project_id=project_id, ref=obj_ref)
     except model_cls.DoesNotExist:

taiga/projects/userstories/serializers.py

@@ -15,7 +15,7 @@
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
 import json
-from django.db.models import get_model
+from django.apps import apps
 from rest_framework import serializers
 
 from taiga.base.serializers import Serializer, PickleField, NeighborsSerializerMixin
@@ -56,7 +56,7 @@ class UserStorySerializer(serializers.ModelSerializer):
         role_points = obj._related_data.pop("role_points", None)
         super().save_object(obj, **kwargs)
 
-        points_modelcls = get_model("projects", "Points")
+        points_modelcls = apps.get_model("projects", "Points")
 
         if role_points:
             for role_id, points_id in role_points.items():

taiga/projects/votes/services.py

@@ -17,7 +17,7 @@
 
 from django.db.models import F
 from django.db.transaction import atomic
-from django.db.models.loading import get_model
+from django.apps import apps
 from django.contrib.auth import get_user_model
 
 from .models import Votes, Vote
@@ -32,7 +32,7 @@ def add_vote(obj, user):
     :param obj: Any Django model instance.
     :param user: User adding the vote. :class:`~taiga.users.models.User` instance.
     """
-    obj_type = get_model("contenttypes", "ContentType").objects.get_for_model(obj)
+    obj_type = apps.get_model("contenttypes", "ContentType").objects.get_for_model(obj)
     with atomic():
         vote, created = Vote.objects.get_or_create(content_type=obj_type, object_id=obj.id, user=user)
 
@@ -54,7 +54,7 @@ def remove_vote(obj, user):
     :param obj: Any Django model instance.
     :param user: User removing her vote. :class:`~taiga.users.models.User` instance.
     """
-    obj_type = get_model("contenttypes", "ContentType").objects.get_for_model(obj)
+    obj_type = apps.get_model("contenttypes", "ContentType").objects.get_for_model(obj)
     with atomic():
         qs = Vote.objects.filter(content_type=obj_type, object_id=obj.id, user=user)
         if not qs.exists():
@@ -74,7 +74,7 @@ def get_voters(obj):
 
     :return: User queryset object representing the users that voted the object.
     """
-    obj_type = get_model("contenttypes", "ContentType").objects.get_for_model(obj)
+    obj_type = apps.get_model("contenttypes", "ContentType").objects.get_for_model(obj)
     return get_user_model().objects.filter(votes__content_type=obj_type, votes__object_id=obj.id)
 
 
@@ -85,7 +85,7 @@ def get_votes(obj):
 
     :return: Number of votes or `0` if the object has no votes at all.
     """
-    obj_type = get_model("contenttypes", "ContentType").objects.get_for_model(obj)
+    obj_type = apps.get_model("contenttypes", "ContentType").objects.get_for_model(obj)
 
     try:
         return Votes.objects.get(content_type=obj_type, object_id=obj.id).count
@@ -101,7 +101,7 @@ def get_voted(user_or_id, model):
 
     :return: Queryset of objects representing the votes of the user.
     """
-    obj_type = get_model("contenttypes", "ContentType").objects.get_for_model(model)
+    obj_type = apps.get_model("contenttypes", "ContentType").objects.get_for_model(model)
     conditions = ('votes_vote.content_type_id = %s',
                   '%s.id = votes_vote.object_id' % model._meta.db_table,
                   'votes_vote.user_id = %s')

taiga/projects/votes/utils.py

@@ -15,7 +15,7 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-from django.db.models.loading import get_model
+from django.apps import apps
 
 
 def attach_votescount_to_queryset(queryset, as_field="votes_count"):
@@ -33,7 +33,7 @@ def attach_votescount_to_queryset(queryset, as_field="votes_count"):
     :return: Queryset object with the additional `as_field` field.
     """
     model = queryset.model
-    type = get_model("contenttypes", "ContentType").objects.get_for_model(model)
+    type = apps.get_model("contenttypes", "ContentType").objects.get_for_model(model)
     sql = ("SELECT coalesce(votes_votes.count, 0) FROM votes_votes "
            "WHERE votes_votes.content_type_id = {type_id} AND votes_votes.object_id = {tbl}.id")
     sql = sql.format(type_id=type.id, tbl=model._meta.db_table)

taiga/searches/api.py

@@ -14,7 +14,7 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-from django.db.models.loading import get_model
+from django.apps import apps
 
 from rest_framework.response import Response
 from rest_framework import viewsets
@@ -31,7 +31,7 @@ from . import services
 
 class SearchViewSet(viewsets.ViewSet):
     def list(self, request, **kwargs):
-        project_model = get_model("projects", "Project")
+        project_model = apps.get_model("projects", "Project")
 
         text = request.QUERY_PARAMS.get('text', "")
         project_id = request.QUERY_PARAMS.get('project', None)
@@ -55,7 +55,7 @@ class SearchViewSet(viewsets.ViewSet):
         return Response(result)
 
     def _get_project(self, project_id):
-        project_model = get_model("projects", "Project")
+        project_model = apps.get_model("projects", "Project")
         return project_model.objects.get(pk=project_id)
 
     def _search_user_stories(self, project, text):

taiga/searches/services.py

@@ -14,7 +14,7 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-from django.db.models.loading import get_model
+from django.apps import apps
 from django.conf import settings
 
 
@@ -22,7 +22,7 @@ MAX_RESULTS = getattr(settings, "SEARCHES_MAX_RESULTS", 150)
 
 
 def search_user_stories(project, text):
-    model_cls = get_model("userstories", "UserStory")
+    model_cls = apps.get_model("userstories", "UserStory")
     where_clause = ("to_tsvector(coalesce(userstories_userstory.subject) || ' ' || "
                     "coalesce(userstories_userstory.description)) @@ plainto_tsquery(%s)")
 
@@ -34,7 +34,7 @@ def search_user_stories(project, text):
 
 
 def search_tasks(project, text):
-    model_cls = get_model("tasks", "Task")
+    model_cls = apps.get_model("tasks", "Task")
     where_clause = ("to_tsvector(coalesce(tasks_task.subject, '') || ' ' || coalesce(tasks_task.description, '')) "
                     "@@ plainto_tsquery(%s)")
 
@@ -46,7 +46,7 @@ def search_tasks(project, text):
 
 
 def search_issues(project, text):
-    model_cls = get_model("issues", "Issue")
+    model_cls = apps.get_model("issues", "Issue")
     where_clause = ("to_tsvector(coalesce(issues_issue.subject) || ' ' || coalesce(issues_issue.description)) "
                     "@@ plainto_tsquery(%s)")
 
@@ -58,7 +58,7 @@ def search_issues(project, text):
 
 
 def search_wiki_pages(project, text):
-    model_cls = get_model("wiki", "WikiPage")
+    model_cls = apps.get_model("wiki", "WikiPage")
     where_clause = ("to_tsvector(coalesce(wiki_wikipage.slug) || ' ' || coalesce(wiki_wikipage.content)) "
                     "@@ plainto_tsquery(%s)")
 

taiga/timeline/signals.py

@@ -14,9 +14,6 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-from django.db.models.loading import get_model
-from django.db.models import signals
-
 from taiga.timeline.service import push_to_timeline
 
 # TODO: Add events to followers timeline when followers are implemented.

taiga/users/api.py

@@ -16,7 +16,7 @@
 
 import uuid
 
-from django.db.models.loading import get_model
+from django.apps import apps
 from django.db.models import Q
 from django.shortcuts import get_object_or_404
 from django.utils.translation import ugettext_lazy as _
@@ -48,7 +48,7 @@ class MembersFilterBackend(BaseFilterBackend):
     def filter_queryset(self, request, queryset, view):
         project_id = request.QUERY_PARAMS.get('project', None)
         if project_id:
-            Project = get_model('projects', 'Project')
+            Project = apps.get_model('projects', 'Project')
             project = get_object_or_404(Project, pk=project_id)
             if project.memberships.filter(user=request.user).exists() or project.owner == request.user:
                 return queryset.filter(Q(memberships__project=project) | Q(id=project.owner.id)).distinct()
@@ -201,7 +201,7 @@ class UsersViewSet(ModelCrudViewSet):
         user = self.get_object()
         self.check_permissions(request, 'starred', user)
 
-        stars = votes_service.get_voted(user.pk, model=get_model('projects', 'Project'))
+        stars = votes_service.get_voted(user.pk, model=apps.get_model('projects', 'Project'))
         stars_data = StarredSerializer(stars, many=True)
         return Response(stars_data.data)
 

taiga/users/services.py

@@ -18,7 +18,7 @@
 This model contains a domain logic for users application.
 """
 
-from django.db.models.loading import get_model
+from django.apps import apps
 from django.db.models import Q
 from django.conf import settings
 
@@ -40,7 +40,7 @@ def get_and_validate_user(*, username:str, password:str) -> bool:
     exception is raised.
     """
 
-    user_model = get_model("users", "User")
+    user_model = apps.get_model("users", "User")
     qs = user_model.objects.filter(Q(username=username) |
                                    Q(email=username))
     if len(qs) == 0:

tests/factories.py

@@ -19,7 +19,6 @@ import uuid
 import threading
 from datetime import date, timedelta
 
-from django.db.models.loading import get_model
 from django.conf import settings
 
 import factory

tests/integration/resources_permissions/test_projects_resource.py

@@ -1,5 +1,5 @@
 from django.core.urlresolvers import reverse
-from django.db.models.loading import get_model
+from django.apps import apps
 
 from taiga.base.utils import json
 from taiga.projects.serializers import ProjectDetailSerializer
@@ -53,8 +53,8 @@ def data():
                                        role__project=m.private_project2,
                                        role__permissions=[])
 
-    ContentType = get_model("contenttypes", "ContentType")
-    Project = get_model("projects", "Project")
+    ContentType = apps.get_model("contenttypes", "ContentType")
+    Project = apps.get_model("projects", "Project")
 
     project_ct = ContentType.objects.get_for_model(Project)
 

tests/integration/test_auth_api.py

@@ -18,7 +18,7 @@
 import pytest
 from unittest.mock import patch, Mock
 
-from django.db.models.loading import get_model
+from django.apps import apps
 from django.core.urlresolvers import reverse
 from .. import factories
 
@@ -88,7 +88,7 @@ def test_response_200_in_registration_with_github_account(client):
 
 
 def test_response_200_in_registration_with_github_account_in_a_project(client):
-    membership_model = get_model("projects", "Membership")
+    membership_model = apps.get_model("projects", "Membership")
     membership = factories.MembershipFactory(user=None)
     form = {"type": "github",
             "code": "xxxxxx",

tests/integration/test_project_history.py

@@ -21,7 +21,6 @@ from unittest.mock import MagicMock
 from unittest.mock import patch
 
 from django.core.urlresolvers import reverse
-from django.db.models.loading import get_model
 from .. import factories as f
 
 from taiga.projects.history import services

tests/integration/test_project_notifications.py

@@ -20,7 +20,7 @@ import pytest
 from unittest.mock import MagicMock, patch
 
 from django.core.urlresolvers import reverse
-from django.db.models.loading import get_model
+from django.apps import apps
 from .. import factories as f
 
 from taiga.projects.notifications import services
@@ -59,7 +59,7 @@ def test_attach_notify_policy_to_project_queryset():
 def test_create_retrieve_notify_policy():
     project = f.ProjectFactory.create()
 
-    policy_model_cls = get_model("notifications", "NotifyPolicy")
+    policy_model_cls = apps.get_model("notifications", "NotifyPolicy")
     current_number = policy_model_cls.objects.all().count()
     assert current_number == 0
 
@@ -102,7 +102,7 @@ def test_users_to_notify():
     member2 = f.MembershipFactory.create(project=project)
     member3 = f.MembershipFactory.create(project=project)
 
-    policy_model_cls = get_model("notifications", "NotifyPolicy")
+    policy_model_cls = apps.get_model("notifications", "NotifyPolicy")
 
     policy1 = policy_model_cls.objects.get(user=member1.user)
     policy2 = policy_model_cls.objects.get(user=member2.user)