add support for new flags #111 #112

muchdogesec · Feb 5, 2025 · 31269d8 · 31269d8
1 parent 302da67
commit 31269d8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/vulmatch/server/serializers.py b/vulmatch/server/serializers.py
@@ -24,6 +24,8 @@ class NVDTaskSerializer(serializers.Serializer):
     last_modified_earliest = serializers.DateField(help_text="(`YYYY-MM-DD`): earliest date")
     last_modified_latest = serializers.DateField(help_text="(`YYYY-MM-DD`): latest date \n* default is `1980-01-01`")
     ignore_embedded_relationships = serializers.BooleanField(default=False)
+    ignore_embedded_relationships_sro = serializers.BooleanField(default=True)
+    ignore_embedded_relationships_smo = serializers.BooleanField(default=True)
     always_latest = serializers.BooleanField(default=False, help_text="Skip _is_latest check when uploading, this should only be used when working with large backfill tasks")
 
     def validate(self, attrs):
@@ -38,6 +40,8 @@ class StixVersionsSerializer(serializers.Serializer):
 
 class ACPSerializer(serializers.Serializer):
     ignore_embedded_relationships = serializers.BooleanField(default=False)
+    ignore_embedded_relationships_sro = serializers.BooleanField(default=True)
+    ignore_embedded_relationships_smo = serializers.BooleanField(default=True)
     modified_min = serializers.DateTimeField(required=False)
     created_min = serializers.DateTimeField(required=False)
 

diff --git a/vulmatch/worker/tasks.py b/vulmatch/worker/tasks.py
@@ -128,11 +128,12 @@ def upload_file(filename, collection_name, stix2arango_note=None, job_id=None, p
         database=settings.ARANGODB_DATABASE,
         collection=collection_name,
         stix2arango_note=stix2arango_note,
-        ignore_embedded_relationships=params.get('ignore_embedded_relationships', False),
         host_url=settings.ARANGODB_HOST_URL,
         username=settings.ARANGODB_USERNAME,
         password=settings.ARANGODB_PASSWORD,
-        always_latest=params.get('always_latest', False),
+        # always_latest=params.get('always_latest', False),
+        # ignore_embedded_relationships=params.get('ignore_embedded_relationships', False),
+        **params
     )
     s2a.run()