github-mirrors/bazarr
1
0
Fork 0
mirror of https://github.com/morpheus65535/bazarr.git synced 2025-04-23 22:27:17 -04:00
Code Issues Projects Releases Packages Wiki Activity

Added additional variables for custom post processing

Browse source
This commit is contained in:
JayZed 2024-01-05 21:43:37 -05:00 • committed by GitHub
parent c0bbd4f150
commit 0ac9ecea6e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 16 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
bazarr/subtitles/processing.py
View file

@ -43,6 +43,8 @@ def process_subtitle(subtitle, media_type, audio_language, path, max_score, is_u
postprocessing_cmd = settings.general.postprocessing_cmd
downloaded_provider = subtitle.provider_name
uploader = subtitle.uploader
release_info = subtitle.release_info
downloaded_language_code3 = _get_download_code3(subtitle)
downloaded_language = language_from_alpha3(downloaded_language_code3)
@ -111,7 +113,7 @@ def process_subtitle(subtitle, media_type, audio_language, path, max_score, is_u
if use_postprocessing is True:
command = pp_replace(postprocessing_cmd, path, downloaded_path, downloaded_language, downloaded_language_code2,
downloaded_language_code3, audio_language, audio_language_code2, audio_language_code3,
percent_score, subtitle_id, downloaded_provider, series_id, episode_id)
percent_score, subtitle_id, downloaded_provider, uploader, release_info, series_id, episode_id)
if media_type == 'series':
use_pp_threshold = settings.general.use_postprocessing_threshold

2
bazarr/subtitles/upload.py
View file

@ -151,7 +151,7 @@ def manual_upload_subtitle(path, language, forced, hi, media_type, subtitle, aud
if use_postprocessing:
command = pp_replace(postprocessing_cmd, path, subtitle_path, uploaded_language, uploaded_language_code2,
uploaded_language_code3, audio_language['name'], audio_language['code2'],
audio_language['code3'], 100, "1", "manual", series_id, episode_id)
audio_language['code3'], 100, "1", "manual", "user", "unknown", series_id, episode_id)
postprocessing(command, path)
set_chmod(subtitles_path=subtitle_path)

5
bazarr/utilities/post_processing.py
View file

@ -16,7 +16,8 @@ def _escape(in_str):
def pp_replace(pp_command, episode, subtitles, language, language_code2, language_code3, episode_language,
episode_language_code2, episode_language_code3, score, subtitle_id, provider, series_id, episode_id):
episode_language_code2, episode_language_code3, score, subtitle_id, provider, uploader,
release_info, series_id, episode_id):
pp_command = re.sub(r'[\'"]?{{directory}}[\'"]?', _escape(os.path.dirname(episode)), pp_command)
pp_command = re.sub(r'[\'"]?{{episode}}[\'"]?', _escape(episode), pp_command)
pp_command = re.sub(r'[\'"]?{{episode_name}}[\'"]?', _escape(os.path.splitext(os.path.basename(episode))[0]),
@ -35,6 +36,8 @@ def pp_replace(pp_command, episode, subtitles, language, language_code2, languag
pp_command = re.sub(r'[\'"]?{{score}}[\'"]?', _escape(str(score)), pp_command)
pp_command = re.sub(r'[\'"]?{{subtitle_id}}[\'"]?', _escape(str(subtitle_id)), pp_command)
pp_command = re.sub(r'[\'"]?{{provider}}[\'"]?', _escape(str(provider)), pp_command)
pp_command = re.sub(r'[\'"]?{{uploader}}[\'"]?', _escape(str(uploader)), pp_command)
pp_command = re.sub(r'[\'"]?{{release_info}}[\'"]?', _escape(str(release_info)), pp_command)
pp_command = re.sub(r'[\'"]?{{series_id}}[\'"]?', _escape(str(series_id)), pp_command)
pp_command = re.sub(r'[\'"]?{{episode_id}}[\'"]?', _escape(str(episode_id)), pp_command)
return pp_command

8
frontend/src/pages/Settings/Subtitles/index.tsx
View file

@ -98,6 +98,14 @@ const commandOptions: CommandOption[] = [
option: "provider",
description: "Provider of the subtitle file",
},
{
option: "uploader",
description: "Uploader of the subtitle file",
},
{
option: "release_info",
description: "Release info for the subtitle file",
},
{
option: "series_id",
description: "Sonarr series ID (Empty if movie)",