B /g^!9c @sdZddlZddlZddddddd d d d d dg Zdad!ddZddZddZddZddZ ddZ dd Z dd Z dd Z Gdd d Zdd Zd"ddZddZed kredS)#a Helper class to quickly write a loop over all standard input files. Typical use is: import fileinput for line in fileinput.input(): process(line) This iterates over the lines of all files listed in sys.argv[1:], defaulting to sys.stdin if the list is empty. If a filename is '-' it is also replaced by sys.stdin and the optional arguments mode and openhook are ignored. To specify an alternative list of filenames, pass it as the argument to input(). A single file name is also allowed. Functions filename(), lineno() return the filename and cumulative line number of the line that has just been read; filelineno() returns its line number in the current file; isfirstline() returns true iff the line just read is the first line of its file; isstdin() returns true iff the line was read from sys.stdin. Function nextfile() closes the current file so that the next iteration will read the first line from the next file (if any); lines not read from the file will not count towards the cumulative line count; the filename is not changed until after the first line of the next file has been read. Function close() closes the sequence. Before any lines have been read, filename() returns None and both line numbers are zero; nextfile() has no effect. After all lines have been read, filename() and the line number functions return the values pertaining to the last line read; nextfile() has no effect. All files are opened in text mode by default, you can override this by setting the mode parameter to input() or FileInput.__init__(). If an I/O error occurs during opening or reading a file, the OSError exception is raised. If sys.stdin is used more than once, the second and further use will return no lines, except perhaps for interactive use, or if it has been explicitly reset (e.g. using sys.stdin.seek(0)). Empty files are opened and immediately closed; the only time their presence in the list of filenames is noticeable at all is when the last file opened is empty. It is possible that the last line of a file doesn't end in a newline character; otherwise lines are returned including the trailing newline. Class FileInput is the implementation; its methods filename(), lineno(), fileline(), isfirstline(), isstdin(), nextfile() and close() correspond to the functions in the module. In addition it has a readline() method which returns the next input line, and a __getitem__() method which implements the sequence behavior. The sequence must be accessed in strictly sequential order; sequence access and readline() cannot be mixed. Optional in-place filtering: if the keyword argument inplace=1 is passed to input() or to the FileInput constructor, the file is moved to a backup file and standard output is directed to the input file. This makes it possible to write a filter that rewrites its input file in place. If the keyword argument backup="." is also given, it specifies the extension for the backup file, and the backup file remains around; by default, the extension is ".bak" and it is deleted when the output file is closed. In-place filtering is disabled when standard input is read. XXX The current implementation does not work for MS-DOS 8+3 filesystems. XXX Possible additions: - optional getopt argument processing - isatty() - read(), read(size), even readlines() Ninputclosenextfilefilenamelineno filelinenofileno isfirstlineisstdin FileInputhook_compressed hook_encodedFrcCs(trtjrtdt||||||atS)aReturn an instance of the FileInput class, which can be iterated. The parameters are passed to the constructor of the FileInput class. The returned instance, in addition to being an iterator, keeps global state for the functions of this module,. zinput() already active)_state_file RuntimeErrorr )filesinplacebackupbufsizemodeopenhookr!/usr/lib64/python3.7/fileinput.pyrSs cCst}da|r|dS)zClose the sequence.N)rr)ZstaterrrrascCts tdtS)a Close the current file so that the next iteration will read the first line from the next file (if any); lines not read from the file will not count towards the cumulative line count. The filename is not changed until after the first line of the next file has been read. Before the first line has been read, this function has no effect; it cannot be used to skip the first file. After the last line of the last file has been read, this function has no effect. no active input())rrrrrrrris cCr)zr Return the name of the file currently being read. Before the first line has been read, returns None. r)rrrrrrrrwcCr)z Return the cumulative line number of the line that has just been read. Before the first line has been read, returns 0. After the last line of the last file has been read, returns the line number of that line. r)rrrrrrrrcCr)z Return the line number in the current file. Before the first line has been read, returns 0. After the last line of the last file has been read, returns the line number of that line within the file. r)rrrrrrrrrcCr)zg Return the file number of the current file. When no file is currently opened, returns -1. r)rrrrrrrrrcCr)ze Returns true the line just read is the first line of its file, otherwise returns false. r)rrrrrrrrrcCr)z] Returns true if the last line was read from sys.stdin, otherwise returns false. r)rrr rrrrr rc@seZdZdZd)ddZd d Zd d Zd dZddZddZ ddZ ddZ ddZ ddZ ddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(ZdS)*r a?FileInput([files[, inplace[, backup[, bufsize, [, mode[, openhook]]]]]]) Class FileInput is the implementation of the module; its methods filename(), lineno(), fileline(), isfirstline(), isstdin(), fileno(), nextfile() and close() correspond to the functions of the same name in the module. In addition it has a readline() method which returns the next input line, and a __getitem__() method which implements the sequence behavior. The sequence must be accessed in strictly sequential order; random access and readline() cannot be mixed. NFr rrcCst|tr|f}nBt|tjr,t|f}n(|dkrBtjdd}|sLd}nt|}||_||_ ||_ |rddl }|j dt ddd|_d|_d|_d|_d|_d|_d|_d|_|dkrtd d |krddl }| d t d||_|r |rtd t|s td ||_dS)N)-rz!bufsize is deprecated and ignoredi)Z stacklevelF)rZrUUZrbz=FileInput opening mode must be one of 'r', 'rU', 'U' and 'rb'r z'U' mode is deprecatedz4FileInput cannot use an opening hook in inplace modez#FileInput openhook must be callable)Z isinstanceZstrosZPathLikefspathsysargvZtuple_files_inplace_backupwarningsZwarnZDeprecationWarning _savestdout_output _filename _startlineno _filelinenor_isstdin_backupfilename ValueError_modeZcallable _openhook)selfrrrrrrr(rrr__init__sJ    zFileInput.__init__cC |dSNrr3rrr__del__zFileInput.__del__cCsz |Wdd|_XdS)Nr)rr%r8rrrrs zFileInput.closecC|Sr6rr8rrr __enter__r:zFileInput.__enter__cCr5r6r7)r3ZtypeZvalueZ tracebackrrr__exit__r:zFileInput.__exit__cCr;r6rr8rrr__iter__r:zFileInput.__iter__cCs:x4|}|r |jd7_|S|js*t|qWdSNr) _readliner-r StopIterationrr3linerrr__next__zFileInput.__next__cCs>||krtdy|Stk r8tdYnXdS)Nzaccessing lines out of orderzend of input reached)rrrDrAZ IndexError)r3Zirrr __getitem__s  zFileInput.__getitem__cCs|j}d|_|r|t_|j}d|_z|r0|Wd|j}d|_y|`Wntk r\YnXz|rr|jsr|Wd|j }d|_ |r|j syt |Wnt k rYnXd|_XXdS)NF)r)r#stdoutr*rrr@ZAttributeErrorr.r/r'r!unlinkOSError)r3Z savestdoutZoutputZfileZbackupfilenamerrrr s4    zFileInput.nextfilecCs:x4|}|r |jd7_|S|js*|S|qWdSr?)r@r-rrrBrrrreadline+rEzFileInput.readlinecCs|jsd|jkrdSdS|jd|_|jdd|_||_d|_d|_d|_d|_|jdkrd|_d|jkrt t j d t j |_nt j |_d |_n\|j rt |j|jpd |_yt |jWntk rYnXt |j|jt|j|j|_yt |jj}Wn$tk r6t|jd |_Yn~Xt jt jBt jB}tt d r`|t jO}t |j||}t |d |_ytt drt |j|Wntk rYnXt j|_ |jt _n,|j!r|!|j|j|_nt|j|j|_|jj"|_#|#S)NZbsr rrFrzZbufferTz.bakZwO_BINARYchmod)$r%r1r+rr,r-rr.r/Zgetattrr#Zstdinr&r!r"r'rHrIZrenameopenZfstatrZst_moder*ZO_CREATZO_WRONLYZO_TRUNCZhasattrrKZfdopenrLrGr)r2rJr@)r3ZpermrZfdrrrr@6s\           zFileInput._readlinecC|jSr6)r+r8rrrrpr:zFileInput.filenamecCs |j|jSr6)r,r-r8rrrrsr:zFileInput.linenocCrNr6r-r8rrrrvr:zFileInput.filelinenocCs0|jr(y |jStk r$dSXndSdS)N)rrr0r8rrrrys  zFileInput.filenocCs |jdkSr?rOr8rrrrr:zFileInput.isfirstlinecCrNr6)r.r8rrrr r:zFileInput.isstdinNFr rrN)__name__Z __module__Z __qualname____doc__r4r9rr<r=r>rDrFrrJr@rrrrrr rrrrr s&  +  : cCsVtj|d}|dkr,ddl}|||S|dkrHddl}|||St||SdS)Nrz.gzrz.bz2)r!ZpathZsplitextgziprMbz2ZBZ2File)rrZextrTrUrrrr s  csfdd}|S)Ncst||dS)Nencodingerrors)rM)rrrVrrrr:zhook_encoded..openhookr)rWrXrrrVrr scCsddl}d}d}|tjddd\}}x(|D] \}}|dkrDd}|dkr0|}q0Wxtt|||dD]b}|d dd kr|dd }|d dd kr|dd }td ttttrd pd|fqdWtdtttfdS)NrFrzib:z-iTz-b)rrrPZ z z%d: %s[%d]%s %sZ*r z %d: %s[%d]) getoptr#r$rZprintrrrr)rYrrZoptsZargsZoZarCrrr_tests"  rZZ__main__rQr6)rSr#r!Z__all__rrrrrrrrrr r r r rZrRrrrrZIs,         [