Mirror
/
cpython
mirror of https://github.com/python/cpython
4
1
Fork 0
Code Issues Releases Wiki Activity

Sjoerd's StringIO speed-up

This commit is contained in:
Guido van Rossum 1996-06-19 14:54:19 +00:00
parent d4cc04c672
commit d3901028cf
1 changed files with 30 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
Lib/StringIO.py
View File

@ -31,6 +31,8 @@ import string
class StringIO: class StringIO:
def __init__(self, buf = ''): def __init__(self, buf = ''):
self.buf = buf self.buf = buf
self.len = len(buf)
self.buflist = []
self.pos = 0 self.pos = 0
self.closed = 0 self.closed = 0
self.softspace = 0 self.softspace = 0
@ -41,25 +43,34 @@ class StringIO:
def isatty(self): def isatty(self):
return 0 return 0
def seek(self, pos, mode = 0): def seek(self, pos, mode = 0):
if self.buflist:
self.buf = self.buf + string.joinfields(self.buflist, '')
self.buflist = []
if mode == 1: if mode == 1:
pos = pos + self.pos pos = pos + self.pos
elif mode == 2: elif mode == 2:
pos = pos + len(self.buf) pos = pos + self.len
self.pos = max(0, pos) self.pos = max(0, pos)
def tell(self): def tell(self):
return self.pos return self.pos
def read(self, n = -1): def read(self, n = -1):
if self.buflist:
self.buf = self.buf + string.joinfields(self.buflist, '')
self.buflist = []
if n < 0: if n < 0:
newpos = len(self.buf) newpos = self.len
else: else:
newpos = min(self.pos+n, len(self.buf)) newpos = min(self.pos+n, self.len)
r = self.buf[self.pos:newpos] r = self.buf[self.pos:newpos]
self.pos = newpos self.pos = newpos
return r return r
def readline(self): def readline(self):
if self.buflist:
self.buf = self.buf + string.joinfields(self.buflist, '')
self.buflist = []
i = string.find(self.buf, '\n', self.pos) i = string.find(self.buf, '\n', self.pos)
if i < 0: if i < 0:
newpos = len(self.buf) newpos = self.len
else: else:
newpos = i+1 newpos = i+1
r = self.buf[self.pos:newpos] r = self.buf[self.pos:newpos]
@ -74,16 +85,28 @@ class StringIO:
return lines return lines
def write(self, s): def write(self, s):
if not s: return if not s: return
if self.pos > len(self.buf): if self.pos > self.len:
self.buf = self.buf + '\0'*(self.pos - len(self.buf)) self.buflist.append('\0'*(self.pos - self.len))
self.len = self.pos
newpos = self.pos + len(s) newpos = self.pos + len(s)
self.buf = self.buf[:self.pos] + s + self.buf[newpos:] if self.pos < self.len:
if self.buflist:
self.buf = self.buf + string.joinfields(self.buflist, '')
self.buflist = []
self.buflist = [self.buf[:self.pos], s, self.buf[newpos:]]
self.buf = ''
else:
self.buflist.append(s)
self.len = newpos
self.pos = newpos self.pos = newpos
def writelines(self, list): def writelines(self, list):
self.write(string.joinfields(list, '')) self.write(string.joinfields(list, ''))
def flush(self): def flush(self):
pass pass
def getvalue(self): def getvalue(self):
if self.buflist:
self.buf = self.buf + string.joinfields(self.buflist, '')
self.buflist = []
return self.buf return self.buf