mirror of
https://github.com/thegeeklab/ansible-later.git
synced 2024-11-30 00:30:35 +00:00
67 lines
2.5 KiB
Python
67 lines
2.5 KiB
Python
|
# lamosity needed to make things reliable on Windows :-(
|
||
|
|
# (borrowed from Python's test_support.py)
|
||
|
|
import errno
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
import sys
|
||
|
|
import time
|
||
|
|
import warnings
|
||
|
|
|
||
|
|
if sys.platform.startswith("win"): # pragma: no cover
|
||
|
|
def _waitfor(func, pathname, waitall=False):
|
||
|
|
# Perform the operation
|
||
|
|
func(pathname)
|
||
|
|
# Now setup the wait loop
|
||
|
|
if waitall:
|
||
|
|
dirname = pathname
|
||
|
|
else:
|
||
|
|
dirname, name = os.path.split(pathname)
|
||
|
|
dirname = dirname or '.'
|
||
|
|
# Check for `pathname` to be removed from the filesystem.
|
||
|
|
# The exponential backoff of the timeout amounts to a total
|
||
|
|
# of ~1 second after which the deletion is probably an error
|
||
|
|
# anyway.
|
||
|
|
# Testing on a i7@4.3GHz shows that usually only 1 iteration is
|
||
|
|
# required when contention occurs.
|
||
|
|
timeout = 0.001
|
||
|
|
while timeout < 1.0: # pragma: no branch
|
||
|
|
# Note we are only testing for the existence of the file(s) in
|
||
|
|
# the contents of the directory regardless of any security or
|
||
|
|
# access rights. If we have made it this far, we have sufficient
|
||
|
|
# permissions to do that much using Python's equivalent of the
|
||
|
|
# Windows API FindFirstFile.
|
||
|
|
# Other Windows APIs can fail or give incorrect results when
|
||
|
|
# dealing with files that are pending deletion.
|
||
|
|
L = os.listdir(dirname)
|
||
|
|
if not (L if waitall else name in L): # pragma: no branch
|
||
|
|
return
|
||
|
|
# Increase the timeout and try again
|
||
|
|
time.sleep(timeout) # pragma: no cover
|
||
|
|
timeout *= 2 # pragma: no cover
|
||
|
|
warnings.warn('tests may fail, delete still pending for '
|
||
|
|
+ pathname, # pragma: no cover
|
||
|
|
RuntimeWarning, stacklevel=4)
|
||
|
|
|
||
|
|
def _rmtree(path):
|
||
|
|
def _rmtree_inner(path):
|
||
|
|
for name in os.listdir(path):
|
||
|
|
fullname = os.path.join(path, name)
|
||
|
|
if os.path.isdir(fullname):
|
||
|
|
_waitfor(_rmtree_inner, fullname, waitall=True)
|
||
|
|
os.rmdir(fullname)
|
||
|
|
else:
|
||
|
|
os.unlink(fullname)
|
||
|
|
_waitfor(_rmtree_inner, path, waitall=True)
|
||
|
|
_waitfor(os.rmdir, path)
|
||
|
|
else:
|
||
|
|
_rmtree = shutil.rmtree
|
||
|
|
|
||
|
|
|
||
|
|
def rmtree(path):
|
||
|
|
try:
|
||
|
|
_rmtree(path)
|
||
|
|
except OSError as e: # pragma: no cover
|
||
|
|
# Unix returns ENOENT, Windows returns ESRCH.
|
||
|
|
if e.errno not in (errno.ENOENT, errno.ESRCH): # pragma: no branch
|
||
|
|
raise
|