1
0
Форк 0

Fix silently failing on Windows (#33)

* Fix #31: Silently failing on Windows

* Update Windows test

* Fix test folders not getting removed
This commit is contained in:
sharkykh 2019-04-30 19:28:09 +03:00 зафіксовано Virgil Dupras
джерело 8f684a9c8b
коміт 66afce7252
2 змінених файлів з 39 додано та 10 видалено

@ -7,7 +7,8 @@
from __future__ import unicode_literals
from ctypes import (windll, Structure, byref, c_uint,
create_unicode_buffer, addressof)
create_unicode_buffer, addressof,
GetLastError, FormatError)
from ctypes.wintypes import HWND, UINT, LPCWSTR, BOOL
import os.path as op
@ -49,6 +50,11 @@ def get_short_path_name(long_name):
if not long_name.startswith('\\\\?\\'):
long_name = '\\\\?\\' + long_name
buf_size = GetShortPathNameW(long_name, None, 0)
# FIX: https://github.com/hsoft/send2trash/issues/31
# If buffer size is zero, an error has occurred.
if not buf_size:
err_no = GetLastError()
raise WindowsError(err_no, FormatError(err_no), long_name[4:])
output = create_unicode_buffer(buf_size)
GetShortPathNameW(long_name, output, buf_size)
return output.value[4:] # Remove '\\?\' for SHFileOperationW
@ -83,4 +89,4 @@ def send2trash(path):
fileop.lpszProgressTitle = None
result = SHFileOperationW(byref(fileop))
if result:
raise WindowsError(None, None, path, result)
raise WindowsError(result, FormatError(result), path)

@ -1,5 +1,6 @@
# coding: utf-8
import os
import shutil
import sys
import unittest
from os import path as op
@ -8,12 +9,37 @@ from tempfile import gettempdir
from send2trash import send2trash as s2t
@unittest.skipIf(sys.platform != 'win32', 'Windows only')
class TestNormal(unittest.TestCase):
def setUp(self):
self.dirname = '\\\\?\\' + op.join(gettempdir(), 'python.send2trash')
self.file = op.join(self.dirname, 'testfile.txt')
self._create_tree(self.file)
def tearDown(self):
shutil.rmtree(self.dirname, ignore_errors=True)
def _create_tree(self, path):
dirname = op.dirname(path)
if not op.isdir(dirname):
os.makedirs(dirname)
with open(path, 'w') as writer:
writer.write('send2trash test')
def test_trash_file(self):
s2t(self.file)
self.assertFalse(op.exists(self.file))
def test_file_not_found(self):
file = op.join(self.dirname, 'otherfile.txt')
self.assertRaises(WindowsError, s2t, file)
@unittest.skipIf(sys.platform != 'win32', 'Windows only')
class TestLongPath(unittest.TestCase):
def setUp(self):
filename = 'A' * 100
self.dirname = '\\\\?\\' + os.path.join(gettempdir(), filename)
self.file = os.path.join(
self.dirname = '\\\\?\\' + op.join(gettempdir(), filename)
self.file = op.join(
self.dirname,
filename,
filename, # From there, the path is not trashable from Explorer
@ -22,14 +48,11 @@ class TestLongPath(unittest.TestCase):
self._create_tree(self.file)
def tearDown(self):
try:
os.remove(self.dirname)
except OSError:
pass
shutil.rmtree(self.dirname, ignore_errors=True)
def _create_tree(self, path):
dirname = os.path.dirname(path)
if not os.path.isdir(dirname):
dirname = op.dirname(path)
if not op.isdir(dirname):
os.makedirs(dirname)
with open(path, 'w') as writer:
writer.write('Looong filename!')