traverseda
/
cpython
1
0
Fork 0
Code Issues Pull Requests 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:
def __init__(self, buf = ''):
self.buf = buf
self.len = len(buf)
self.buflist = []
self.pos = 0
self.closed = 0
self.softspace = 0
@ -41,25 +43,34 @@ class StringIO:
def isatty(self):
return 0
def seek(self, pos, mode = 0):
if self.buflist:
self.buf = self.buf + string.joinfields(self.buflist, '')
self.buflist = []
if mode == 1:
pos = pos + self.pos
elif mode == 2:
pos = pos + len(self.buf)
pos = pos + self.len
self.pos = max(0, pos)
def tell(self):
return self.pos
def read(self, n = -1):
if self.buflist:
self.buf = self.buf + string.joinfields(self.buflist, '')
self.buflist = []
if n < 0:
newpos = len(self.buf)
newpos = self.len
else:
newpos = min(self.pos+n, len(self.buf))
newpos = min(self.pos+n, self.len)
r = self.buf[self.pos:newpos]
self.pos = newpos
return r
def readline(self):
if self.buflist:
self.buf = self.buf + string.joinfields(self.buflist, '')
self.buflist = []
i = string.find(self.buf, '\n', self.pos)
if i < 0:
newpos = len(self.buf)
newpos = self.len
else:
newpos = i+1
r = self.buf[self.pos:newpos]
@ -74,16 +85,28 @@ class StringIO:
return lines
def write(self, s):
if not s: return
if self.pos > len(self.buf):
self.buf = self.buf + '\0'*(self.pos - len(self.buf))
if self.pos > self.len:
self.buflist.append('\0'*(self.pos - self.len))
self.len = self.pos
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
def writelines(self, list):
self.write(string.joinfields(list, ''))
def flush(self):
pass
def getvalue(self):
if self.buflist:
self.buf = self.buf + string.joinfields(self.buflist, '')
self.buflist = []
return self.buf