Minor cleanup in plat_other

- Add OSError code values
- Use INFO_SUFFIX constant in tests
- Remove old PathLike conversions
This commit is contained in:
Andrew Senetar 2021-08-24 01:00:02 -05:00
parent 7686647389
commit 18e51c0b5a
Signed by: arsenetar
GPG Key ID: C63300DCE48AB2F1
2 changed files with 11 additions and 13 deletions

View File

@ -182,26 +182,23 @@ def send2trash(paths):
path_b = fsencode(path) path_b = fsencode(path)
elif isinstance(path, bytes): elif isinstance(path, bytes):
path_b = path path_b = path
elif hasattr(path, "__fspath__"):
# Python 3.6 PathLike protocol
return send2trash(path.__fspath__())
else: else:
raise TypeError("str, bytes or PathLike expected, not %r" % type(path)) raise TypeError("str, bytes or PathLike expected, not %r" % type(path))
if not op.exists(path_b): if not op.exists(path_b):
raise OSError("File not found: %s" % path) raise OSError(errno.ENOENT, "File not found: %s" % path)
# ...should check whether the user has the necessary permissions to delete # ...should check whether the user has the necessary permissions to delete
# it, before starting the trashing operation itself. [2] # it, before starting the trashing operation itself. [2]
if not os.access(path_b, os.W_OK): if not os.access(path_b, os.W_OK):
raise OSError("Permission denied: %s" % path) raise OSError(errno.EACCES, "Permission denied: %s" % path)
# if the file to be trashed is on the same device as HOMETRASH we
# want to move it there.
path_dev = get_dev(path_b)
path_dev = get_dev(path_b)
# If XDG_DATA_HOME or HOMETRASH do not yet exist we need to stat the # If XDG_DATA_HOME or HOMETRASH do not yet exist we need to stat the
# home directory, and these paths will be created further on if needed. # home directory, and these paths will be created further on if needed.
trash_dev = get_dev(op.expanduser(b"~")) trash_dev = get_dev(op.expanduser(b"~"))
# if the file to be trashed is on the same device as HOMETRASH we
# want to move it there.
if path_dev == trash_dev: if path_dev == trash_dev:
topdir = XDG_DATA_HOME topdir = XDG_DATA_HOME
dest_trash = HOMETRASH_B dest_trash = HOMETRASH_B

View File

@ -22,6 +22,7 @@ if sys.platform != "win32":
import send2trash.plat_other import send2trash.plat_other
from send2trash.plat_other import send2trash as s2t from send2trash.plat_other import send2trash as s2t
INFO_SUFFIX = send2trash.plat_other.INFO_SUFFIX
HOMETRASH = send2trash.plat_other.HOMETRASH HOMETRASH = send2trash.plat_other.HOMETRASH
else: else:
pytest.skip("Skipping non-windows tests", allow_module_level=True) pytest.skip("Skipping non-windows tests", allow_module_level=True)
@ -39,7 +40,7 @@ def testfile():
# Remove trash files if they exist # Remove trash files if they exist
if op.exists(op.join(HOMETRASH, "files", name)): if op.exists(op.join(HOMETRASH, "files", name)):
os.remove(op.join(HOMETRASH, "files", name)) os.remove(op.join(HOMETRASH, "files", name))
os.remove(op.join(HOMETRASH, "info", name + ".trashinfo")) os.remove(op.join(HOMETRASH, "info", name + INFO_SUFFIX))
if op.exists(file.name): if op.exists(file.name):
os.remove(file.name) os.remove(file.name)
@ -59,7 +60,7 @@ def testfiles():
yield files yield files
filenames = [op.basename(file.name) for file in files] filenames = [op.basename(file.name) for file in files]
[os.remove(op.join(HOMETRASH, "files", filename)) for filename in filenames] [os.remove(op.join(HOMETRASH, "files", filename)) for filename in filenames]
[os.remove(op.join(HOMETRASH, "info", filename + ".trashinfo")) for filename in filenames] [os.remove(op.join(HOMETRASH, "info", filename + INFO_SUFFIX)) for filename in filenames]
def test_trash(testfile): def test_trash(testfile):
@ -94,7 +95,7 @@ def gen_unicode_file():
# Cleanup trash files on supported platforms # Cleanup trash files on supported platforms
if sys.platform != "win32" and op.exists(op.join(HOMETRASH, "files", name)): if sys.platform != "win32" and op.exists(op.join(HOMETRASH, "files", name)):
os.remove(op.join(HOMETRASH, "files", name)) os.remove(op.join(HOMETRASH, "files", name))
os.remove(op.join(HOMETRASH, "info", name + ".trashinfo")) os.remove(op.join(HOMETRASH, "info", name + INFO_SUFFIX))
if op.exists(file): if op.exists(file):
os.remove(file) os.remove(file)
@ -162,11 +163,11 @@ def test_trash_topdir(gen_ext_vol):
s2t(gen_ext_vol[2]) s2t(gen_ext_vol[2])
assert op.exists(gen_ext_vol[2]) is False assert op.exists(gen_ext_vol[2]) is False
assert op.exists(op.join(trash_dir, str(os.getuid()), "files", gen_ext_vol[1])) is True assert op.exists(op.join(trash_dir, str(os.getuid()), "files", gen_ext_vol[1])) is True
assert op.exists(op.join(trash_dir, str(os.getuid()), "info", gen_ext_vol[1] + ".trashinfo",)) is True assert op.exists(op.join(trash_dir, str(os.getuid()), "info", gen_ext_vol[1] + INFO_SUFFIX,)) is True
# info relative path (if another test is added, with the same fileName/Path, # info relative path (if another test is added, with the same fileName/Path,
# then it gets renamed etc.) # then it gets renamed etc.)
cfg = ConfigParser() cfg = ConfigParser()
cfg.read(op.join(trash_dir, str(os.getuid()), "info", gen_ext_vol[1] + ".trashinfo")) cfg.read(op.join(trash_dir, str(os.getuid()), "info", gen_ext_vol[1] + INFO_SUFFIX))
assert (gen_ext_vol[1] == cfg.get("Trash Info", "Path", raw=True)) is True assert (gen_ext_vol[1] == cfg.get("Trash Info", "Path", raw=True)) is True