Quantcast
Channel: Planet Plone - Where Developers And Integrators Write
Viewing all articles
Browse latest Browse all 3535

Zopatista - Martijn Pieters: Easy in-place file rewriting

$
0
0

Using a context manager to allow painless rewriting of files

Whenever you need to process a file in-place, transforming the contents and writing it out again in the same location, you can reach out for the fileinput module and use its inplace option:

importfileinputforlineinfileinput.input(somefilename,inplace=True):line='additional information '+line.rstrip('\n')printline

There are a few problems with the fileinput module, however. My biggest nitpick with the module is that it has an API that relies heavily on globals; fileinput.input() creates a global fileinput.FileInput() object, which other functions in the module then access. You can of course ignore all that and reach straight for the fileinput.FileInput() constructor, but fileinput.input() is presented as the main API entrypoint.

The other is that the in-place modus hijacks sys.stdout as the means to write back to the replacement file. Obstensibly this is to make it easy to use a print statement, but then you have to remember to remove the newline from the lines read from the old file.

Last, but not least, the fileinput version in the Python 3 standard library does not support specifying an encoding, error mode or newline handling. You can open the input file in binary mode, but output is always handled in text mode. This greatly diminishes the usefulness of this library.

So I wrote my own replacement, using the excellent @contextlib.contextmanager decorator. This version works on both Python 2 and 3, relying on io.open() to remain compatible between Python versions:

fromcontextlibimportcontextmanagerimportioimportos@contextmanagerdefinplace(filename,mode='r',buffering=-1,encoding=None,errors=None,newline=None,backup_extension=None):"""Allow for a file to be replaced with new content.    yields a tuple of (readable, writable) file objects, where writable    replaces readable.    If an exception occurs, the old file is restored, removing the    written data.    mode should *not* use 'w', 'a' or '+'; only read-only-modes are supported."""# move existing file to backup, create new file with same permissions# borrowed extensively from the fileinput moduleifset(mode).intersection('wa+'):raiseValueError('Only read-only file modes can be used')backupfilename=filename+(backup_extensionoros.extsep+'bak')try:os.unlink(backupfilename)exceptos.error:passos.rename(filename,backupfilename)readable=io.open(backupfilename,mode,buffering=buffering,encoding=encoding,errors=errors,newline=newline)try:perm=os.fstat(readable.fileno()).st_modeexceptOSError:writable=open(filename,'w'+mode.replace('r',''),buffering=buffering,encoding=encoding,errors=errors,newline=newline)else:os_mode=os.O_CREAT|os.O_WRONLY|os.O_TRUNCifhasattr(os,'O_BINARY'):os_mode|=os.O_BINARYfd=os.open(filename,os_mode,perm)writable=io.open(fd,"w"+mode.replace('r',''),buffering=buffering,encoding=encoding,errors=errors,newline=newline)try:ifhasattr(os,'chmod'):os.chmod(filename,perm)exceptOSError:passtry:yieldreadable,writableexceptException:# move backup backtry:os.unlink(filename)exceptos.error:passos.rename(backupfilename,filename)raisefinally:readable.close()writable.close()try:os.unlink(backupfilename)exceptos.error:pass

This context manager deliberately focuses on just one file, and ignores sys.stdin, unline the fileinput module. It is aimed squarly at just replacing a file in-place.

Usage example, in Python 2, with the CSV module:

importcsvwithinplace(csvfilename,'rb')as(infh,outfh):reader=csv.reader(infh)writer=csv.writer(outfh)forrowinreader:row+=['new','columns']writer.writerow(row)

and the Python 3 version:

importcsvwithinplace(csvfilename,'r',newline='')as(infh,outfh):reader=csv.reader(infh)writer=csv.writer(outfh)forrowinreader:row+=['new','columns']writer.writerow(row)

Viewing all articles
Browse latest Browse all 3535

Trending Articles