home *** CD-ROM | disk | FTP | other *** search
/ Maximum CD 2010 November / maximum-cd-2010-11.iso / DiscContents / calibre-0.7.13.msi / file_2806 (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2010-08-06  |  5.9 KB  |  256 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.6)
  3.  
  4.  
  5. try:
  6.     from errno import EINVAL
  7. except ImportError:
  8.     EINVAL = 22
  9.  
  10. __all__ = [
  11.     'StringIO']
  12.  
  13. def _complain_ifclosed(closed):
  14.     if closed:
  15.         raise ValueError, 'I/O operation on closed file'
  16.     closed
  17.  
  18.  
  19. class StringIO:
  20.     
  21.     def __init__(self, buf = ''):
  22.         if not isinstance(buf, basestring):
  23.             buf = str(buf)
  24.         
  25.         self.buf = buf
  26.         self.len = len(buf)
  27.         self.buflist = []
  28.         self.pos = 0
  29.         self.closed = False
  30.         self.softspace = 0
  31.  
  32.     
  33.     def __iter__(self):
  34.         return self
  35.  
  36.     
  37.     def next(self):
  38.         _complain_ifclosed(self.closed)
  39.         r = self.readline()
  40.         if not r:
  41.             raise StopIteration
  42.         r
  43.         return r
  44.  
  45.     
  46.     def close(self):
  47.         if not self.closed:
  48.             self.closed = True
  49.             del self.buf
  50.             del self.pos
  51.         
  52.  
  53.     
  54.     def isatty(self):
  55.         _complain_ifclosed(self.closed)
  56.         return False
  57.  
  58.     
  59.     def seek(self, pos, mode = 0):
  60.         _complain_ifclosed(self.closed)
  61.         if self.buflist:
  62.             self.buf += ''.join(self.buflist)
  63.             self.buflist = []
  64.         
  65.         if mode == 1:
  66.             pos += self.pos
  67.         elif mode == 2:
  68.             pos += self.len
  69.         
  70.         self.pos = max(0, pos)
  71.  
  72.     
  73.     def tell(self):
  74.         _complain_ifclosed(self.closed)
  75.         return self.pos
  76.  
  77.     
  78.     def read(self, n = -1):
  79.         _complain_ifclosed(self.closed)
  80.         if self.buflist:
  81.             self.buf += ''.join(self.buflist)
  82.             self.buflist = []
  83.         
  84.         if n < 0:
  85.             newpos = self.len
  86.         else:
  87.             newpos = min(self.pos + n, self.len)
  88.         r = self.buf[self.pos:newpos]
  89.         self.pos = newpos
  90.         return r
  91.  
  92.     
  93.     def readline(self, length = None):
  94.         _complain_ifclosed(self.closed)
  95.         if self.buflist:
  96.             self.buf += ''.join(self.buflist)
  97.             self.buflist = []
  98.         
  99.         i = self.buf.find('\n', self.pos)
  100.         if i < 0:
  101.             newpos = self.len
  102.         else:
  103.             newpos = i + 1
  104.         if length is not None:
  105.             if self.pos + length < newpos:
  106.                 newpos = self.pos + length
  107.             
  108.         
  109.         r = self.buf[self.pos:newpos]
  110.         self.pos = newpos
  111.         return r
  112.  
  113.     
  114.     def readlines(self, sizehint = 0):
  115.         total = 0
  116.         lines = []
  117.         line = self.readline()
  118.         while line:
  119.             lines.append(line)
  120.             total += len(line)
  121.             if sizehint < sizehint:
  122.                 pass
  123.             elif sizehint <= total:
  124.                 break
  125.             
  126.             line = self.readline()
  127.             continue
  128.             0
  129.         return lines
  130.  
  131.     
  132.     def truncate(self, size = None):
  133.         _complain_ifclosed(self.closed)
  134.         if size is None:
  135.             size = self.pos
  136.         elif size < 0:
  137.             raise IOError(EINVAL, 'Negative size not allowed')
  138.         elif size < self.pos:
  139.             self.pos = size
  140.         
  141.         self.buf = self.getvalue()[:size]
  142.         self.len = size
  143.  
  144.     
  145.     def write(self, s):
  146.         _complain_ifclosed(self.closed)
  147.         if not s:
  148.             return None
  149.         if not isinstance(s, basestring):
  150.             s = str(s)
  151.         
  152.         spos = self.pos
  153.         slen = self.len
  154.         if spos == slen:
  155.             self.buflist.append(s)
  156.             self.len = self.pos = spos + len(s)
  157.             return None
  158.         if spos > slen:
  159.             self.buflist.append('\x00' * (spos - slen))
  160.             slen = spos
  161.         
  162.         newpos = spos + len(s)
  163.         if spos < slen:
  164.             if self.buflist:
  165.                 self.buf += ''.join(self.buflist)
  166.             
  167.             self.buflist = [
  168.                 self.buf[:spos],
  169.                 s,
  170.                 self.buf[newpos:]]
  171.             self.buf = ''
  172.             if newpos > slen:
  173.                 slen = newpos
  174.             
  175.         else:
  176.             self.buflist.append(s)
  177.             slen = newpos
  178.         self.len = slen
  179.         self.pos = newpos
  180.  
  181.     
  182.     def writelines(self, iterable):
  183.         write = self.write
  184.         for line in iterable:
  185.             write(line)
  186.         
  187.  
  188.     
  189.     def flush(self):
  190.         _complain_ifclosed(self.closed)
  191.  
  192.     
  193.     def getvalue(self):
  194.         if self.buflist:
  195.             self.buf += ''.join(self.buflist)
  196.             self.buflist = []
  197.         
  198.         return self.buf
  199.  
  200.  
  201.  
  202. def test():
  203.     import sys
  204.     if sys.argv[1:]:
  205.         file = sys.argv[1]
  206.     else:
  207.         file = '/etc/passwd'
  208.     lines = open(file, 'r').readlines()
  209.     text = open(file, 'r').read()
  210.     f = StringIO()
  211.     for line in lines[:-2]:
  212.         f.write(line)
  213.     
  214.     f.writelines(lines[-2:])
  215.     if f.getvalue() != text:
  216.         raise RuntimeError, 'write failed'
  217.     f.getvalue() != text
  218.     length = f.tell()
  219.     print 'File length =', length
  220.     f.seek(len(lines[0]))
  221.     f.write(lines[1])
  222.     f.seek(0)
  223.     print 'First line =', repr(f.readline())
  224.     print 'Position =', f.tell()
  225.     line = f.readline()
  226.     print 'Second line =', repr(line)
  227.     f.seek(-len(line), 1)
  228.     line2 = f.read(len(line))
  229.     if line != line2:
  230.         raise RuntimeError, 'bad result after seek back'
  231.     line != line2
  232.     f.seek(len(line2), 1)
  233.     list = f.readlines()
  234.     line = list[-1]
  235.     f.seek(f.tell() - len(line))
  236.     line2 = f.read()
  237.     if line != line2:
  238.         raise RuntimeError, 'bad result after seek back from EOF'
  239.     line != line2
  240.     print 'Read', len(list), 'more lines'
  241.     print 'File length =', f.tell()
  242.     if f.tell() != length:
  243.         raise RuntimeError, 'bad length'
  244.     f.tell() != length
  245.     f.truncate(length / 2)
  246.     f.seek(0, 2)
  247.     print 'Truncated length =', f.tell()
  248.     if f.tell() != length / 2:
  249.         raise RuntimeError, 'truncate did not adjust length'
  250.     f.tell() != length / 2
  251.     f.close()
  252.  
  253. if __name__ == '__main__':
  254.     test()
  255.  
  256.