From a7440e06a94ed71d16b9a774855832f3f9996db7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Pawe=C5=82=20Jastrz=C4=99bski?= Date: Tue, 1 Sep 2015 18:53:09 +0200 Subject: [PATCH] Additional temp cleanup --- kcc/comic2ebook.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/kcc/comic2ebook.py b/kcc/comic2ebook.py index ce4cfa6..f074c43 100755 --- a/kcc/comic2ebook.py +++ b/kcc/comic2ebook.py @@ -28,7 +28,7 @@ from urllib.request import Request, urlopen from re import sub from stat import S_IWRITE, S_IREAD, S_IEXEC from zipfile import ZipFile, ZIP_STORED, ZIP_DEFLATED -from tempfile import mkdtemp +from tempfile import mkdtemp, gettempdir from shutil import move, copytree, rmtree from optparse import OptionParser, OptionGroup from multiprocessing import Pool @@ -841,6 +841,13 @@ def sanitizePermissions(filetree): os.chmod(os.path.join(root, name), S_IWRITE | S_IREAD | S_IEXEC) +def sanitizeTemp(): + for root, dirs, _ in walkLevel(gettempdir(), 0): + for tempdir in dirs: + if tempdir.startswith('KCC-'): + rmtree(os.path.join(root, tempdir), True) + + # noinspection PyUnboundLocalVariable def splitDirectory(path): # Detect directory stucture @@ -1216,6 +1223,7 @@ def makeBook(source, qtGUI=None): GUI.progressBarTick.emit('1') else: checkTools(source) + sanitizeTemp() path = getWorkFolder(source) print("\nChecking images...") getComicInfo(os.path.join(path, "OEBPS", "Images"), source)