From b8434f7cc08859357e6efca15b60d09ec14e1f46 Mon Sep 17 00:00:00 2001 From: Virgil Dupras Date: Tue, 6 Apr 2010 10:28:43 +0100 Subject: [PATCH] Added the trash_win module. --- modules/trash_win.c | 57 +++++++++++++++++++++++++++++++++++++++++++++ setup.py | 18 ++++++++++---- 2 files changed, 70 insertions(+), 5 deletions(-) create mode 100644 modules/trash_win.c diff --git a/modules/trash_win.c b/modules/trash_win.c new file mode 100644 index 0000000..54d8368 --- /dev/null +++ b/modules/trash_win.c @@ -0,0 +1,57 @@ +#define PY_SSIZE_T_CLEAN +#include "Python.h" + +#define WINDOWS_LEAN_AND_MEAN +#include "windows.h" +#include "shlobj.h" + +/* WARNING: If the filepath is not fully qualify, Windows deletes the file + rather than sending it to trash. + */ + +static PyObject* trash_win_send(PyObject *self, PyObject *args) +{ + SHFILEOPSTRUCTW op; + PyObject *filepath; + Py_ssize_t len, cpysize; + WCHAR filechars[MAX_PATH+1]; + int r; + + if (!PyArg_ParseTuple(args, "O", &filepath)) { + return NULL; + } + + if (!PyUnicode_Check(filepath)) { + PyErr_SetString(PyExc_TypeError, "Unicode filename required"); + return NULL; + } + + len = PyUnicode_GET_SIZE(filepath); + /* +2 because we are going to add two null chars at the end */ + cpysize = sizeof(WCHAR) * (len + 2); + memcpy(filechars, PyUnicode_AsUnicode(filepath), cpysize); + filechars[len] = '\0'; + filechars[len+1] = '\0'; + + op.hwnd = 0; + op.wFunc = FO_DELETE; + op.pFrom = (LPCWSTR)&filechars; + op.pTo = NULL; + op.fFlags = FOF_ALLOWUNDO | FOF_NOCONFIRMATION | FOF_NOERRORUI | FOF_SILENT; + r = SHFileOperationW(&op); + return Py_None; +} + +static PyMethodDef TrashMethods[] = { + {"send", trash_win_send, METH_VARARGS, ""}, + {NULL, NULL, 0, NULL} +}; + +PyMODINIT_FUNC +init_trash_win(void) +{ + PyObject *m = Py_InitModule("_trash_win", TrashMethods); + if (m == NULL) { + return; + } +} \ No newline at end of file diff --git a/setup.py b/setup.py index 77594d9..3a85b40 100644 --- a/setup.py +++ b/setup.py @@ -1,3 +1,4 @@ +import sys import os.path as op from distutils.core import setup @@ -5,11 +6,18 @@ from distutils.extension import Extension exts = [] -exts.append(Extension( - '_trash_osx', - [op.join('modules', 'trash_osx.c')], - extra_link_args=['-framework', 'CoreServices'], -)) +if sys.platform == 'darwin': + exts.append(Extension( + '_trash_osx', + [op.join('modules', 'trash_osx.c')], + extra_link_args=['-framework', 'CoreServices'], + )) +if sys.platform == 'win32': + exts.append(Extension( + '_trash_win', + [op.join('modules', 'trash_win.c')], + extra_link_args = ['shell32.lib'], + )) setup( name='Send2Trash',