From 3cec49e1f444f7ac2246b3cde70bc52004925297 Mon Sep 17 00:00:00 2001
From: isox <isox@vulners.com>
Date: Tue, 28 Jul 2020 22:54:40 +0300
Subject: [PATCH] Minor 1.5.7

Added references option to the documents call.
---
 vulners/__init__.py | 2 +-
 vulners/api.py      | 8 ++++----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/vulners/__init__.py b/vulners/__init__.py
index eedb35a..8fed86f 100644
--- a/vulners/__init__.py
+++ b/vulners/__init__.py
@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
 
-__version__ = "1.5.6"
+__version__ = "1.5.7"
 
 from vulners.api import Vulners
diff --git a/vulners/api.py b/vulners/api.py
index 0f8f290..7569083 100644
--- a/vulners/api.py
+++ b/vulners/api.py
@@ -519,7 +519,7 @@ def cpeVulnerabilities(self, cpeString, maxVulnerabilities = 50):
             dataDocs[elementData.get('bulletinFamily')] = dataDocs.get(elementData.get('bulletinFamily'), []) + [elementData]
         return dataDocs
 
-    def document(self, identificator, fields = None):
+    def document(self, identificator, fields = None, references = False):
         """
         Fetch information about bulletin by identificator
 
@@ -527,7 +527,7 @@ def document(self, identificator, fields = None):
         :param references: Search for the references in all collections
         :return: bulletin data dict
         """
-        results = self.__id(identificator, references=False, fields = fields or self.default_fields)
+        results = self.__id(identificator, references=references, fields = fields or self.default_fields)
         return results.get('documents', {}).get(identificator, {})
 
     def audit(self, os, os_version, package):
@@ -597,7 +597,7 @@ def kbUpdates(self, kb_identificator, fields = None):
                 dataDocs.append(element.get('_source'))
         return AttributeList(dataDocs, total=total)
 
-    def documentList(self, identificatorList, fields = None):
+    def documentList(self, identificatorList, fields = None, references = False):
         """
         Fetch information about multiple bulletin identificators
 
@@ -607,7 +607,7 @@ def documentList(self, identificatorList, fields = None):
 
         if not isinstance(identificatorList, (list,set)) or not all(isinstance(item, string_types) for item in identificatorList):
             raise TypeError('Identificator list is expected to be a list of strings')
-        return self.__id(identificatorList, references=False, fields=fields or self.default_fields).get('documents')
+        return self.__id(identificatorList, references=references, fields=fields or self.default_fields).get('documents')
 
     def references(self, identificator, fields = None):
         """