Merge pull request #7 from intelowlproject/develop

fix to resolve client issues when requesting all analyzers

Author: mlodic

intel_owl_client.py

@@ -52,6 +52,10 @@ def intel_owl_client():
 
     logger = get_logger(args.debug, args.log_to_file)
 
+    _pyintelowl_logic(args, logger)
+
+
+def _pyintelowl_logic(args, logger):
     md5 = None
     results = []
     elapsed_time = None
@@ -92,8 +96,10 @@ def intel_owl_client():
             logger.info("about to request ask_analysis_availability for md5: {}, analyzers: {}"
                         "".format(md5, args.analyzers_list))
 
-            api_request_result = pyintelowl_client.ask_analysis_availability(md5, args.analyzers_list,
-                                                                             args.check_reported_analysis_too)
+            api_request_result = \
+                pyintelowl_client.ask_analysis_availability(md5, args.analyzers_list,
+                                                            args.run_all_available_analyzers,
+                                                            args.check_reported_analysis_too)
             errors = api_request_result.get('errors', [])
             if errors:
                 raise IntelOwlClientException("API ask_analysis_availability failed. Errors: {}"

pyintelowl/pyintelowl.py

@@ -30,20 +30,24 @@ def session(self):
             session.verify = self.certificate
             session.headers.update({
                 'Authorization': 'Token {}'.format(self.api_key),
-                'User-Agent': 'IntelOwlClient/0.2.0',
+                'User-Agent': 'IntelOwlClient/0.2.1',
             })
             self._session = session
 
         return self._session
 
-    def ask_analysis_availability(self, md5, analyzers_needed, check_reported_analysis_too=False):
+    def ask_analysis_availability(self, md5, analyzers_needed,
+                                  run_all_available_analyzers=False,
+                                  check_reported_analysis_too=False):
         answer = {}
         errors = []
         try:
             params = {
                 "md5": md5,
                 "analyzers_needed": analyzers_needed
             }
+            if run_all_available_analyzers:
+                params['run_all_available_analyzers'] = True
             if not check_reported_analysis_too:
                 params['running_only'] = True
             url = self.instance + "/api/ask_analysis_availability"

setup.py

@@ -10,7 +10,7 @@
 # This call to setup() does all the work
 setup(
     name="pyintelowl",
-    version="0.2.0",
+    version="0.2.1",
     description="Client and Library for Intel Owl",
     long_description=README,
     long_description_content_type="text/markdown",