mirror of https://github.com/morpheus65535/bazarr
104 lines
3.9 KiB
Python
104 lines
3.9 KiB
Python
# coding=utf-8
|
|
|
|
from flask import request, jsonify
|
|
from flask_restful import Resource
|
|
|
|
from database import TableMovies, get_audio_profile_languages, get_profile_id
|
|
from helper import path_mappings
|
|
from get_providers import get_providers
|
|
from get_subtitle.manual import manual_search, manual_download_subtitle
|
|
from utils import history_log_movie
|
|
from config import settings
|
|
from notifier import send_notifications_movie
|
|
from list_subtitles import store_subtitles_movie
|
|
|
|
from ..utils import authenticate
|
|
import logging
|
|
|
|
|
|
class ProviderMovies(Resource):
|
|
@authenticate
|
|
def get(self):
|
|
# Manual Search
|
|
radarrId = request.args.get('radarrid')
|
|
movieInfo = TableMovies.select(TableMovies.title,
|
|
TableMovies.path,
|
|
TableMovies.sceneName,
|
|
TableMovies.profileId) \
|
|
.where(TableMovies.radarrId == radarrId) \
|
|
.dicts() \
|
|
.get_or_none()
|
|
|
|
if not movieInfo:
|
|
return 'Movie not found', 500
|
|
|
|
title = movieInfo['title']
|
|
moviePath = path_mappings.path_replace_movie(movieInfo['path'])
|
|
sceneName = movieInfo['sceneName'] or "None"
|
|
profileId = movieInfo['profileId']
|
|
|
|
providers_list = get_providers()
|
|
|
|
data = manual_search(moviePath, profileId, providers_list, sceneName, title, 'movie')
|
|
if not data:
|
|
data = []
|
|
return jsonify(data=data)
|
|
|
|
@authenticate
|
|
def post(self):
|
|
# Manual Download
|
|
radarrId = request.args.get('radarrid')
|
|
movieInfo = TableMovies.select(TableMovies.title,
|
|
TableMovies.path,
|
|
TableMovies.sceneName,
|
|
TableMovies.audio_language) \
|
|
.where(TableMovies.radarrId == radarrId) \
|
|
.dicts() \
|
|
.get_or_none()
|
|
|
|
if not movieInfo:
|
|
return 'Movie not found', 500
|
|
|
|
title = movieInfo['title']
|
|
moviePath = path_mappings.path_replace_movie(movieInfo['path'])
|
|
sceneName = movieInfo['sceneName'] or "None"
|
|
|
|
hi = request.form.get('hi').capitalize()
|
|
forced = request.form.get('forced').capitalize()
|
|
use_original_format = request.form.get('original_format').capitalize()
|
|
logging.debug(f"use_original_format {use_original_format}")
|
|
selected_provider = request.form.get('provider')
|
|
subtitle = request.form.get('subtitle')
|
|
|
|
audio_language_list = get_audio_profile_languages(movie_id=radarrId)
|
|
if len(audio_language_list) > 0:
|
|
audio_language = audio_language_list[0]['name']
|
|
else:
|
|
audio_language = 'None'
|
|
|
|
try:
|
|
result = manual_download_subtitle(moviePath, audio_language, hi, forced, subtitle, selected_provider,
|
|
sceneName, title, 'movie', use_original_format, profile_id=get_profile_id(movie_id=radarrId))
|
|
if result is not None:
|
|
message = result[0]
|
|
path = result[1]
|
|
forced = result[5]
|
|
if result[8]:
|
|
language_code = result[2] + ":hi"
|
|
elif forced:
|
|
language_code = result[2] + ":forced"
|
|
else:
|
|
language_code = result[2]
|
|
provider = result[3]
|
|
score = result[4]
|
|
subs_id = result[6]
|
|
subs_path = result[7]
|
|
history_log_movie(2, radarrId, message, path, language_code, provider, score, subs_id, subs_path)
|
|
if not settings.general.getboolean('dont_notify_manual_actions'):
|
|
send_notifications_movie(radarrId, message)
|
|
store_subtitles_movie(path, moviePath)
|
|
except OSError:
|
|
pass
|
|
|
|
return '', 204
|