assemblyline-v4-service 4.4.1.dev226__py3-none-any.whl → 4.4.1.dev230__py3-none-any.whl

assemblyline_v4_service/VERSION

@@ -1 +1 @@
-4.4.1.dev226
+4.4.1.dev230

assemblyline_v4_service/common/result.py

@@ -39,6 +39,11 @@ BODY_FORMAT = StringTable('BODY_FORMAT', [
     ('TIMELINE', 12)
 ])
 
+PROMOTE_TO = StringTable('PROMOTE_TO', [
+    ('SCREENSHOT', 0),
+    ('ENTROPY', 1)
+])
+
 
 class InvalidHeuristicException(Exception):
     pass
@@ -407,7 +412,6 @@ class TableSectionBody(SectionBody):
         self._config = {'column_order': order}
 
 
-
 class ImageSectionBody(SectionBody):
     def __init__(self, request: ServiceRequest) -> None:
         self._request = request
@@ -484,6 +488,7 @@ class ResultSection:
         self.zeroize_on_tag_safe = zeroize_on_tag_safe
         self.auto_collapse = auto_collapse
         self.zeroize_on_sig_safe = zeroize_on_sig_safe
+        self._promote_to = None
 
         if isinstance(title_text, list):
             title_text = ''.join(title_text)
@@ -517,6 +522,10 @@ class ResultSection:
     def heuristic(self):
         return self._heuristic
 
+    @property
+    def promote_to(self):
+        return self._promote_to
+
     @property
     def subsections(self):
         return self._subsections
@@ -681,6 +690,9 @@ class ResultGraphSection(TypeSpecificResultSection):
     def set_colormap(self, cmap_min: int, cmap_max: int, values: List[int]) -> None:
         self.section_body.set_colormap(cmap_min, cmap_max, values)
 
+    def promote_as_entropy(self):
+        self._promote_to = PROMOTE_TO.ENTROPY
+
 
 class ResultURLSection(TypeSpecificResultSection):
     def __init__(self, title_text: Union[str, List], **kwargs):
@@ -764,6 +776,9 @@ class ResultImageSection(TypeSpecificResultSection):
 
         return ocr_section
 
+    def promote_as_screenshot(self):
+        self._promote_to = PROMOTE_TO.SCREENSHOT
+
 
 class ResultTimelineSection(TypeSpecificResultSection):
     def __init__(self, title_text: Union[str, List], **kwargs):
@@ -802,7 +817,8 @@ class Result:
             tags=unflatten(section.tags),
             title_text=section.title_text,
             zeroize_on_tag_safe=section.zeroize_on_tag_safe,
-            auto_collapse=section.auto_collapse
+            auto_collapse=section.auto_collapse,
+            promote_to=section.promote_to
         ))
 
     def _flatten_sections(self, section: ResultSection, root: bool = True) -> None:

{assemblyline_v4_service-4.4.1.dev226.dist-info → assemblyline_v4_service-4.4.1.dev230.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: assemblyline-v4-service
-Version: 4.4.1.dev226
+Version: 4.4.1.dev230
 Summary: Assemblyline 4 - Service base
 Home-page: https://github.com/CybercentreCanada/assemblyline-v4-service/
 Author: CCCS Assemblyline development team

{assemblyline_v4_service-4.4.1.dev226.dist-info → assemblyline_v4_service-4.4.1.dev230.dist-info}/RECORD

@@ -1,4 +1,4 @@
-assemblyline_v4_service/VERSION,sha256=EB5Y2GlVD6c-QH3uF7FS3hlZMmK1DIRn6KZT0BAELTQ,13
+assemblyline_v4_service/VERSION,sha256=mSbg-uE2Y5EbQ2Tqr0iXBZJhzGuCJMaRX9ddLVtEx1c,13
 assemblyline_v4_service/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 assemblyline_v4_service/healthz.py,sha256=sS1cFkDLw8hUPMpj7tbHXFv8ZmHcazrwZ0l6oQDwwkQ,1575
 assemblyline_v4_service/run_privileged_service.py,sha256=9uTfHetXR5G-EDKMDrgfWUOw34yr64-cj6Cm9eZaCbQ,14547
@@ -10,7 +10,7 @@ assemblyline_v4_service/common/helper.py,sha256=xs9quuf-M1JOdKieBqOmWaOece0CtzXF
 assemblyline_v4_service/common/ocr.py,sha256=gFPVAGVsmVk4SKL7hCMCT7NEyG-lFLniEFQup6WtCsc,4384
 assemblyline_v4_service/common/ontology_helper.py,sha256=Wao4VZ_YrlmLd3-CXwqsK6c5VTiVZ89CKEhZOgZrm0Q,7792
 assemblyline_v4_service/common/request.py,sha256=zgmKvYnK1KcBWsQQUZmeBXGIi-P5m3ltxqGY2Xo696c,9764
-assemblyline_v4_service/common/result.py,sha256=qOwyFP4zJMpngZWDIuc0bFN-tNjLuPP2RDOgkPNwEXI,30349
+assemblyline_v4_service/common/result.py,sha256=3d--Xa0dSqfXUVTeLG3WUU0geYG4xsmx7Ukz7XlnPEw,30751
 assemblyline_v4_service/common/task.py,sha256=22dl6FiQwqeL0CxkSzmHn9mKSL5xf9zE5QjxT-nDLs4,12504
 assemblyline_v4_service/common/utils.py,sha256=71ndFdbdfCeRCM-eYs1htKqJ-1miiWbsF_UgHnKKNZ4,3112
 assemblyline_v4_service/dev/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -22,8 +22,8 @@ assemblyline_v4_service/updater/client.py,sha256=VKmkiDz3klvMzXiGSQTH7QN-MfHuJnw
 assemblyline_v4_service/updater/gunicorn_config.py,sha256=p3j2KPBeD5jvMw9O5i7vAtlRgPSVVxIG9AO0DfN82J8,1247
 assemblyline_v4_service/updater/helper.py,sha256=CO0zY6R0gx36iOCSfZyCYCvUAUP76qQMqZ1z_HkZMME,9123
 assemblyline_v4_service/updater/updater.py,sha256=VB8jt5qg604HZyG6epHqGqG-7yPUCE5-i6uHOO2n72s,32045
-assemblyline_v4_service-4.4.1.dev226.dist-info/LICENCE.md,sha256=NSkYo9EH8h5oOkzg4VhjAHF4339MqPP2cQ8msTPgl-c,1396
-assemblyline_v4_service-4.4.1.dev226.dist-info/METADATA,sha256=1-aBoyAIbRkyh-n2VCQpFucd3UsiFHKRnZFrpI4nOoQ,9369
-assemblyline_v4_service-4.4.1.dev226.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-assemblyline_v4_service-4.4.1.dev226.dist-info/top_level.txt,sha256=Ut5IqePObcxlJ8rv2--dOAzYbxzqlllfiV_51cbqjbA,24
-assemblyline_v4_service-4.4.1.dev226.dist-info/RECORD,,
+assemblyline_v4_service-4.4.1.dev230.dist-info/LICENCE.md,sha256=NSkYo9EH8h5oOkzg4VhjAHF4339MqPP2cQ8msTPgl-c,1396
+assemblyline_v4_service-4.4.1.dev230.dist-info/METADATA,sha256=lAvKY-Te01Ov-lDTLSoFH_IlBbCLm6Rny-uNk2GtEpc,9369
+assemblyline_v4_service-4.4.1.dev230.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+assemblyline_v4_service-4.4.1.dev230.dist-info/top_level.txt,sha256=Ut5IqePObcxlJ8rv2--dOAzYbxzqlllfiV_51cbqjbA,24
+assemblyline_v4_service-4.4.1.dev230.dist-info/RECORD,,