home *** CD-ROM | disk | FTP | other *** search
/ GameStar 2006 January / Gamestar_80_2006-01_dvd.iso / Dema / Civilization4 / data1.cab / Civ4DemoComponent / Assets / Python / System / fileinput.pyc (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2005-11-09  |  12.3 KB  |  434 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.4)
  3.  
  4. '''Helper class to quickly write a loop over all standard input files.
  5.  
  6. Typical use is:
  7.  
  8.     import fileinput
  9.     for line in fileinput.input():
  10.         process(line)
  11.  
  12. This iterates over the lines of all files listed in sys.argv[1:],
  13. defaulting to sys.stdin if the list is empty.  If a filename is \'-\' it
  14. is also replaced by sys.stdin.  To specify an alternative list of
  15. filenames, pass it as the argument to input().  A single file name is
  16. also allowed.
  17.  
  18. Functions filename(), lineno() return the filename and cumulative line
  19. number of the line that has just been read; filelineno() returns its
  20. line number in the current file; isfirstline() returns true iff the
  21. line just read is the first line of its file; isstdin() returns true
  22. iff the line was read from sys.stdin.  Function nextfile() closes the
  23. current file so that the next iteration will read the first line from
  24. the next file (if any); lines not read from the file will not count
  25. towards the cumulative line count; the filename is not changed until
  26. after the first line of the next file has been read.  Function close()
  27. closes the sequence.
  28.  
  29. Before any lines have been read, filename() returns None and both line
  30. numbers are zero; nextfile() has no effect.  After all lines have been
  31. read, filename() and the line number functions return the values
  32. pertaining to the last line read; nextfile() has no effect.
  33.  
  34. All files are opened in text mode.  If an I/O error occurs during
  35. opening or reading a file, the IOError exception is raised.
  36.  
  37. If sys.stdin is used more than once, the second and further use will
  38. return no lines, except perhaps for interactive use, or if it has been
  39. explicitly reset (e.g. using sys.stdin.seek(0)).
  40.  
  41. Empty files are opened and immediately closed; the only time their
  42. presence in the list of filenames is noticeable at all is when the
  43. last file opened is empty.
  44.  
  45. It is possible that the last line of a file doesn\'t end in a newline
  46. character; otherwise lines are returned including the trailing
  47. newline.
  48.  
  49. Class FileInput is the implementation; its methods filename(),
  50. lineno(), fileline(), isfirstline(), isstdin(), nextfile() and close()
  51. correspond to the functions in the module.  In addition it has a
  52. readline() method which returns the next input line, and a
  53. __getitem__() method which implements the sequence behavior.  The
  54. sequence must be accessed in strictly sequential order; sequence
  55. access and readline() cannot be mixed.
  56.  
  57. Optional in-place filtering: if the keyword argument inplace=1 is
  58. passed to input() or to the FileInput constructor, the file is moved
  59. to a backup file and standard output is directed to the input file.
  60. This makes it possible to write a filter that rewrites its input file
  61. in place.  If the keyword argument backup=".<some extension>" is also
  62. given, it specifies the extension for the backup file, and the backup
  63. file remains around; by default, the extension is ".bak" and it is
  64. deleted when the output file is closed.  In-place filtering is
  65. disabled when standard input is read.  XXX The current implementation
  66. does not work for MS-DOS 8+3 filesystems.
  67.  
  68. Performance: this module is unfortunately one of the slower ways of
  69. processing large numbers of input lines.  Nevertheless, a significant
  70. speed-up has been obtained by using readlines(bufsize) instead of
  71. readline().  A new keyword argument, bufsize=N, is present on the
  72. input() function and the FileInput() class to override the default
  73. buffer size.
  74.  
  75. XXX Possible additions:
  76.  
  77. - optional getopt argument processing
  78. - specify open mode (\'r\' or \'rb\')
  79. - fileno()
  80. - isatty()
  81. - read(), read(size), even readlines()
  82.  
  83. '''
  84. import sys
  85. import os
  86. __all__ = [
  87.     'input',
  88.     'close',
  89.     'nextfile',
  90.     'filename',
  91.     'lineno',
  92.     'filelineno',
  93.     'isfirstline',
  94.     'isstdin',
  95.     'FileInput']
  96. _state = None
  97. DEFAULT_BUFSIZE = 8 * 1024
  98.  
  99. def input(files = None, inplace = 0, backup = '', bufsize = 0):
  100.     '''input([files[, inplace[, backup]]])
  101.  
  102.     Create an instance of the FileInput class. The instance will be used
  103.     as global state for the functions of this module, and is also returned
  104.     to use during iteration. The parameters to this function will be passed
  105.     along to the constructor of the FileInput class.
  106.     '''
  107.     global _state
  108.     if _state and _state._file:
  109.         raise RuntimeError, 'input() already active'
  110.     
  111.     _state = FileInput(files, inplace, backup, bufsize)
  112.     return _state
  113.  
  114.  
  115. def close():
  116.     '''Close the sequence.'''
  117.     global _state
  118.     state = _state
  119.     _state = None
  120.     if state:
  121.         state.close()
  122.     
  123.  
  124.  
  125. def nextfile():
  126.     '''
  127.     Close the current file so that the next iteration will read the first
  128.     line from the next file (if any); lines not read from the file will
  129.     not count towards the cumulative line count. The filename is not
  130.     changed until after the first line of the next file has been read.
  131.     Before the first line has been read, this function has no effect;
  132.     it cannot be used to skip the first file. After the last line of the
  133.     last file has been read, this function has no effect.
  134.     '''
  135.     if not _state:
  136.         raise RuntimeError, 'no active input()'
  137.     
  138.     return _state.nextfile()
  139.  
  140.  
  141. def filename():
  142.     '''
  143.     Return the name of the file currently being read.
  144.     Before the first line has been read, returns None.
  145.     '''
  146.     if not _state:
  147.         raise RuntimeError, 'no active input()'
  148.     
  149.     return _state.filename()
  150.  
  151.  
  152. def lineno():
  153.     '''
  154.     Return the cumulative line number of the line that has just been read.
  155.     Before the first line has been read, returns 0. After the last line
  156.     of the last file has been read, returns the line number of that line.
  157.     '''
  158.     if not _state:
  159.         raise RuntimeError, 'no active input()'
  160.     
  161.     return _state.lineno()
  162.  
  163.  
  164. def filelineno():
  165.     '''
  166.     Return the line number in the current file. Before the first line
  167.     has been read, returns 0. After the last line of the last file has
  168.     been read, returns the line number of that line within the file.
  169.     '''
  170.     if not _state:
  171.         raise RuntimeError, 'no active input()'
  172.     
  173.     return _state.filelineno()
  174.  
  175.  
  176. def isfirstline():
  177.     '''
  178.     Returns true the line just read is the first line of its file,
  179.     otherwise returns false.
  180.     '''
  181.     if not _state:
  182.         raise RuntimeError, 'no active input()'
  183.     
  184.     return _state.isfirstline()
  185.  
  186.  
  187. def isstdin():
  188.     '''
  189.     Returns true if the last line was read from sys.stdin,
  190.     otherwise returns false.
  191.     '''
  192.     if not _state:
  193.         raise RuntimeError, 'no active input()'
  194.     
  195.     return _state.isstdin()
  196.  
  197.  
  198. class FileInput:
  199.     '''class FileInput([files[, inplace[, backup]]])
  200.  
  201.     Class FileInput is the implementation of the module; its methods
  202.     filename(), lineno(), fileline(), isfirstline(), isstdin(), nextfile()
  203.     and close() correspond to the functions of the same name in the module.
  204.     In addition it has a readline() method which returns the next
  205.     input line, and a __getitem__() method which implements the
  206.     sequence behavior. The sequence must be accessed in strictly
  207.     sequential order; random access and readline() cannot be mixed.
  208.     '''
  209.     
  210.     def __init__(self, files = None, inplace = 0, backup = '', bufsize = 0):
  211.         if type(files) == type(''):
  212.             files = (files,)
  213.         elif files is None:
  214.             files = sys.argv[1:]
  215.         
  216.         if not files:
  217.             files = ('-',)
  218.         else:
  219.             files = tuple(files)
  220.         self._files = files
  221.         self._inplace = inplace
  222.         self._backup = backup
  223.         if not bufsize:
  224.             pass
  225.         self._bufsize = DEFAULT_BUFSIZE
  226.         self._savestdout = None
  227.         self._output = None
  228.         self._filename = None
  229.         self._lineno = 0
  230.         self._filelineno = 0
  231.         self._file = None
  232.         self._isstdin = False
  233.         self._backupfilename = None
  234.         self._buffer = []
  235.         self._bufindex = 0
  236.  
  237.     
  238.     def __del__(self):
  239.         self.close()
  240.  
  241.     
  242.     def close(self):
  243.         self.nextfile()
  244.         self._files = ()
  245.  
  246.     
  247.     def __iter__(self):
  248.         return self
  249.  
  250.     
  251.     def next(self):
  252.         
  253.         try:
  254.             line = self._buffer[self._bufindex]
  255.         except IndexError:
  256.             pass
  257.  
  258.         self._bufindex += 1
  259.         self._lineno += 1
  260.         self._filelineno += 1
  261.         return line
  262.         line = self.readline()
  263.         return line
  264.  
  265.     
  266.     def __getitem__(self, i):
  267.         if i != self._lineno:
  268.             raise RuntimeError, 'accessing lines out of order'
  269.         
  270.         
  271.         try:
  272.             return self.next()
  273.         except StopIteration:
  274.             raise IndexError, 'end of input reached'
  275.  
  276.  
  277.     
  278.     def nextfile(self):
  279.         savestdout = self._savestdout
  280.         self._savestdout = 0
  281.         if savestdout:
  282.             sys.stdout = savestdout
  283.         
  284.         output = self._output
  285.         self._output = 0
  286.         if output:
  287.             output.close()
  288.         
  289.         file = self._file
  290.         self._file = 0
  291.         if file and not (self._isstdin):
  292.             file.close()
  293.         
  294.         backupfilename = self._backupfilename
  295.         self._backupfilename = 0
  296.         if backupfilename and not (self._backup):
  297.             
  298.             try:
  299.                 os.unlink(backupfilename)
  300.             except OSError:
  301.                 pass
  302.             except:
  303.                 None<EXCEPTION MATCH>OSError
  304.             
  305.  
  306.         None<EXCEPTION MATCH>OSError
  307.         self._isstdin = False
  308.         self._buffer = []
  309.         self._bufindex = 0
  310.  
  311.     
  312.     def readline(self):
  313.         
  314.         try:
  315.             line = self._buffer[self._bufindex]
  316.         except IndexError:
  317.             pass
  318.  
  319.         self._bufindex += 1
  320.         self._lineno += 1
  321.         self._filelineno += 1
  322.         return line
  323.         if not self._file:
  324.             self._filename = self._files[0]
  325.             self._files = self._files[1:]
  326.             self._filelineno = 0
  327.             self._file = None
  328.             self._isstdin = False
  329.             self._backupfilename = 0
  330.             if self._filename == '-':
  331.                 self._filename = '<stdin>'
  332.                 self._file = sys.stdin
  333.                 self._isstdin = True
  334.             elif self._inplace:
  335.                 if not self._backup:
  336.                     pass
  337.                 self._backupfilename = self._filename + os.extsep + 'bak'
  338.                 
  339.                 try:
  340.                     os.unlink(self._backupfilename)
  341.                 except os.error:
  342.                     self if not self._files else self
  343.                     self if not self._files else self
  344.                 except:
  345.                     self if not self._files else self
  346.  
  347.                 os.rename(self._filename, self._backupfilename)
  348.                 self._file = open(self._backupfilename, 'r')
  349.                 
  350.                 try:
  351.                     perm = os.fstat(self._file.fileno()).st_mode
  352.                 except OSError:
  353.                     self if not self._files else self
  354.                     self if not self._files else self
  355.                     self._output = open(self._filename, 'w')
  356.                 except:
  357.                     self if not self._files else self
  358.  
  359.                 fd = os.open(self._filename, os.O_CREAT | os.O_WRONLY | os.O_TRUNC, perm)
  360.                 self._output = os.fdopen(fd, 'w')
  361.                 
  362.                 try:
  363.                     if hasattr(os, 'chmod'):
  364.                         os.chmod(self._filename, perm)
  365.                 except OSError:
  366.                     self if not self._files else self
  367.                     self if not self._files else self
  368.                 except:
  369.                     self if not self._files else self
  370.  
  371.                 self._savestdout = sys.stdout
  372.                 sys.stdout = self._output
  373.             else:
  374.                 self._file = open(self._filename, 'r')
  375.         
  376.         self._buffer = self._file.readlines(self._bufsize)
  377.         self._bufindex = 0
  378.         if not self._buffer:
  379.             self.nextfile()
  380.         
  381.         return self.readline()
  382.  
  383.     
  384.     def filename(self):
  385.         return self._filename
  386.  
  387.     
  388.     def lineno(self):
  389.         return self._lineno
  390.  
  391.     
  392.     def filelineno(self):
  393.         return self._filelineno
  394.  
  395.     
  396.     def isfirstline(self):
  397.         return self._filelineno == 1
  398.  
  399.     
  400.     def isstdin(self):
  401.         return self._isstdin
  402.  
  403.  
  404.  
  405. def _test():
  406.     import getopt
  407.     inplace = 0
  408.     backup = 0
  409.     (opts, args) = getopt.getopt(sys.argv[1:], 'ib:')
  410.     for o, a in opts:
  411.         if o == '-i':
  412.             inplace = 1
  413.         
  414.         if o == '-b':
  415.             backup = a
  416.             continue
  417.     
  418.     for line in input(args, inplace = inplace, backup = backup):
  419.         if line[-1:] == '\n':
  420.             line = line[:-1]
  421.         
  422.         if line[-1:] == '\r':
  423.             line = line[:-1]
  424.         
  425.         if not isfirstline() or '*':
  426.             pass
  427.         print '%d: %s[%d]%s %s' % (lineno(), filename(), filelineno(), '', line)
  428.     
  429.     print '%d: %s[%d]' % (lineno(), filename(), filelineno())
  430.  
  431. if __name__ == '__main__':
  432.     _test()
  433.  
  434.