assemblyline-v4-service 4.5.1.dev439__py3-none-any.whl → 4.5.1.dev440__py3-none-any.whl

assemblyline_v4_service/VERSION

@@ -1 +1 @@
-4.5.1.dev439
+4.5.1.dev440

assemblyline_v4_service/updater/client.py

@@ -20,8 +20,12 @@ class SyncableBadlistClient(BadlistClient):
         self.classification_override = None
 
     def add_update_many(self, data: List[Union[dict, BadlistModel]]) -> Dict[str, Any]:
-        return hashlist_add_update_many(self, self.datastore.badlist, data)
+        response = hashlist_add_update_many(self, self.datastore.badlist, data)
 
+        # Ensure new data is available to query
+        self.datastore.badlist.commit()
+
+        return response
 
 class SyncableSafelistClient(SafelistClient):
     def __init__(self, datastore, config=None):
@@ -30,7 +34,12 @@ class SyncableSafelistClient(SafelistClient):
         self.classification_override = None
 
     def add_update_many(self, data: List[Union[dict, SafelistModel]]) -> Dict[str, Any]:
-        return hashlist_add_update_many(self, self.datastore.safelist, data)
+        response = hashlist_add_update_many(self, self.datastore.safelist, data)
+
+        # Ensure new data is available to query
+        self.datastore.safelist.commit()
+
+        return response
 
 
 def hashlist_add_update_many(client: Union[SyncableBadlistClient, SyncableSafelistClient],
@@ -187,10 +196,11 @@ class SyncableSignatureClient(SignatureClient):
                                                     [(self.datastore.signature.UPDATE_SET, 'status', 'DISABLED'),
                                                      (self.datastore.signature.UPDATE_SET, 'last_modified', 'NOW')])
 
+        response = None
         # Proceed with adding/updating signatures
         if len(data) < SIGNATURE_UPDATE_BATCH:
             # Update all of them in a single batch
-            return super().add_update_many(source, sig_type, data, dedup_name)
+            response = super().add_update_many(source, sig_type, data, dedup_name)
         else:
             response = {
                 'success': 0,
@@ -213,7 +223,10 @@ class SyncableSignatureClient(SignatureClient):
                 batch_num += 1
                 start = batch_num * SIGNATURE_UPDATE_BATCH
 
-            return response
+        # Ensure new data is available to query
+        self.datastore.signature.commit()
+
+        return response
 
 
 class UpdaterClient(object):

{assemblyline_v4_service-4.5.1.dev439.dist-info → assemblyline_v4_service-4.5.1.dev440.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: assemblyline-v4-service
-Version: 4.5.1.dev439
+Version: 4.5.1.dev440
 Summary: Assemblyline 4 - Service base
 Home-page: https://github.com/CybercentreCanada/assemblyline-v4-service/
 Author: CCCS Assemblyline development team

{assemblyline_v4_service-4.5.1.dev439.dist-info → assemblyline_v4_service-4.5.1.dev440.dist-info}/RECORD

@@ -1,4 +1,4 @@
-assemblyline_v4_service/VERSION,sha256=Ax92kDfpZpiOZ3TnVlYtc94xyBt9UVwGlh_hjcRrMPk,13
+assemblyline_v4_service/VERSION,sha256=H7M57wxLrsdx3Rh9tXOhqnTbXS1BxKdT8XghlednKqc,13
 assemblyline_v4_service/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 assemblyline_v4_service/healthz.py,sha256=3QGBg0EZuXC6UN411HFwpLNEop9UvS9feFhvBUTP-k4,1576
 assemblyline_v4_service/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -19,7 +19,7 @@ assemblyline_v4_service/dev/run_service_once.py,sha256=1sdxkRBI7x9Sht8EVCTMIW3Mg
 assemblyline_v4_service/updater/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 assemblyline_v4_service/updater/__main__.py,sha256=9Os-u8Tf7MD73JSrUSPmOaErTgfvesNLiEeszU4ujXA,133
 assemblyline_v4_service/updater/app.py,sha256=Mtmx4bkXfP4nFqqa5q15jW8QIXr4JK84lCovxAVyvPs,3317
-assemblyline_v4_service/updater/client.py,sha256=eBOK8doMu9nsIO6kXWtaBMy4hJIKRtQG0iC_21a5CG4,10849
+assemblyline_v4_service/updater/client.py,sha256=tLY84gaGdFBVIDaMgRHIEa7x2S8jBl7lQLzp4seC6aI,11200
 assemblyline_v4_service/updater/gunicorn_config.py,sha256=p3j2KPBeD5jvMw9O5i7vAtlRgPSVVxIG9AO0DfN82J8,1247
 assemblyline_v4_service/updater/helper.py,sha256=Eo8HNyHRZ3OL_b6onWIFj_BykwYj94JgRrT9GQx2Tsg,10076
 assemblyline_v4_service/updater/updater.py,sha256=HqRVUs5nNIfKmHu7g1GUkgC7bvYWs2EC9ZFy1kj-Ltw,31609
@@ -38,8 +38,8 @@ test/test_common/test_request.py,sha256=Ceyds8BNO1O0f1kH1VEb84faJcaupvSjVKIrGdHe
 test/test_common/test_result.py,sha256=6BiOKxEPrKBjOY44jv3TY-yiXm0qI1ok_CZBnjP9TM4,45447
 test/test_common/test_task.py,sha256=P44mNcSe-3tJgDk9ppN3KbM7oN4LBVIuhONG-Gveh74,19007
 test/test_common/test_utils.py,sha256=TbnBxqpS_ZC5ptXR9XJX3xtbItD0mTbtiBxxdyP8J5k,5904
-assemblyline_v4_service-4.5.1.dev439.dist-info/LICENCE.md,sha256=NSkYo9EH8h5oOkzg4VhjAHF4339MqPP2cQ8msTPgl-c,1396
-assemblyline_v4_service-4.5.1.dev439.dist-info/METADATA,sha256=AhGU366uS5olq3jNyh9kfxHdEhyA1rgjxffiuw8XL38,5603
-assemblyline_v4_service-4.5.1.dev439.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-assemblyline_v4_service-4.5.1.dev439.dist-info/top_level.txt,sha256=LpTOEaVCatkrvbVq3EZseMSIa2PQZU-2rhuO_FTpZgY,29
-assemblyline_v4_service-4.5.1.dev439.dist-info/RECORD,,
+assemblyline_v4_service-4.5.1.dev440.dist-info/LICENCE.md,sha256=NSkYo9EH8h5oOkzg4VhjAHF4339MqPP2cQ8msTPgl-c,1396
+assemblyline_v4_service-4.5.1.dev440.dist-info/METADATA,sha256=7GM_yfY93KURNOaDK5S3B2mo1RGNWVv-DToGolqwPmM,5603
+assemblyline_v4_service-4.5.1.dev440.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+assemblyline_v4_service-4.5.1.dev440.dist-info/top_level.txt,sha256=LpTOEaVCatkrvbVq3EZseMSIa2PQZU-2rhuO_FTpZgY,29
+assemblyline_v4_service-4.5.1.dev440.dist-info/RECORD,,