Working resetdb

ProjetSigma · Nov 30, 2016 · 1e97be6 · 1e97be6
1 parent 222b2e7
commit 1e97be6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sigma_core/models/user.py b/sigma_core/models/user.py
@@ -1,6 +1,4 @@
 from django.db import models
-from sigma_core.models.group_member import GroupMember
-from sigma_core.models.group import Group
 from django.db.models import Q
 
 from django.contrib.auth.models import AbstractBaseUser, BaseUserManager
@@ -97,6 +95,8 @@ def has_common_group(self, user):
         return len(set(self.memberships.values_list('group', flat=True)).intersection(user.memberships.all().values_list('group', flat=True))) > 0
 
     def get_group_membership(self, group):
+        from sigma_core.models.group_member import GroupMember
+        from sigma_core.models.group import Group
         try:
             return self.memberships.get(group=group)
         except GroupMember.DoesNotExist: