inventree/InvenTree
2
0
Fork 0
mirror of https://github.com/inventree/InvenTree.git synced 2025-06-17 20:45:44 +00:00
Code Activity

Fixed test cases?

Browse Source
This commit is contained in:
eeintech
2021-01-12 15:02:44 -05:00
parent 452d22579a
commit 49c954aa68
3 changed files with 44 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
InvenTree/users/models.py
View File

@ -413,26 +413,34 @@ class Owner(models.Model):
@classmethod
def create(cls, owner):
# Check if owner already exists
existing_owner = cls.get_owner(owner)
if not existing_owner:
# Create new owner
try:
return cls.objects.create(owner=owner)
except IntegrityError:
return None
@classmethod
def get_owner(cls, user_or_group):
# Get corresponding owner
try:
group = Owner.objects.get(owner_id=owner.id,
group = Owner.objects.get(owner_id=user_or_group.id,
owner_type=ContentType.objects.get_for_model(Group).id)
return group
except Owner.DoesNotExist:
pass
try:
user = Owner.objects.get(owner_id=owner.id,
user = Owner.objects.get(owner_id=user_or_group.id,
owner_type=ContentType.objects.get_for_model(User).id)
return user
except Owner.DoesNotExist:
pass
# Create new owner
try:
return cls.objects.create(owner=owner)
except IntegrityError:
return None
return None
def get_users(self, include_group=False):