dosage/scripts/create-cbz.py

72 lines
2.1 KiB
Python
Raw Normal View History

2013-05-01 06:33:04 +00:00
#!/usr/bin/env python
2016-04-12 22:52:16 +00:00
# -*- coding: utf-8 -*-
2016-10-28 22:21:41 +00:00
# Copyright (C) 2004-2008 Tristan Seligmann and Jonathan Jacobs
2016-04-12 22:52:16 +00:00
# Copyright (C) 2012-2014 Bastian Kleineidam
# Copyright (C) 2015-2016 Tobias Gruetzmacher
2013-05-01 06:33:04 +00:00
"""
Creates a CBZ file in the comic directory.
Uses an ordered symlink directory (see order-symlinks.py) if it exists,
else the plain files are used.
"""
2016-04-12 22:52:16 +00:00
from __future__ import absolute_import, division, print_function
2013-05-01 06:33:04 +00:00
import sys
import os
import zipfile
2016-04-12 22:52:16 +00:00
sys.path.append(os.path.join(os.path.dirname(__file__), "..")) # noqa
2013-05-01 06:33:04 +00:00
from dosagelib.configuration import App
# known image file extensions
ImageExts = (
".jpg",
".jpeg",
".gif",
".png",
)
2016-04-12 22:52:16 +00:00
2013-05-01 06:33:04 +00:00
def is_image(filename):
"""Determine if given filename is an image."""
# note: isfile() also accepts symlinks
return os.path.isfile(filename) and filename.lower().endswith(ImageExts)
def get_cbz_comment():
"""Return a UTF-8 encoded comment no longer than 65535 bytes.
At the moment this just returns the application name and version,
since cbz readers do not seem to use the comment string anyway."""
return (u"Created by %s" % App).encode('utf-8')
def create_cbz(directory):
"""Creates or updates a CBZ from files in the given comic directory."""
if not os.path.isdir(directory):
print("ERROR: Directory", directory, "not found.")
return
base = os.path.basename(directory.rstrip(os.path.sep))
zipname = '%s.cbz' % base
zipname = os.path.join(directory, zipname)
d = os.path.join(directory, 'inorder')
if os.path.isdir(d):
# use directory with ordered symlinks
directory = d
if os.path.exists(zipname):
os.remove(zipname)
with zipfile.ZipFile(zipname, 'w') as myzip:
for filename in sorted(os.listdir(d)):
fullname = os.path.join(d, filename)
if is_image(fullname):
myzip.write(fullname)
myzip.comment = get_cbz_comment()
print("INFO: Created", zipname)
if __name__ == '__main__':
if len(sys.argv) > 1:
for arg in sys.argv[1:]:
create_cbz(arg)
else:
print("Usage: %s <comic-dir>..." % os.path.basename(sys.argv[0]))