final code review changes

pull/81/head
msramalho 2023-06-26 17:32:19 +01:00
rodzic 0cba2c25c6
commit a0971fc601
3 zmienionych plików z 3 dodań i 3 usunięć

Wyświetl plik

@ -50,6 +50,7 @@ configurations:
text: textual content
screenshot: screenshot
hash: hash
pdq_hash: perceptual hashes
wacz: wacz
replaywebpage: replaywebpage
instagram_tbot_archiver:

Wyświetl plik

@ -35,7 +35,7 @@ class Media:
def all_inner_media(self, include_self=False):
""" Media can be inside media properties, examples include transformations on original media.
This function return a generator for all the inner media.
This function returns a generator for all the inner media.
"""
if include_self: yield self
for prop in self.properties.values():

Wyświetl plik

@ -28,8 +28,7 @@ class PdqHashEnricher(Enricher):
for m in to_enrich.media:
for media in m.all_inner_media(True):
if media.is_image() and media.get("id") != "screenshot":
if len(hd := self.calculate_pdq_hash(media.filename)):
if media.is_image() and media.get("id") != "screenshot" and len(hd := self.calculate_pdq_hash(media.filename)):
media.set("pdq_hash", hd)
def calculate_pdq_hash(self, filename):