fix: [retro hunt] fix object tag queue + decoded content

This commit is contained in:
Terrtia 2023-06-19 10:53:15 +02:00
parent 6c21a8f8cf
commit b64c70e0cd
No known key found for this signature in database
GPG key ID: 1E1B1F50D84613D0
2 changed files with 23 additions and 5 deletions

View file

@ -111,13 +111,25 @@ class Decoded(AbstractDaterangeObject):
def get_rel_path(self, mimetype=None):
if not mimetype:
mimetype = self.get_mimetype()
if not mimetype:
self.logger.warning(f'Decoded {self.id}: Empty mimetype')
return None
return os.path.join(HASH_DIR, mimetype, self.id[0:2], self.id)
def get_filepath(self, mimetype=None):
return os.path.join(os.environ['AIL_HOME'], self.get_rel_path(mimetype=mimetype))
rel_path = self.get_rel_path(mimetype=mimetype)
if not rel_path:
return None
else:
return os.path.join(os.environ['AIL_HOME'], rel_path)
def get_content(self, mimetype=None, r_type='str'):
filepath = self.get_filepath(mimetype=mimetype)
if not filepath:
if r_type == 'str':
return ''
else:
return b''
if r_type == 'str':
with open(filepath, 'r') as f:
content = f.read()
@ -442,7 +454,7 @@ def get_all_decodeds_objects(filters={}):
if i >= len(files):
files = []
for file in files:
yield Decoded(file).id
yield Decoded(file)
############################################################################

View file

@ -128,10 +128,16 @@ class Retro_Hunt_Module(AbstractModule):
self.retro_hunt.add(self.obj.get_type(), self.obj.get_subtype(), obj_id)
# TODO FILTER Tags
# TODO refactor Tags module for all object type
# Tags
for tag in self.tags:
msg = f'{tag};{id}'
self.add_message_to_queue(msg, 'Tags')
if self.obj.get_type() == 'item':
for tag in self.tags:
msg = f'{tag};{obj_id}'
self.add_message_to_queue(msg, 'Tags')
else:
for tag in self.tags:
self.obj.add_tag(tag)
# # Mails
# EXPORTER MAILS