Ejemplo n.º 1
0
 def find(self, *args, **kwargs):
     """
     provide generator over rides that matched the given parameter
     :param args: ``Collection.find``
     :param kwargs: ``Collection.find``
     """
     for doc in self.mongo_crud.find(self.COLLECTION_NAME, *args, **kwargs):
         doc[self.DOC_ID_KEY] = doc.pop(self.DB_ID_KEY)
         yield SiriRide.from_json(doc)
Ejemplo n.º 2
0
 def read(self, doc_id) -> SiriRide:
     """
     Get a SiriRide instance from the database.
     :param doc_id: represent document id in database.
     :return: SiriRide instance with requested id
     :raise: FilterError in case no SiriRide record with given doc_id
     """
     res = self.mongo_crud.read(self.COLLECTION_NAME, doc_id)
     res[self.DOC_ID_KEY] = res.pop(self.DB_ID_KEY)
     return SiriRide.from_json(res)