Skip to content

Commit

Permalink
Fix silently failing on Windows (#33)
Browse files Browse the repository at this point in the history
* Fix #31: Silently failing on Windows

* Update Windows test

* Fix test folders not getting removed
  • Loading branch information
sharkykh authored and Virgil Dupras committed Apr 30, 2019
1 parent 8f684a9 commit 66afce7
Show file tree
Hide file tree
Showing 2 changed files with 39 additions and 10 deletions.
10 changes: 8 additions & 2 deletions send2trash/plat_win.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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)
39 changes: 31 additions & 8 deletions tests/test_plat_win.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
# coding: utf-8
import os
import shutil
import sys
import unittest
from os import path as op
Expand All @@ -8,12 +9,37 @@
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
Expand All @@ -22,14 +48,11 @@ def setUp(self):
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!')
Expand Down

0 comments on commit 66afce7

Please sign in to comment.