manyi/venv/Lib/site-packages/Tea/stream.py

# -*- coding: utf-8 -*-
from Tea.converter import py2
if py2:
    from _io import BytesIO
else:
    from _io import (
        TextIOWrapper,
        BufferedReader, BytesIO,
        BufferedWriter
    )


class BaseStream(object):
    def __init__(self, size=1024):
        self.size = size

    def read(self, size=1024):
        raise NotImplementedError('read method must be overridden')

    def __len__(self):
        raise NotImplementedError('__len__ method must be overridden')

    def next(self):
        raise NotImplementedError('__next__ method must be overridden')

    def __iter__(self):
        return self


if py2:
    STREAM_CLASS = (file, BaseStream, BytesIO)
    READABLE = (BaseStream, file, BytesIO)
    WRITABLE = (file, )
else:
    STREAM_CLASS = (TextIOWrapper, BufferedReader, BaseStream, BytesIO)
    READABLE = (BaseStream, BufferedReader, BytesIO)
    WRITABLE = (BufferedWriter,)