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