From 88591715c225b703c28be6993d1bb1ca4deaf623 Mon Sep 17 00:00:00 2001 From: phate1 Date: Wed, 2 Oct 2019 22:43:42 +0100 Subject: [PATCH] fixed user attribute issue not sure if it the "best" way to fix but it works :) also updated a couple of the other attributes to match the current handle_file --- virustotal_download_service/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/virustotal_download_service/__init__.py b/virustotal_download_service/__init__.py index 6da78a91..9e21fa42 100644 --- a/virustotal_download_service/__init__.py +++ b/virustotal_download_service/__init__.py @@ -200,9 +200,9 @@ def run(self, obj, config): handle_file(filename = obj.md5, data = data, source = "VirusTotal", - reference = "Binary downloaded from VT based on MD5", - user = "VT Download Service", - method = "VirusTotal Download Service", + source_reference = "Binary downloaded from VT based on MD5", + user = self.current_task.user, + source_method = "VirusTotal Download Service", md5_digest = obj.md5 ) except Exception as e: logger.error("VirusTotal: Sample creation failed ({0})".format(e))