wrap_write

wrap_write_bar.txt

dfedf
fqe
as

wrap_write_bar3.txt

dfedf
fqe
as

wrap_write_foo.txt

dfedf
fqe
as

wrap_write_wrap_write.py

from mock import patch


def no_write_option_decorator(f):
    builtin_open = open

    def fake_write(*args):
        pass

    def patch_open(path, options):
        ret = builtin_open(path, options)
        ret.write = fake_write
        return ret

    def wrapped(src, dst, can_write=True):
        with patch("builtins.open", open if can_write else patch_open):
            return f(src, dst)

    return wrapped


@no_write_option_decorator
def copy(src, dst):
    with open(src, 'r') as foo:
        with open(dst, 'w') as bar:
            bar.write(foo.read())



if __name__ == '__main__':
    copy('foo.txt', 'bar.txt')
    copy('foo.txt', 'bar2.txt', False)
    copy('foo.txt', 'bar3.txt')