Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Method for dry runs?

Tags:

python

at the moment my python code often looks like this:

...
if not dry_run:
    result = shutil.copyfile(...)
else:
    print "   DRY-RUN: shutil.copyfile(...) "
...

I now think about writting something like a dry runner method:

def dry_runner(cmd, dry_run, message, before="", after=""):
    if dry_run:
        print before + "DRY-RUN: " + message + after
     # return execute(cmd)

But the cmd will be executed first and the result is given to dry_runner method.

How can I code such a method the pythonic way?

like image 755
Marten Bauer Avatar asked Jul 19 '10 09:07

Marten Bauer


2 Answers

You could use this generic wrapper function:

def execute(func, *args):
    print 'before', func
    if not dry:
        func(*args)
    print 'after', func

>>> execute(shutil.copyfile, 'src', 'dst')
like image 88
compie Avatar answered Oct 27 '22 22:10

compie


This isn't perfect in its display, but the functionality works. Hopefully this is clear enough:

dry = True

def dryrun(f):
    def wrapper(*args, **kwargs):
        if dry:
            print "DRY RUN: %s(%s)" % (f.__name__, 
                                       ','.join(list(args) + ["%s=%s" % (k, v) for (k, v) in kwargs.iteritems()])) 
        else:
            f(*args, **kwargs)
    return wrapper

import shutil
copyfile = dryrun(shutil.copyfile)

copyfile('a', 'b')
like image 26
xorsyst Avatar answered Oct 28 '22 00:10

xorsyst