31 lines
886 B
Bash
31 lines
886 B
Bash
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import os
|
||
|
|
import zipfile
|
||
|
|
from PIL import Image
|
||
|
|
import sys
|
||
|
|
from io import BytesIO
|
||
|
|
|
||
|
|
def extract_cover(epub, output_path, size):
|
||
|
|
for fileinfo in epub.filelist:
|
||
|
|
if fileinfo.filename.lower().endswith(('.jpg', '.jpeg', '.png')):
|
||
|
|
cover = epub.open(fileinfo.filename)
|
||
|
|
im = Image.open(BytesIO(cover.read()))
|
||
|
|
im.thumbnail((size, size), Image.LANCZOS)
|
||
|
|
im.save(os.path.join(output_path, os.path.basename(input_file) + '.png'), "PNG")
|
||
|
|
return True
|
||
|
|
return False
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
input_file = sys.argv[1]
|
||
|
|
output_path = os.environ.get('THUMBNAIL_DIR', os.path.dirname(input_file))
|
||
|
|
size = int(sys.argv[2])
|
||
|
|
|
||
|
|
epub = zipfile.ZipFile(input_file, "r")
|
||
|
|
if extract_cover(epub, output_path, size):
|
||
|
|
exit(0)
|
||
|
|
else:
|
||
|
|
print("Error extracting cover")
|
||
|
|
exit(1)
|
||
|
|
|