0

Add force_writable argument to mirror_dir

Fixes #1611.
This commit is contained in:
Nicolas F
2019-07-12 23:37:35 +02:00
parent 70c64cd4c5
commit a422270666
2 changed files with 11 additions and 5 deletions

View File

@@ -62,7 +62,7 @@ def does_rename_work(dir_to_test):
return renameworks
## useful recursive copy, that ignores common OS cruft
def mirror_dir(src, dst, entities=None, capabilities=default_caps):
def mirror_dir(src, dst, entities=None, capabilities=default_caps, force_writable=False):
'''copies all of the entities from src to dst'''
chmod_works = capabilities.get("chmod_works")
if not os.path.exists(dst):
@@ -80,11 +80,14 @@ def mirror_dir(src, dst, entities=None, capabilities=default_caps):
continue
if os.path.isdir(os.path.join(src,entry)):
mirror_dir(os.path.join(src, entry), os.path.join(dst, entry), capabilities=capabilities)
mirror_dir(os.path.join(src, entry), os.path.join(dst, entry), capabilities=capabilities, force_writable=force_writable)
elif os.path.isfile(os.path.join(src,entry)):
try:
if chmod_works:
shutil.copy(os.path.join(src, entry), os.path.join(dst, entry))
if force_writable: # for shitty neckbeard ware
dst_stat = os.stat(os.path.join(dst, entry))
os.chmod(os.path.join(dst, entry), dst_stat.st_mode | stat.S_IWUSR)
else:
shutil.copyfile(os.path.join(src, entry), os.path.join(dst, entry))
except IOError as outer:
@@ -99,6 +102,9 @@ def mirror_dir(src, dst, entities=None, capabilities=default_caps):
# try again; if this stills throws an error, let it propagate up
if chmod_works:
shutil.copy(os.path.join(src, entry), os.path.join(dst, entry))
if force_writable:
dst_stat = os.stat(os.path.join(dst, entry))
os.chmod(os.path.join(dst, entry), dst_stat.st_mode | stat.S_IWUSR)
else:
shutil.copyfile(os.path.join(src, entry), os.path.join(dst, entry))