command_cursor.py 796 B

12345678910111213141516171819202122232425262728293031323334353637383940
  1. class CommandCursor(object):
  2. def __init__(self, collection, curser_info=None, address=None, retrieved=0):
  3. self._collection = iter(collection)
  4. self._id = None
  5. self._address = address
  6. self._data = {}
  7. self._retrieved = retrieved
  8. self._batch_size = 0
  9. self._killed = (self._id == 0)
  10. @property
  11. def address(self):
  12. return self._address
  13. def close(self):
  14. pass
  15. def batch_size(self, batch_size):
  16. return self
  17. @property
  18. def alive(self):
  19. return True
  20. def __iter__(self):
  21. return self
  22. def next(self):
  23. return next(self._collection)
  24. __next__ = next
  25. def __enter__(self):
  26. return self
  27. def __exit__(self, exc_type, exc_val, exc_tb):
  28. return