Added zero-padding to .img files
This commit is contained in:
christopher-owen
GitHub
parent
ee4df84a9c
commit
ae08fb893a
284
mur.py
284
mur.py
@ -19,7 +19,7 @@ from api.exceptions import IneligibleError
|
||||
client = api.Client()
|
||||
|
||||
def print_title():
|
||||
print("""
|
||||
print("""
|
||||
_____ _____ _____
|
||||
| | | | __ |
|
||||
| | | | | | -|
|
||||
@ -27,166 +27,178 @@ def print_title():
|
||||
""")
|
||||
|
||||
def get_os():
|
||||
if platform.system() == 'Windows':
|
||||
return True
|
||||
return False
|
||||
if platform.system() == 'Windows':
|
||||
return True
|
||||
return False
|
||||
|
||||
def set_con_title():
|
||||
if get_os():
|
||||
os.system('title MUR R1 (by Sorrow446)')
|
||||
else:
|
||||
sys.stdout.write('\x1b]2;MUR R1 (by Sorrow446)\x07')
|
||||
if get_os():
|
||||
os.system('title MUR R1 (by Sorrow446)')
|
||||
else:
|
||||
sys.stdout.write('\x1b]2;MUR R1 (by Sorrow446)\x07')
|
||||
|
||||
def sanitize(fn):
|
||||
if get_os():
|
||||
return re.sub(r'[\/:*?"><|]', '_', fn)
|
||||
else:
|
||||
return re.sub('/', '_', fn)
|
||||
if get_os():
|
||||
return re.sub(r'[\/:*?"><|]', '_', fn)
|
||||
else:
|
||||
return re.sub('/', '_', fn)
|
||||
|
||||
def parse_args():
|
||||
parser = argparse.ArgumentParser(
|
||||
description='Sorrow446.'
|
||||
)
|
||||
parser.add_argument(
|
||||
'-u', '--url',
|
||||
help="URL - marvel.com/comics/issue/ or read.marvel.com/#/book/.",
|
||||
nargs='*',
|
||||
required=True
|
||||
)
|
||||
parser.add_argument(
|
||||
'-f', '--format',
|
||||
help="Export format.",
|
||||
choices=['cbz', 'pdf'],
|
||||
required=True
|
||||
)
|
||||
parser.add_argument(
|
||||
'-m', '--meta',
|
||||
help="Write comic's metadata to JSON file.",
|
||||
action='store_true'
|
||||
)
|
||||
return parser.parse_args()
|
||||
parser = argparse.ArgumentParser(
|
||||
description='Sorrow446.'
|
||||
)
|
||||
parser.add_argument(
|
||||
'-u', '--url',
|
||||
help="URL - marvel.com/comics/issue/ or read.marvel.com/#/book/.",
|
||||
nargs='*',
|
||||
required=True
|
||||
)
|
||||
parser.add_argument(
|
||||
'-f', '--format',
|
||||
help="Export format.",
|
||||
choices=['cbz', 'pdf'],
|
||||
required=True
|
||||
)
|
||||
parser.add_argument(
|
||||
'-m', '--meta',
|
||||
help="Write comic's metadata to JSON file.",
|
||||
action='store_true'
|
||||
)
|
||||
return parser.parse_args()
|
||||
|
||||
def parse_cookies(cd, out_cookies={}):
|
||||
with open(os.path.join(cd, 'cookies.txt')) as f:
|
||||
for line in f.readlines():
|
||||
if not line.startswith('#'):
|
||||
field = line.strip().split('\t')
|
||||
out_cookies[field[5]] = field[6]
|
||||
client.set_cookies(out_cookies)
|
||||
with open(os.path.join(cd, 'cookies.txt')) as f:
|
||||
for line in f.readlines():
|
||||
if not line.startswith('#'):
|
||||
field = line.strip().split('\t')
|
||||
out_cookies[field[5]] = field[6]
|
||||
client.set_cookies(out_cookies)
|
||||
|
||||
def exist_check(f):
|
||||
if os.path.isfile(f):
|
||||
return True
|
||||
return False
|
||||
if os.path.isfile(f):
|
||||
return True
|
||||
return False
|
||||
|
||||
def dir_setup(tmp_dir, dl_dir):
|
||||
if os.path.isdir(tmp_dir):
|
||||
shutil.rmtree(tmp_dir)
|
||||
if not os.path.isdir(dl_dir):
|
||||
os.makedirs(dl_dir)
|
||||
os.makedirs(tmp_dir)
|
||||
if os.path.isdir(tmp_dir):
|
||||
shutil.rmtree(tmp_dir)
|
||||
if not os.path.isdir(dl_dir):
|
||||
os.makedirs(dl_dir)
|
||||
os.makedirs(tmp_dir)
|
||||
|
||||
def check_url(url):
|
||||
regexes=[
|
||||
r'http[s]://(read).marvel.com/#/book/([0-9]+$)',
|
||||
r'http[s]://(www).marvel.com/comics/issue/([0-9]+)/.+'
|
||||
]
|
||||
for regex in regexes:
|
||||
match = re.match(regex, url)
|
||||
if match:
|
||||
return match.group(1), match.group(2)
|
||||
regexes=[
|
||||
r'http[s]://(read).marvel.com/#/book/([0-9]+$)',
|
||||
r'http[s]://(www).marvel.com/comics/issue/([0-9]+)/.+'
|
||||
]
|
||||
for regex in regexes:
|
||||
match = re.match(regex, url)
|
||||
if match:
|
||||
return match.group(1), match.group(2)
|
||||
|
||||
def download(urls, tmp_dir, cur=0):
|
||||
total = len(urls)
|
||||
for url in urls:
|
||||
cur += 1
|
||||
print('Downloading image {} of {}...'.format(cur, total))
|
||||
r = client.session.get(url, stream=True)
|
||||
r.raise_for_status()
|
||||
size = int(r.headers.get('content-length', 0))
|
||||
abs = os.path.join(tmp_dir, str(cur) + '.jpg')
|
||||
with open(abs, 'wb') as f:
|
||||
with tqdm(total=size, unit='B',
|
||||
unit_scale=True, unit_divisor=1024,
|
||||
initial=0, miniters=1) as bar:
|
||||
for chunk in r.iter_content(32*1024):
|
||||
if chunk:
|
||||
f.write(chunk)
|
||||
bar.update(len(chunk))
|
||||
total = len(urls)
|
||||
for url in urls:
|
||||
cur += 1
|
||||
print('Downloading image {} of {}...'.format(cur, total))
|
||||
r = client.session.get(url, stream=True)
|
||||
r.raise_for_status()
|
||||
size = int(r.headers.get('content-length', 0))
|
||||
abs = os.path.join(tmp_dir, str(cur) + '.jpg')
|
||||
with open(abs, 'wb') as f:
|
||||
with tqdm(total=size, unit='B',
|
||||
unit_scale=True, unit_divisor=1024,
|
||||
initial=0, miniters=1) as bar:
|
||||
for chunk in r.iter_content(32*1024):
|
||||
if chunk:
|
||||
f.write(chunk)
|
||||
bar.update(len(chunk))
|
||||
|
||||
def make_pdf(abs, images, title):
|
||||
with open(abs, 'wb') as f:
|
||||
f.write(img2pdf.convert(images, title=title))
|
||||
with open(abs, 'wb') as f:
|
||||
f.write(img2pdf.convert(images, title=title))
|
||||
|
||||
def make_cbz(abs, images):
|
||||
with zipfile.ZipFile(abs, 'w', zipfile.ZIP_STORED) as f:
|
||||
for i in images:
|
||||
f.write(i)
|
||||
with zipfile.ZipFile(abs, 'w', zipfile.ZIP_STORED) as f:
|
||||
for i in images:
|
||||
f.write(i)
|
||||
|
||||
def write_meta(meta_abs, meta):
|
||||
with open(meta_abs, 'w') as f:
|
||||
json.dump(meta, f, indent=4)
|
||||
with open(meta_abs, 'w') as f:
|
||||
json.dump(meta, f, indent=4)
|
||||
|
||||
def err(e, cur, tot):
|
||||
print(e)
|
||||
if cur == tot:
|
||||
sys.exit(1)
|
||||
print(e)
|
||||
if cur == tot:
|
||||
sys.exit(1)
|
||||
|
||||
def main():
|
||||
if hasattr(sys, 'frozen'):
|
||||
cd = os.path.dirname(sys.executable)
|
||||
else:
|
||||
cd = os.path.dirname(__file__)
|
||||
tmp_dir = os.path.join(cd, 'mur_tmp')
|
||||
dl_dir = os.path.join(cd, 'MUR downloads')
|
||||
dir_setup(tmp_dir, dl_dir)
|
||||
parse_cookies(cd)
|
||||
args = parse_args()
|
||||
tot = len(args.url)
|
||||
cur = 0
|
||||
for url in args.url:
|
||||
cur += 1
|
||||
try:
|
||||
print("Comic {} of {}:".format(cur, tot))
|
||||
try:
|
||||
type, id = check_url(url)
|
||||
except TypeError:
|
||||
err('Invalid URL: '+str(url), cur, tot)
|
||||
continue
|
||||
if type == "www":
|
||||
id = client.get_id(url)
|
||||
fmt = args.format
|
||||
meta = client.get_comic_meta(id)
|
||||
title = meta['title']
|
||||
title_s = sanitize(title)
|
||||
print(str(title) + "\n")
|
||||
abs = os.path.join(dl_dir, '{}.{}'.format(title_s, fmt))
|
||||
if exist_check(abs):
|
||||
err('Comic already exists locally.', cur, tot)
|
||||
continue
|
||||
try:
|
||||
download(client.get_comic(id), tmp_dir)
|
||||
except IneligibleError as e:
|
||||
print(e)
|
||||
sys.exit(1)
|
||||
images = [os.path.join(tmp_dir, i) for i in os.listdir(tmp_dir)]
|
||||
print('Converting to {}...'.format(fmt.upper()))
|
||||
if fmt == 'pdf':
|
||||
make_pdf(abs, images, title)
|
||||
else:
|
||||
make_cbz(abs, images)
|
||||
if args.meta:
|
||||
print("Writing metadata to JSON file...")
|
||||
meta_abs = os.path.join(dl_dir, '{}_meta.json'.format(title_s))
|
||||
write_meta(meta_abs, meta)
|
||||
for i in images:
|
||||
os.remove(i)
|
||||
except HTTPError as e:
|
||||
err(e, cur, tot)
|
||||
except Exception as e:
|
||||
err(e, cur, tot)
|
||||
if hasattr(sys, 'frozen'):
|
||||
cd = os.path.dirname(sys.executable)
|
||||
else:
|
||||
cd = os.path.dirname(__file__)
|
||||
tmp_dir = os.path.join(cd, 'mur_tmp')
|
||||
dl_dir = os.path.join(cd, 'MUR downloads')
|
||||
dir_setup(tmp_dir, dl_dir)
|
||||
parse_cookies(cd)
|
||||
args = parse_args()
|
||||
tot = len(args.url)
|
||||
cur = 0
|
||||
for url in args.url:
|
||||
cur += 1
|
||||
try:
|
||||
print("Comic {} of {}:".format(cur, tot))
|
||||
try:
|
||||
type, id = check_url(url)
|
||||
except TypeError:
|
||||
err('Invalid URL: '+str(url), cur, tot)
|
||||
continue
|
||||
if type == "www":
|
||||
id = client.get_id(url)
|
||||
fmt = args.format
|
||||
meta = client.get_comic_meta(id)
|
||||
title = meta['title']
|
||||
title_s = sanitize(title)
|
||||
print(str(title) + "\n")
|
||||
abs = os.path.join(dl_dir, '{}.{}'.format(title_s, fmt))
|
||||
if exist_check(abs):
|
||||
err('Comic already exists locally.', cur, tot)
|
||||
continue
|
||||
try:
|
||||
download(client.get_comic(id), tmp_dir)
|
||||
except IneligibleError as e:
|
||||
print(e)
|
||||
sys.exit(1)
|
||||
#input("Press Enter to continue...")
|
||||
##NEW
|
||||
sourcedir = "tmp_dir"; number_ofdigits = 5; extensions = (".jpg", ".jpeg")
|
||||
|
||||
files = os.listdir(tmp_dir)
|
||||
for item in files:
|
||||
if item.endswith(extensions):
|
||||
name = item.split("."); zeros = number_ofdigits-len(name[0])
|
||||
newname = str(zeros*"0")+name[0]+"."+name[1]
|
||||
shutil.move(tmp_dir+"/"+item, tmp_dir+"/"+newname)
|
||||
##END NEW
|
||||
#input("Press Enter to continue...")
|
||||
images = [os.path.join(tmp_dir, i) for i in os.listdir(tmp_dir)]
|
||||
print('Converting to {}...'.format(fmt.upper()))
|
||||
if fmt == 'pdf':
|
||||
make_pdf(abs, images, title)
|
||||
else:
|
||||
make_cbz(abs, images)
|
||||
if args.meta:
|
||||
print("Writing metadata to JSON file...")
|
||||
meta_abs = os.path.join(dl_dir, '{}_meta.json'.format(title_s))
|
||||
write_meta(meta_abs, meta)
|
||||
for i in images:
|
||||
os.remove(i)
|
||||
except HTTPError as e:
|
||||
err(e, cur, tot)
|
||||
except Exception as e:
|
||||
err(e, cur, tot)
|
||||
|
||||
if __name__ == '__main__':
|
||||
print_title()
|
||||
set_con_title()
|
||||
main()
|
||||
print_title()
|
||||
set_con_title()
|
||||
main()
|
||||
|
||||
Loading…
Reference in New Issue
Block a user