85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
import re
|
|
import shutil
|
|
from pathlib import Path
|
|
from zipfile import ZipFile
|
|
|
|
|
|
# create a cbz archive
|
|
def make_archive(chapter_path):
|
|
image_folder = Path(chapter_path)
|
|
zip_path = Path(f"{chapter_path}.zip")
|
|
if not image_folder.exists():
|
|
print(f"ERR: Folder: {image_folder} does not exist")
|
|
return False
|
|
with ZipFile(f"{image_folder}.zip", "w") as zip_archive:
|
|
for file in image_folder.iterdir():
|
|
zip_archive.write(file, file.name)
|
|
zip_path.rename(zip_path.with_suffix(".cbz"))
|
|
shutil.rmtree(image_folder)
|
|
|
|
return True
|
|
|
|
|
|
# check if the file already exists
|
|
def check_existence(chapter_path, manga_nocbz):
|
|
# check for folder if option nocbz is given. if nocbz is not given, the folder will be overwritten
|
|
chapter_path = Path(chapter_path)
|
|
cbz_path = chapter_path.parent / f"{chapter_path.name}.cbz"
|
|
if manga_nocbz and chapter_path.exists():
|
|
return True
|
|
# check for cbz archive
|
|
elif not manga_nocbz and cbz_path.exists():
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
|
|
# create a list of chapters
|
|
def get_chapter_list(chapters):
|
|
chapter_list = []
|
|
for chapter in chapters.split(","):
|
|
if "-" in chapter and ":" in chapter:
|
|
lower = chapter.split("-")[0].split(":")
|
|
upper = chapter.split("-")[1].split(":")
|
|
for n in range(int(lower[1]), int(upper[1]) + 1):
|
|
chapter_list.append(str(f"{lower[0]}:{n}"))
|
|
elif "-" in chapter:
|
|
lower = chapter.split("-")[0]
|
|
upper = chapter.split("-")[1]
|
|
for n in range(int(lower), int(upper) + 1):
|
|
chapter_list.append(str(n))
|
|
else:
|
|
chapter_list.append(chapter)
|
|
|
|
return chapter_list
|
|
|
|
|
|
# remove illegal characters etc
|
|
def fix_name(filename):
|
|
# remove illegal characters
|
|
filename = re.sub("[\\\/\<\>\:\;\|\?\*\!\@]", "", filename)
|
|
# remove multiple dots
|
|
filename = re.sub("([\.]{2,})", ".", filename)
|
|
# remove dot(s) at the beginning and end of the filename
|
|
filename = re.sub("(^[\.]{1,})|([\.]{1,}$)", "", filename)
|
|
# remove trailing and beginning spaces
|
|
filename = re.sub("([ \t]+$)|(^[ \t]+)", "", filename)
|
|
|
|
return filename
|
|
|
|
|
|
# create name for chapter
|
|
def get_filename(chapter_name, chapter_vol, chapter_num, manga_forcevol):
|
|
# filename for chapter
|
|
if chapter_name == "Oneshot" or chapter_num == "Oneshot":
|
|
chapter_filename = "Oneshot"
|
|
elif not chapter_name and manga_forcevol:
|
|
chapter_filename = f"Vol. {chapter_vol} Ch. {chapter_num}"
|
|
elif not chapter_name:
|
|
chapter_filename = f"Ch. {chapter_num}"
|
|
elif manga_forcevol:
|
|
chapter_filename = f"Vol. {chapter_vol} Ch. {chapter_num} - {chapter_name}"
|
|
else:
|
|
chapter_filename = f"Ch. {chapter_num} - {chapter_name}"
|
|
|
|
return chapter_filename
|