class SandboxInOut(object):
"""Collect written text, and return it when called."""
class InnerSandboxInOut(object):
def __init__(self, printer, _getattr_=None):
self.printer = printer
self._getattr_ = _getattr_
def write(self, text: str):
self.printer.write(text, consumer=False)
def read(self):
line = self.printer.readline(consumer=True)
if len(line) == 0:
return None
return str(line)
def __call__(self, testing=None):
return self
def _call_print(self, *objects, **kwargs):
if kwargs.get('file', None) is None:
kwargs['file'] = self
else:
self._getattr_(kwargs['file'], 'write')
print(*objects, **kwargs)
def __init__(self, _getattr_=None):
self.stream = BufferedRandom(BytesIO())
self.position = 0
def printer(self):
return self.InnerSandboxInOut(self)
def reader(self):
return lambda: self.readline()
def write(self, line, consumer=True):
if not consumer:
self.stream.seek(2) # go to the end of the stream
else:
self.stream.seek(0, self.position)
self.stream.write(bytes(line, 'utf-8'))
if consumer:
self.position = self.stream.tell()
def readline(self, consumer=True):
if not consumer:
self.stream.seek(2) # go to the end of the stream
else:
self.stream.seek(0, self.position)
line = self.stream.readline()
if len(line) == 0:
return None
if consumer:
self.position = self.stream.tell()
return str(line.decode('utf-8')).strip()
def get_stream(self):
return self.stream
def __init__(self, file_: BufferedRandom, length: int) -> None:
# Read the file into a buffer.
file_.seek(0, SEEK_END)
buf = bytearray(file_.tell())
file_.seek(0, SEEK_SET)
file_.readinto(buf)
# Overwrite the header, in case it is currently being written to.
header = list_header(length)
buf[:len(header)] = header
# Parse the buffer's contents as list items.
super().__init__(namespacify(cbor2.loads(buf)))
# Store the file pointer for `extend` calls.
self._file = file_
def FileTell(file_handle: io.BufferedRandom) -> int:
return file_handle.tell()
