mirror of https://github.com/morpheus65535/bazarr
85 lines
3.1 KiB
Python
85 lines
3.1 KiB
Python
# coding=utf-8
|
|
|
|
import operator
|
|
|
|
from flask_restx import Resource, Namespace, reqparse, fields
|
|
from functools import reduce
|
|
|
|
from app.database import get_exclusion_clause, TableMovies, database, select, func
|
|
from api.swaggerui import subtitles_language_model
|
|
|
|
from api.utils import authenticate, postprocess
|
|
|
|
|
|
api_ns_movies_wanted = Namespace('Movies Wanted', description='List movies wanted subtitles')
|
|
|
|
|
|
@api_ns_movies_wanted.route('movies/wanted')
|
|
class MoviesWanted(Resource):
|
|
get_request_parser = reqparse.RequestParser()
|
|
get_request_parser.add_argument('start', type=int, required=False, default=0, help='Paging start integer')
|
|
get_request_parser.add_argument('length', type=int, required=False, default=-1, help='Paging length integer')
|
|
get_request_parser.add_argument('radarrid[]', type=int, action='append', required=False, default=[],
|
|
help='Movies ID to list')
|
|
|
|
get_subtitles_language_model = api_ns_movies_wanted.model('subtitles_language_model', subtitles_language_model)
|
|
|
|
data_model = api_ns_movies_wanted.model('wanted_movies_data_model', {
|
|
'title': fields.String(),
|
|
'missing_subtitles': fields.Nested(get_subtitles_language_model),
|
|
'radarrId': fields.Integer(),
|
|
'sceneName': fields.String(),
|
|
'tags': fields.List(fields.String),
|
|
})
|
|
|
|
get_response_model = api_ns_movies_wanted.model('MovieWantedGetResponse', {
|
|
'data': fields.Nested(data_model),
|
|
'total': fields.Integer(),
|
|
})
|
|
|
|
@authenticate
|
|
@api_ns_movies_wanted.marshal_with(get_response_model, code=200)
|
|
@api_ns_movies_wanted.response(401, 'Not Authenticated')
|
|
@api_ns_movies_wanted.doc(parser=get_request_parser)
|
|
def get(self):
|
|
"""List movies wanted subtitles"""
|
|
args = self.get_request_parser.parse_args()
|
|
radarrid = args.get("radarrid[]")
|
|
|
|
wanted_conditions = [(TableMovies.missing_subtitles != '[]')]
|
|
if len(radarrid) > 0:
|
|
wanted_conditions.append((TableMovies.radarrId.in_(radarrid)))
|
|
start = 0
|
|
length = 0
|
|
else:
|
|
start = args.get('start')
|
|
length = args.get('length')
|
|
|
|
wanted_conditions += get_exclusion_clause('movie')
|
|
wanted_condition = reduce(operator.and_, wanted_conditions)
|
|
|
|
stmt = select(TableMovies.title,
|
|
TableMovies.missing_subtitles,
|
|
TableMovies.radarrId,
|
|
TableMovies.sceneName,
|
|
TableMovies.tags) \
|
|
.where(wanted_condition)
|
|
if length > 0:
|
|
stmt = stmt.order_by(TableMovies.radarrId.desc()).limit(length).offset(start)
|
|
|
|
results = [postprocess({
|
|
'title': x.title,
|
|
'missing_subtitles': x.missing_subtitles,
|
|
'radarrId': x.radarrId,
|
|
'sceneName': x.sceneName,
|
|
'tags': x.tags,
|
|
}) for x in database.execute(stmt).all()]
|
|
|
|
count = database.execute(
|
|
select(func.count())
|
|
.select_from(TableMovies)
|
|
.where(wanted_condition)) \
|
|
.scalar()
|
|
|
|
return {'data': results, 'total': count}
|