1
0
mirror of https://github.com/arsenetar/send2trash.git synced 2025-05-10 02:39:50 +00:00

Compare commits

..

No commits in common. "7686647389d394aa8affa0c04a2741225d3b65f5" and "2aa834be94448fb3b63618aad9c7c4e1ffba4bdf" have entirely different histories.

3 changed files with 107 additions and 64 deletions

View File

@ -37,10 +37,54 @@ class FileOperationProgressSink(DesignatedWrapPolicy):
# but that may need some additional considerations before implementing. # but that may need some additional considerations before implementing.
return 0 if flags & shellcon.TSF_DELETE_RECYCLE_IF_POSSIBLE else 0x80004005 # S_OK, or E_FAIL return 0 if flags & shellcon.TSF_DELETE_RECYCLE_IF_POSSIBLE else 0x80004005 # S_OK, or E_FAIL
def PostDeleteItem(self, flags, item, hr_delete, newly_created): def PostDeleteItem(self, flags, item, hrDelete, newlyCreated):
if newly_created: if newlyCreated:
self.newItem = newly_created.GetDisplayName(shellcon.SHGDN_FORPARSING) self.newItem = newlyCreated.GetDisplayName(shellcon.SHGDN_FORPARSING)
def StartOperations(self):
pass
def FinishOperations(self, Result):
pass
def PreRenameItem(self, Flags, Item, NewName):
pass
def PostRenameItem(self, Flags, Item, NewName, hrRename, NewlyCreated):
pass
def PreMoveItem(self, Flags, Item, DestinationFolder, NewName):
pass
def PostMoveItem(self, Flags, Item, DestinationFolder, NewName, hrMove, NewlyCreated):
pass
def PreCopyItem(self, Flags, Item, DestinationFolder, NewName):
pass
def PostCopyItem(self, Flags, Item, DestinationFolder, NewName, hrCopy, NewlyCreated):
pass
def PreNewItem(self, Flags, DestinationFolder, NewName):
pass
def PostNewItem(
self, Flags, DestinationFolder, NewName, TemplateName, FileAttributes, hrNew, NewItem,
):
pass
def UpdateProgress(self, WorkTotal, WorkSoFar):
pass
def ResetTimer(self):
pass
def PauseTimer(self):
pass
def ResumeTimer(self):
pass
def create_sink(): def CreateSink():
return pythoncom.WrapObject(FileOperationProgressSink(), shell.IID_IFileOperationProgressSink) return pythoncom.WrapObject(FileOperationProgressSink(), shell.IID_IFileOperationProgressSink)

View File

@ -12,7 +12,7 @@ from platform import version
import pythoncom import pythoncom
import pywintypes import pywintypes
from win32com.shell import shell, shellcon from win32com.shell import shell, shellcon
from .IFileOperationProgressSink import create_sink from .IFileOperationProgressSink import CreateSink
def send2trash(paths): def send2trash(paths):
@ -42,7 +42,7 @@ def send2trash(paths):
# actually try to perform the operation, this section may throw a # actually try to perform the operation, this section may throw a
# pywintypes.com_error which does not seem to create as nice of an # pywintypes.com_error which does not seem to create as nice of an
# error as OSError so wrapping with try to convert # error as OSError so wrapping with try to convert
sink = create_sink() sink = CreateSink()
try: try:
for path in paths: for path in paths:
item = shell.SHCreateItemFromParsingName(path, None, shell.IID_IShellItem) item = shell.SHCreateItemFromParsingName(path, None, shell.IID_IShellItem)

View File

@ -13,10 +13,9 @@ except ImportError:
# py2 # py2
from ConfigParser import ConfigParser # noqa: F401 from ConfigParser import ConfigParser # noqa: F401
from tempfile import mkdtemp, NamedTemporaryFile from tempfile import mkdtemp, NamedTemporaryFile, mktemp
import shutil import shutil
import stat import stat
import uuid
if sys.platform != "win32": if sys.platform != "win32":
import send2trash.plat_other import send2trash.plat_other
@ -85,50 +84,52 @@ def _filesys_enc():
@pytest.fixture @pytest.fixture
def gen_unicode_file(): def testUnicodefile():
name = u"send2trash_tést1" name = u"send2trash_tést1"
file = op.join(op.expanduser(b"~"), name.encode("utf-8")) file = op.join(op.expanduser(b"~"), name.encode("utf-8"))
touch(file) touch(file)
assert op.exists(file) is True assert op.exists(file) is True
yield file yield 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":
os.remove(op.join(HOMETRASH, "files", name)) # Remove trash files if they exist
os.remove(op.join(HOMETRASH, "info", name + ".trashinfo")) if op.exists(op.join(HOMETRASH, "files", name)):
os.remove(op.join(HOMETRASH, "files", name))
os.remove(op.join(HOMETRASH, "info", name + ".trashinfo"))
if op.exists(file): if op.exists(file):
os.remove(file) os.remove(file)
@pytest.mark.skipif(_filesys_enc() == "ascii", reason="Requires Unicode filesystem") @pytest.mark.skipif(_filesys_enc() == "ascii", reason="Requires Unicode filesystem")
def test_trash_bytes(gen_unicode_file): def test_trash_bytes(testUnicodefile):
s2t(gen_unicode_file) s2t(testUnicodefile)
assert not op.exists(gen_unicode_file) assert not op.exists(testUnicodefile)
@pytest.mark.skipif(_filesys_enc() == "ascii", reason="Requires Unicode filesystem") @pytest.mark.skipif(_filesys_enc() == "ascii", reason="Requires Unicode filesystem")
def test_trash_unicode(gen_unicode_file): def test_trash_unicode(testUnicodefile):
s2t(gen_unicode_file.decode(sys.getfilesystemencoding())) s2t(testUnicodefile.decode(sys.getfilesystemencoding()))
assert not op.exists(gen_unicode_file) assert not op.exists(testUnicodefile)
class ExtVol: class ExtVol:
def __init__(self, path): def __init__(self, path):
self.trash_topdir = path self.trashTopdir = path
if PY3: if PY3:
self.trash_topdir_b = os.fsencode(self.trash_topdir) self.trashTopdir_b = os.fsencode(self.trashTopdir)
else: else:
self.trash_topdir_b = self.trash_topdir self.trashTopdir_b = self.trashTopdir
def s_getdev(path): def s_getdev(path):
from send2trash.plat_other import is_parent from send2trash.plat_other import is_parent
st = os.lstat(path) st = os.lstat(path)
if is_parent(self.trash_topdir, path): if is_parent(self.trashTopdir, path):
return "dev" return "dev"
return st.st_dev return st.st_dev
def s_ismount(path): def s_ismount(path):
if op.realpath(path) in (op.realpath(self.trash_topdir), op.realpath(self.trash_topdir_b),): if op.realpath(path) in (op.realpath(self.trashTopdir), op.realpath(self.trashTopdir_b),):
return True return True
return old_ismount(path) return old_ismount(path)
@ -140,60 +141,58 @@ class ExtVol:
def cleanup(self): def cleanup(self):
send2trash.plat_other.get_dev = self.old_getdev send2trash.plat_other.get_dev = self.old_getdev
send2trash.plat_other.os.path.ismount = self.old_ismount send2trash.plat_other.os.path.ismount = self.old_ismount
shutil.rmtree(self.trash_topdir) shutil.rmtree(self.trashTopdir)
@pytest.fixture @pytest.fixture
def gen_ext_vol(): def testExtVol():
trash_topdir = mkdtemp(prefix="s2t") trashTopdir = mkdtemp(prefix="s2t")
volume = ExtVol(trash_topdir) volume = ExtVol(trashTopdir)
file_name = "test.txt" fileName = "test.txt"
file_path = op.join(volume.trash_topdir, file_name) filePath = op.join(volume.trashTopdir, fileName)
touch(file_path) touch(filePath)
assert op.exists(file_path) is True assert op.exists(filePath) is True
yield volume, file_name, file_path yield volume, fileName, filePath
volume.cleanup() volume.cleanup()
def test_trash_topdir(gen_ext_vol): def test_trash_topdir(testExtVol):
trash_dir = op.join(gen_ext_vol[0].trash_topdir, ".Trash") trashDir = op.join(testExtVol[0].trashTopdir, ".Trash")
os.mkdir(trash_dir, 0o777 | stat.S_ISVTX) os.mkdir(trashDir, 0o777 | stat.S_ISVTX)
s2t(gen_ext_vol[2]) s2t(testExtVol[2])
assert op.exists(gen_ext_vol[2]) is False assert op.exists(testExtVol[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(trashDir, str(os.getuid()), "files", testExtVol[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(trashDir, str(os.getuid()), "info", testExtVol[1] + ".trashinfo",)) 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(trashDir, str(os.getuid()), "info", testExtVol[1] + ".trashinfo"))
assert (gen_ext_vol[1] == cfg.get("Trash Info", "Path", raw=True)) is True assert (testExtVol[1] == cfg.get("Trash Info", "Path", raw=True)) is True
def test_trash_topdir_fallback(gen_ext_vol): def test_trash_topdir_fallback(testExtVol):
s2t(gen_ext_vol[2]) s2t(testExtVol[2])
assert op.exists(gen_ext_vol[2]) is False assert op.exists(testExtVol[2]) is False
assert ( assert op.exists(op.join(testExtVol[0].trashTopdir, ".Trash-" + str(os.getuid()), "files", testExtVol[1],)) is True
op.exists(op.join(gen_ext_vol[0].trash_topdir, ".Trash-" + str(os.getuid()), "files", gen_ext_vol[1],)) is True
)
def test_trash_topdir_failure(gen_ext_vol): def test_trash_topdir_failure(testExtVol):
os.chmod(gen_ext_vol[0].trash_topdir, 0o500) # not writable to induce the exception os.chmod(testExtVol[0].trashTopdir, 0o500) # not writable to induce the exception
pytest.raises(TrashPermissionError, s2t, [gen_ext_vol[2]]) pytest.raises(TrashPermissionError, s2t, [testExtVol[2]])
os.chmod(gen_ext_vol[0].trash_topdir, 0o700) # writable to allow deletion os.chmod(testExtVol[0].trashTopdir, 0o700) # writable to allow deletion
def test_trash_symlink(gen_ext_vol): def test_trash_symlink(testExtVol):
# Generating a random uuid named path for symlink # Use mktemp (race conditioney but no symlink equivalent)
sl_dir = op.join(op.expanduser("~"), "s2t_" + str(uuid.uuid4())) # Since is_parent uses realpath(), and our getdev uses is_parent,
os.mkdir(op.join(gen_ext_vol[0].trash_topdir, "subdir"), 0o700) # this should work
file_path = op.join(gen_ext_vol[0].trash_topdir, "subdir", gen_ext_vol[1]) slDir = mktemp(prefix="s2t", dir=op.expanduser("~"))
touch(file_path) os.mkdir(op.join(testExtVol[0].trashTopdir, "subdir"), 0o700)
os.symlink(op.join(gen_ext_vol[0].trash_topdir, "subdir"), sl_dir) filePath = op.join(testExtVol[0].trashTopdir, "subdir", testExtVol[1])
s2t(op.join(sl_dir, gen_ext_vol[1])) touch(filePath)
assert op.exists(file_path) is False os.symlink(op.join(testExtVol[0].trashTopdir, "subdir"), slDir)
assert ( s2t(op.join(slDir, testExtVol[1]))
op.exists(op.join(gen_ext_vol[0].trash_topdir, ".Trash-" + str(os.getuid()), "files", gen_ext_vol[1],)) is True assert op.exists(filePath) is False
) assert op.exists(op.join(testExtVol[0].trashTopdir, ".Trash-" + str(os.getuid()), "files", testExtVol[1],)) is True
os.remove(sl_dir) os.remove(slDir)