comics2pdf-improved/comic2pdf.py at master · Jackmaster98/comics2pdf-improved · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
# Converts .cbr and .cbz files to .pdf
#
# Use:  python comic2pdf.py
# -- Only works with comicbook files that contain JPG's (for now).
# -- The script should be in the same directory the file(s) to convert are in.
#
# Author:  MComas1
# Date:  14-09-18
#
# License:  You can do what you want with it.
# Mainly based on a script by Bransorem (https://github.com/bransorem/comic2pdf)

import os, sys, zipfile, patoolib
from PIL import Image
import PIL.ExifTags
failed = False

def nlog_info (msg, out=open("comic2pdf_log.txt","a")):
    """Print info message to stdout (or any other given output)."""
    print("patool:", msg, file=out)

def olog_info (msg, out=sys.stdout):
    """Print info message to stdout (or any other given output)."""
    print("patool:", msg, file=out)

def handlerar2(filein):
	tmp_dir = os.getcwd()+"\\Teemp\\"
	if not os.path.exists(tmp_dir):
		os.mkdir(tmp_dir)
	original = sys.stdout
	sys.stdout = open("comic2pdf_log.txt","a")
	patoolib.util.log_info = nlog_info
	patoolib.extract_archive(filein, outdir=tmp_dir)
	newfile = filein.replace(filein[-4:],".pdf")
	toPDF2(newfile,tmp_dir,7)
	if os.path.exists(tmp_dir):
		cleanDir(tmp_dir)
	print("------------------------------------------------------------")

	sys.stdout = original
	print("\""+newfile[:-4]+"\" successfully converted!")

def handlezip(filein):
	zip_ref = zipfile.ZipFile(filein, 'r')
	tmp_dir = os.getcwd()+"\\Teemp\\"
	zip_ref.extractall(tmp_dir)
	zip_ref.close()
	newfile = filein.replace(filein[-4:],".pdf")
	toPDF2(newfile,tmp_dir,0)
	try:
		if os.path.exists(tmp_dir):
			os.cleanDir(tmp_dir)
	except:
		aaa = 223
	print("\""+newfile[:-4]+"\" successfully converted!")

def toPDF2(filename, newdir,ii):
	ffiles = os.listdir(newdir)
	if ffiles[0].endswith(".xml"):
		ffiles.remove(ffiles[0])
	if (len(ffiles) == 1):
		toPDF2(filename,newdir+ffiles[0]+"\\",ii)
	else:
		# imagelist is the list with all image filenames
		im_list = list()
		firstP = True
		im = None
		for image in ffiles:
			if (image.endswith(".jpg") or image.endswith(".png") or image.endswith(".webp") or image.endswith(".tif.png.jpg") or image.endswith(".png.jpg") or image.endswith(".JPG") or image.endswith(".jpeg") or image.endswith(".JPEG")):
				image.replace("~", "")
				if (image.endswith(".webp") or image.endswith(".png")):
					os.rename(newdir+image,newdir+image + ".jpg")
					image = image + ".jpg"
				im1 = Image.open(newdir+image)
				try:
					im1.save(newdir+image,dpi=(96,96))
				except:
					aaaaa = 4

				if (firstP):
					im = im1
					firstP = False
				else: im_list.append(im1)
			else: continue
		#print(exif)
		im.save(filename, "PDF" ,resolution=100.0, save_all=True, append_images=im_list)
	cleanDir(newdir)
	#print("OK")

def cleanDir(dir):
	try:
		files = os.listdir(dir)
		for file in files:
			os.remove(dir+"\\"+file)
		os.rmdir(dir)
	except: print("No dir to clean!")

def opendir(directory):
	# look at all files in directory
	#print(os.listdir(directory))
	for file in os.listdir(directory):
		# file extension cbr only
		if (file[-4:] == '.cbz' or file[-4:] == '.zip'):
			# change to zip
			handlezip(file)
		elif (file[-4:] == '.cbr' or file[-4:] == '.rar'):
			# change to rar
			try:
				handlerar2(file)
			except:
				handlerar2(file)
	if failed:
		print ("WARNING: some items were skipped")


#os.chdir(sys.argv[1])
opendir(os.getcwd())