Merge pull request #24 from CybercentreCanada/assemblyline/issues/101

Update to use UpdaterClient

api_vector/updater.py

@@ -3,7 +3,6 @@
 
 from assemblyline.common import forge
 from assemblyline.common.path import strip_path_inclusion
-from assemblyline_client.v4_client.client import Client as Client4
 from assemblyline_v4_service.updater.updater import ServiceUpdater, UpdateSource
 
 classification = forge.get_classification()
@@ -13,7 +12,7 @@ class APIVectorUpdateServer(ServiceUpdater):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
-    def import_update(self, files_sha256: str, al_client: Client4, source: str, default_classification: str = None):
+    def import_update(self, files_sha256: str, source: str, default_classification: str = None):
         if len(files_sha256) != 1:
             # No file were found in the latest source update
             source_config: UpdateSource = [x for x in self._service.update_config.sources if x["name"] == source][0]

service_manifest.yml

@@ -16,6 +16,7 @@ disable_cache: false
 enabled: true
 is_external: false
 licence_count: 0
+privileged: true
 
 config:
   min_confidence: 50
@@ -34,6 +35,9 @@ dependencies:
       command: ["python", "-m", "api_vector.updater"]
       image: ${REGISTRY}cccs/assemblyline-service-apivector:$SERVICE_TAG
       ports: ["5003"]
+      environment:
+        - name: MAX_REQUESTS
+          value: 0
     run_as_core: True
 
 update_config: