mirror/AnimeWwise
1
0
Fork 0
mirror of https://github.com/Escartem/AnimeWwise.git synced 2026-06-06 16:30:26 +08:00
Code Activity

refactor to audio_format var

Browse Source
This commit is contained in:
Marcel
2024-01-21 16:44:27 +00:00
parent ffc6236dca
commit 6bb091f315
1 changed files with 16 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
extract.py
View File

@@ -34,7 +34,9 @@ def main():
parse_args = parser.parse_args()
is_ogg = parse_args.ogg and True or False
print(is_ogg and "Converting to .ogg" or "Converting to .mp3")
audio_format = 'ogg' if is_ogg else 'mp3'
print(f'Format: {audio_format}')
# Initial cleanup
if os.path.exists("temp") and skips[8] != "1":
@@ -219,9 +221,9 @@ def main():
curr += 1
# updates folders and progress bar
os.makedirs(path("temp/ogg" if is_ogg else "temp/mp3"), exist_ok=True)
os.makedirs(path(f"temp/{audio_format}"), exist_ok=True)
bar = PixelBar(
f"[{curr}/{steps}] Converting to {'ogg' if is_ogg else 'mp3'} ",
f"[{curr}/{steps}] Converting to {audio_format} ",
max=len(all_files),
suffix="%(percent).1f%% - %(eta)ds left",
)
@@ -239,11 +241,7 @@ def main():
"libvorbis" if is_ogg else "libmp3lame",
"-b:a",
"192k",
path(
f"temp/ogg/{file.split('.')[0]}.ogg"
if is_ogg
else f"temp/mp3/{file.split('.')[0]}.mp3"
),
path(f"temp/{audio_format}/{file.split('.')[0]}.{audio_format}"),
]
call(args)
@@ -254,20 +252,11 @@ def main():
shutil.rmtree("temp/wav")
# update files list
all_files = [
f"{f.split('.')[0]}.ogg" if is_ogg else f"{f.split('.')[0]}.mp3"
for f in all_files
]
all_files = [f"{f.split('.')[0]}.{audio_format}" for f in all_files]
if not alone:
new_files = [
f"{f.split('.')[0]}.ogg" if is_ogg else f"{f.split('.')[0]}.mp3"
for f in new_files
]
changed_files = [
f"{f.split('.')[0]}.ogg" if is_ogg else f"{f.split('.')[0]}.mp3"
for f in changed_files
]
new_files = [f"{f.split('.')[0]}.{audio_format}" for f in new_files]
changed_files = [f"{f.split('.')[0]}.{audio_format}" for f in changed_files]
#########################
### 7 - Map filenames ###
@@ -280,13 +269,10 @@ def main():
spinner.text = f"[{curr}/{steps}] Mapping names"
spinner.start()
if alone:
os.makedirs(path(f"temp/map/unmapped"), exist_ok=True)
else:
if len(new_files) > 0:
os.makedirs(path(f"temp/map/new_files/unmapped"), exist_ok=True)
if len(changed_files) > 0:
os.makedirs(path(f"temp/map/changed_files/unmapped"), exist_ok=True)
os.makedirs(path(f"temp/map/unmapped"), exist_ok=True)
if not alone:
os.makedirs(path(f"temp/map/new_files/unmapped"), exist_ok=True)
os.makedirs(path(f"temp/map/changed_files/unmapped"), exist_ok=True)
lang = None
@@ -306,11 +292,11 @@ def main():
lang = key_data[1]
print(f"\n: {lang} detected")
dir_path = path(f"{base_path}/{key_data[0]}.{('ogg' if is_ogg else 'mp3')}")
dir_path = path(f"{base_path}/{key_data[0]}.{audio_format}")
os.makedirs(os.path.dirname(dir_path), exist_ok=True)
shutil.copy(path(f"temp/{'ogg' if is_ogg else 'mp3'}/{file}"), dir_path)
shutil.copy(path(f"temp/{audio_format}/{file}"), dir_path)
else:
shutil.copy(path(f"temp/{'ogg' if is_ogg else 'mp3'}/{file}"), path(f"{base_path}/unmapped/{file}"))
shutil.copy(path(f"temp/{audio_format}/{file}"), path(f"{base_path}/unmapped/{file}"))
# stop spinner
spinner.stop()