p15693087
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests 1 Projects Releases Wiki Activity

Using objects.get() instead of objects()[0]

This commit is contained in:
Shay Nehmad 2019-07-22 17:42:20 +03:00
parent 89d49a7d3f
commit 92e400a66f
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
monkey/monkey_island/cc/models/monkey.py
View File

@ -42,16 +42,16 @@ class Monkey(Document):
@staticmethod
def get_single_monkey_by_id(db_id):
try:
return Monkey.objects(id=db_id)[0]
except IndexError:
raise MonkeyNotFoundError("id: {0}".format(str(db_id)))
return Monkey.objects.get(id=db_id)
except mongoengine.DoesNotExist as ex:
raise MonkeyNotFoundError("info: {0} | id: {1}".format(ex.message, str(db_id)))
@staticmethod
def get_single_monkey_by_guid(monkey_guid):
try:
return Monkey.objects(guid=monkey_guid)[0]
except IndexError:
raise MonkeyNotFoundError("guid: {0}".format(str(monkey_guid)))
return Monkey.objects.get(guid=monkey_guid)
except mongoengine.DoesNotExist as ex:
raise MonkeyNotFoundError("info: {0} | guid: {1}".format(ex.message, str(monkey_guid)))
@staticmethod
def get_latest_modifytime():