dasemu/Transcriptarr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

removed check function and moved it under gen_subtitles_queue

Browse Source
This commit is contained in:
McCloudS
2024-04-20 08:45:56 -06:00
committed by GitHub
parent 8ddebaf701
commit 60520cef63
1 changed files with 28 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
subgen.py
View File

@@ -75,7 +75,7 @@ def update_env_variables():
procaddedmedia = convert_to_bool(os.getenv('PROCADDEDMEDIA', True))
procmediaonplay = convert_to_bool(os.getenv('PROCMEDIAONPLAY', True))
namesublang = os.getenv('NAMESUBLANG', 'aa')
skipifinternalsublang = os.getenv('SKIPIFINTERNALSUBLANG', 'eng')
skipifinternalsublang = os.getenv('SKIPIFINTERNALSUBLANG', 'aa')
webhookport = int(os.getenv('WEBHOOKPORT', 9000))
word_level_highlight = convert_to_bool(os.getenv('WORD_LEVEL_HIGHLIGHT', False))
debug = convert_to_bool(os.getenv('DEBUG', True))
@@ -310,7 +310,6 @@ def receive_tautulli_webhook(
logging.debug("Path of file: " + fullpath)
# gen_subtitles(path_mapping(fullpath), transcribe_or_translate, True)
if gen_subtitles_check(path_mapping(fullpath)):
gen_subtitles_queue(path_mapping(fullpath), transcribe_or_translate, True)
else:
return {
@@ -339,7 +338,6 @@ def receive_plex_webhook(
logging.debug("Path of file: " + fullpath)
# gen_subtitles(path_mapping(fullpath), transcribe_or_translate, True)
if gen_subtitles_check(path_mapping(fullpath)):
gen_subtitles_queue(path_mapping(fullpath), transcribe_or_translate, True)
refresh_plex_metadata(plex_json['Metadata']['ratingKey'], plexserver, plextoken)
logging.info(f"Metadata for item {plex_json['Metadata']['ratingKey']} refreshed successfully.")
@@ -366,7 +364,6 @@ def receive_jellyfin_webhook(
logging.debug(f"Path of file: {fullpath}")
# gen_subtitles(path_mapping(fullpath), transcribe_or_translate, True)
if gen_subtitles_check(path_mapping(fullpath)):
gen_subtitles_queue(path_mapping(fullpath), transcribe_or_translate, True)
try:
refresh_jellyfin_metadata(ItemId, jellyfinserver, jellyfintoken)
@@ -401,7 +398,6 @@ def receive_emby_webhook(
if event == "library.new" and procaddedmedia or event == "playback.start" and procmediaonplay:
logging.debug("Path of file: " + fullpath)
# gen_subtitles(path_mapping(fullpath), transcribe_or_translate, True)
if gen_subtitles_check(path_mapping(fullpath)):
gen_subtitles_queue(path_mapping(fullpath), transcribe_or_translate, True)
return ""
@@ -513,28 +509,6 @@ def write_lrc(result, file_path):
fraction = int((segment.start - int(segment.start)) * 100)
file.write(f"[{minutes:02d}:{seconds:02d}.{fraction:02d}] {segment.text}\n")
def gen_subtitles_check(file_path: str):
if not has_audio(file_path):
logging.debug(f"{file_path} doesn't have any audio to transcribe!")
return False
if file_path in files_to_transcribe:
logging.info(f"File {os.path.basename(file_path)} is already in the transcription list. Skipping.")
return
message = None
if has_subtitle_language(file_path, skipifinternalsublang):
message = f"{file_path} already has an internal subtitle we want, skipping generation"
elif os.path.exists(file_path.rsplit('.', 1)[0] + subextension):
message = f"{file_path} already has a subtitle created for this, skipping it"
elif os.path.exists(file_path.rsplit('.', 1)[0] + subextensionSDH):
message = f"{file_path} already has a SDH subtitle created for this, skipping it"
if message:
logging.info(message)
return False
return True
def gen_subtitles(file_path: str, transcription_type: str, add_to_front=True, force_language=None) -> None:
"""Generates subtitles for a video file.
@@ -594,6 +568,26 @@ def gen_subtitles(file_path: str, transcription_type: str, add_to_front=True, fo
def gen_subtitles_queue(file_path: str, transcription_type: str, add_to_front=True, force_language=None) -> None:
global task_queue
if not has_audio(file_path):
logging.debug(f"{file_path} doesn't have any audio to transcribe!")
return
if file_path in files_to_transcribe:
logging.info(f"File {os.path.basename(file_path)} is already in the transcription list. Skipping.")
return
message = None
if has_subtitle_language(file_path, skipifinternalsublang):
message = f"{file_path} already has an internal subtitle we want, skipping generation"
elif os.path.exists(file_path.rsplit('.', 1)[0] + subextension):
message = f"{file_path} already has a subtitle created for this, skipping it"
elif os.path.exists(file_path.rsplit('.', 1)[0] + subextensionSDH):
message = f"{file_path} already has a SDH subtitle created for this, skipping it"
if message:
logging.info(message)
return
task = {
'path': file_path,
'transcribe_or_translate': transcription_type,
@@ -773,7 +767,6 @@ if monitor:
# Call the gen_subtitles function
logging.info(f"File: {path_mapping(file_path)} was added")
# gen_subtitles(path_mapping(file_path), transcribe_or_translate, False)
if gen_subtitles_check(path_mapping(fullpath)):
gen_subtitles_queue(path_mapping(file_path), transcribe_or_translate, False)
def on_created(self, event):
self.create_subtitle(event)
@@ -790,13 +783,11 @@ def transcribe_existing(transcribe_folders, forceLanguage=None):
for file in files:
file_path = os.path.join(root, file)
# gen_subtitles(path_mapping(file_path), transcribe_or_translate, False, forceLanguage)
if gen_subtitles_check(path_mapping(file_path)):
gen_subtitles_queue(path_mapping(file_path), transcribe_or_translate, False, forceLanguage)
# if the path specified was actually a single file and not a folder, process it
if os.path.isfile(path):
if has_audio(path):
# gen_subtitles(path_mapping(path), transcribe_or_translate, False, forceLanguage)
if gen_subtitles_check(path_mapping(path)):
gen_subtitles_queue(path_mapping(path), transcribe_or_translate, False, forceLanguage)
# Set up the observer to watch for new files
if monitor: