public/kcc
1
0
Fork 0
mirror of https://github.com/ciromattia/kcc synced 2026-01-30 17:07:41 +00:00
Code Issues Releases Wiki Activity

Refactored sanitizeTreeReadOnly

Browse Source
This commit is contained in:
Paweł Jastrzębski
2013-06-18 14:30:46 +02:00
parent fd493dba9a
commit 8a9f11b0f5
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
kcc/comic2ebook.py
View File

@@ -26,12 +26,12 @@ import os
import sys
import tempfile
import re
import stat
from shutil import move
from shutil import copyfile
from shutil import copytree
from shutil import rmtree
from shutil import make_archive
from stat import S_IWRITE
from optparse import OptionParser
from multiprocessing import Pool, Queue, freeze_support
try:
@@ -569,8 +569,7 @@ def getWorkFolder(afile):
os.rmdir(workdir) # needed for copytree() fails if dst already exists
fullPath = os.path.join(workdir, 'OEBPS', 'Images')
copytree(afile, fullPath)
if sys.platform == 'win32':
sanitizeTreeReadOnly(fullPath)
sanitizeTreeBeforeConversion(fullPath)
return workdir
except OSError:
raise
@@ -625,12 +624,14 @@ def sanitizeTree(filetree):
os.rename(os.path.join(root, name), os.path.join(root, slugify(name)))
def sanitizeTreeReadOnly(filetree):
def sanitizeTreeBeforeConversion(filetree):
for root, dirs, files in os.walk(filetree, False):
for name in files:
os.chmod(os.path.join(root, name), S_IWRITE)
os.chmod(os.path.join(root, name), stat.S_IWRITE | stat.S_IREAD)
if os.path.getsize(os.path.join(root, name)) == 0:
os.remove(os.path.join(root, name))
for name in dirs:
os.chmod(os.path.join(root, name), S_IWRITE)
os.chmod(os.path.join(root, name), stat.S_IWRITE | stat.S_IREAD)
def Copyright():