from cStringIO import StringIO

class CustomStringIO():

    def __init__(self, max_buffer = 4096):

        self.max_buffer = max_buffer #用来设置stringIO最大存储字节数

        self.buffer = StringIO() #cStringIO.StringIO 不能用于继承，一旦赋值初始化了，就不能改变，但其效率比StringIO.StringIO高，cStringIO没有len和pos属性。（还有，cStringIO不支持Unicode编码）

    def truncate_to_maxsize(self):

        '''自定义裁剪函数，当StringIO存储字节数超过定义的最大字节数时，裁剪最后写入的最大长度

        self.buffer.seek(-self.max_buffer, 2)

        _last = self.buffer.readlines()

        self.buffer.truncate(0)

        self.buffer.writelines(_last)

    def write(self,string):

        self.buffer.write(string)

        self.truncate_to_maxsize()

    def getvalue(self):

        return self.buffer.getvalue()

if __name__ == '__main__':

    mystringio=CustomStringIO(10)

    mystringio.write('lyhlyh')

    mystringio.write('ldfdffdfdsfdsfsdfs')

    mystringio.write('fdfdfdfdf')

    print mystringio.getvalue()