fix: [trackers] fix pgp yara tracker

pull/594/head
Terrtia 2023-05-12 15:43:41 +02:00
parent 9b2e8718d7
commit e365555905
No known key found for this signature in database
GPG Key ID: 1E1B1F50D84613D0
2 changed files with 10 additions and 4 deletions

View File

@ -108,8 +108,11 @@ class AbstractDaterangeObject(AbstractObject, ABC):
sparkline.append(self.get_nb_seen_by_date(date))
return sparkline
def get_content(self):
return self.id
def get_content(self, r_type='str'):
if r_type == 'str':
return self.id
elif r_type == 'bytes':
return self.id.encode()
def _add_create(self):
r_object.sadd(f'{self.type}:all', self.id)

View File

@ -116,8 +116,11 @@ class AbstractSubtypeObject(AbstractObject, ABC):
if date > last_seen:
self.set_last_seen(date)
def get_content(self):
return self.id
def get_content(self, r_type='str'):
if r_type == 'str':
return self.id
if r_type == 'bytes':
return self.id.encode()
def get_sparkline(self):
sparkline = []