Merge branch 'fix/search-regression' into 'master'

Fixed bug in Malwarecage.search methods See merge request mlwr/mwdblib!31
CERT-Polska · Feb 4, 2020 · 2ffbc5c · 2ffbc5c
2 parents 6fb2fc4 + 5d42410
commit 2ffbc5c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/core.py b/src/core.py
@@ -416,7 +416,7 @@ def search_files(self, query):
         :rtype: Iterator[:class:`MalwarecageFile`]
         :raises: requests.exceptions.HTTPError
         """
-        return self._recent("file", query)
+        return self._recent(MalwarecageFile, query)
 
     def search_configs(self, query):
         """
@@ -427,7 +427,7 @@ def search_configs(self, query):
         :rtype: Iterator[:class:`MalwarecageConfig`]
         :raises: requests.exceptions.HTTPError
         """
-        return self._recent("config", query)
+        return self._recent(MalwarecageConfig, query)
 
     def search_blobs(self, query):
         """
@@ -438,7 +438,7 @@ def search_blobs(self, query):
         :rtype: Iterator[:class:`MalwarecageBlob`]
         :raises: requests.exceptions.HTTPError
         """
-        return self._recent("blob", query)
+        return self._recent(MalwarecageBlob, query)
 
     @staticmethod
     def _convert_bytes(data):