Add possibility to supress the _id value in query_data_and_generate_dataframe

cdplib/db_handlers/MongodbHandler.py

@@ -247,15 +247,15 @@ class MongodbHandler:
 
     def query_data_and_generate_dataframe(self, collection_name: str, attribute: str = None,
                                           attribute_value: str = None, comparison_operator: str = '$eq',
-                                          index = None, return_as_dataframe: bool = True):
+                                          index = None, return_as_dataframe: bool = True, return_id: bool = False):
         '''
 
         '''
         try:
             if attribute == None or attribute_value == None:
-                data = self._database[collection_name].find()
+                data = self._database[collection_name].find({'_id': return_id})
             else:
-                data = self._database[collection_name].find({attribute: {comparison_operator: attribute_value}})
+                data = self._database[collection_name].find({attribute: {comparison_operator: attribute_value}, {'_id': return_id}})
 
         except Exception as error:
             self._log.log_and_raise_error(('An error occured trying to query data from {}, with query {}: {}:{}. \nError:{}').format(collection_name, attribute, comparison_operator, attribute_value, error))