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

Updated docs and TODO (we won't get to it this PR)

This commit is contained in:
Shay Nehmad 2019-10-02 09:59:25 +03:00
parent 4d9467bac9
commit 628ebc06c1
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -87,7 +87,8 @@ class Monkey(Document):
os = "windows" os = "windows"
return os return os
# TODO This is not a field therefore cache shouldn't be here # TODO This is not really a field, therefore cache shouldn't be here - we should cache the IP addresses and this
# should be a regular method.
@staticmethod @staticmethod
@ring.lru() @ring.lru()
def get_label_by_id(object_id): def get_label_by_id(object_id):
@ -97,11 +98,17 @@ class Monkey(Document):
@staticmethod @staticmethod
@ring.lru() @ring.lru()
def get_hostname_by_id(object_id): def get_hostname_by_id(object_id):
"""
:param object_id: the object ID of a Monkey in the database.
:return: The hostname of that machine.
:note: Use this and not monkey.hostname for performance - this is lru-cached.
"""
return Monkey.get_single_monkey_by_id(object_id).hostname return Monkey.get_single_monkey_by_id(object_id).hostname
def set_hostname(self, hostname): def set_hostname(self, hostname):
""" """
Need this to clear the cache Sets a new hostname for a machine and clears the cache for getting it.
:param hostname: The new hostname for the machine.
""" """
self.hostname = hostname self.hostname = hostname
self.save() self.save()