mirror of https://github.com/morpheus65535/bazarr
This should be a good start of an API to get new UI on track.
This commit is contained in:
Louis Vézina
parent
1e99de61b3
commit
cafa6ded0b
254
bazarr/api.py
254
bazarr/api.py
|
|
@ -1,23 +1,26 @@
|
|||
import os
|
||||
import ast
|
||||
import libs
|
||||
from datetime import timedelta
|
||||
import datetime
|
||||
import pretty
|
||||
|
||||
from get_args import args
|
||||
from config import settings
|
||||
|
||||
from init import *
|
||||
import logging
|
||||
from database import database
|
||||
from helper import path_replace, path_replace_reverse, path_replace_movie, path_replace_reverse_movie
|
||||
from get_languages import load_language_in_db, alpha2_from_language, alpha3_from_language, language_from_alpha2, \
|
||||
alpha3_from_alpha2
|
||||
|
||||
from flask import Flask, jsonify, request
|
||||
from flask_restful import Resource, Api
|
||||
|
||||
app = Flask(__name__)
|
||||
api = Api(app)
|
||||
|
||||
import os
|
||||
import ast
|
||||
import libs
|
||||
|
||||
from get_args import args
|
||||
from config import settings, url_sonarr, url_radarr, url_radarr_short, url_sonarr_short, base_url
|
||||
|
||||
from init import *
|
||||
import logging
|
||||
from database import database, dict_mapper
|
||||
from helper import path_replace, path_replace_reverse, path_replace_movie, path_replace_reverse_movie
|
||||
from get_languages import load_language_in_db, alpha2_from_language, alpha3_from_language, language_from_alpha2, \
|
||||
alpha3_from_alpha2
|
||||
|
||||
load_language_in_db()
|
||||
|
||||
|
||||
|
|
@ -167,10 +170,233 @@ class Movies(Resource):
|
|||
return jsonify(result)
|
||||
|
||||
|
||||
class HistorySeries(Resource):
|
||||
def get(self):
|
||||
upgradable_episodes_not_perfect = []
|
||||
if settings.general.getboolean('upgrade_subs'):
|
||||
days_to_upgrade_subs = settings.general.days_to_upgrade_subs
|
||||
minimum_timestamp = ((datetime.datetime.now() - timedelta(days=int(days_to_upgrade_subs))) -
|
||||
datetime.datetime(1970, 1, 1)).total_seconds()
|
||||
|
||||
if settings.general.getboolean('upgrade_manual'):
|
||||
query_actions = [1, 2, 3]
|
||||
else:
|
||||
query_actions = [1, 3]
|
||||
|
||||
if settings.sonarr.getboolean('only_monitored'):
|
||||
series_monitored_only_query_string = " AND monitored='True'"
|
||||
else:
|
||||
series_monitored_only_query_string = ''
|
||||
|
||||
upgradable_episodes = database.execute(
|
||||
"SELECT video_path, MAX(timestamp) as timestamp, score FROM table_history "
|
||||
"INNER JOIN table_episodes on table_episodes.sonarrEpisodeId = "
|
||||
"table_history.sonarrEpisodeId WHERE action IN (" +
|
||||
','.join(map(str, query_actions)) + ") AND timestamp > ? AND "
|
||||
"score is not null" + series_monitored_only_query_string + " GROUP BY "
|
||||
"table_history.video_path, table_history.language",
|
||||
(minimum_timestamp,))
|
||||
|
||||
for upgradable_episode in upgradable_episodes:
|
||||
if upgradable_episode['timestamp'] > minimum_timestamp:
|
||||
try:
|
||||
int(upgradable_episode['score'])
|
||||
except ValueError:
|
||||
pass
|
||||
else:
|
||||
if int(upgradable_episode['score']) < 360:
|
||||
upgradable_episodes_not_perfect.append(upgradable_episode)
|
||||
|
||||
data = database.execute("SELECT table_history.action, table_shows.title as seriesTitle, "
|
||||
"table_episodes.season || 'x' || table_episodes.episode as episode_number, "
|
||||
"table_episodes.title as episodeTitle, table_history.timestamp, "
|
||||
"table_history.description, table_history.sonarrSeriesId, table_episodes.path, "
|
||||
"table_history.language, table_history.score FROM table_history LEFT JOIN table_shows "
|
||||
"on table_shows.sonarrSeriesId = table_history.sonarrSeriesId LEFT JOIN table_episodes "
|
||||
"on table_episodes.sonarrEpisodeId = table_history.sonarrEpisodeId WHERE "
|
||||
"table_episodes.title is not NULL ORDER BY timestamp DESC")
|
||||
|
||||
for item in data:
|
||||
# Mark episode as upgradable or not
|
||||
if {"video_path": str(item['path']), "timestamp": float(item['timestamp']), "score": str(item['score'])} in upgradable_episodes_not_perfect:
|
||||
item.update({"upgradable": True})
|
||||
else:
|
||||
item.update({"upgradable": False})
|
||||
|
||||
# Parse language
|
||||
if item['language'] and item['language'] != 'None':
|
||||
splitted_language = item['language'].split(':')
|
||||
item['language'] = {"name": language_from_alpha2(splitted_language[0]),
|
||||
"code2": splitted_language[0],
|
||||
"code3": alpha3_from_alpha2(splitted_language[0]),
|
||||
"forced": True if len(splitted_language) > 1 else False}
|
||||
|
||||
# Make timestamp pretty
|
||||
if item['timestamp']:
|
||||
item['timestamp'] = pretty.date(int(item['timestamp']))
|
||||
|
||||
# Provide mapped path
|
||||
mapped_path = path_replace(item['path'])
|
||||
item.update({"mapped_path": mapped_path})
|
||||
|
||||
# Confirm if path exist
|
||||
item.update({"exist": os.path.isfile(mapped_path)})
|
||||
|
||||
return jsonify(data)
|
||||
|
||||
|
||||
class HistoryMovies(Resource):
|
||||
def get(self):
|
||||
upgradable_movies = []
|
||||
upgradable_movies_not_perfect = []
|
||||
if settings.general.getboolean('upgrade_subs'):
|
||||
days_to_upgrade_subs = settings.general.days_to_upgrade_subs
|
||||
minimum_timestamp = ((datetime.datetime.now() - timedelta(days=int(days_to_upgrade_subs))) -
|
||||
datetime.datetime(1970, 1, 1)).total_seconds()
|
||||
|
||||
if settings.radarr.getboolean('only_monitored'):
|
||||
movies_monitored_only_query_string = ' AND table_movies.monitored = "True"'
|
||||
else:
|
||||
movies_monitored_only_query_string = ""
|
||||
|
||||
if settings.general.getboolean('upgrade_manual'):
|
||||
query_actions = [1, 2, 3]
|
||||
else:
|
||||
query_actions = [1, 3]
|
||||
|
||||
upgradable_movies = database.execute(
|
||||
"SELECT video_path, MAX(timestamp) as timestamp, score FROM table_history_movie "
|
||||
"INNER JOIN table_movies on table_movies.radarrId=table_history_movie.radarrId WHERE action IN (" +
|
||||
','.join(map(str, query_actions)) + ") AND timestamp > ? AND score is not NULL" +
|
||||
movies_monitored_only_query_string + " GROUP BY video_path, language", (minimum_timestamp,))
|
||||
|
||||
for upgradable_movie in upgradable_movies:
|
||||
if upgradable_movie['timestamp'] > minimum_timestamp:
|
||||
try:
|
||||
int(upgradable_movie['score'])
|
||||
except ValueError:
|
||||
pass
|
||||
else:
|
||||
if int(upgradable_movie['score']) < 120:
|
||||
upgradable_movies_not_perfect.append(upgradable_movie)
|
||||
|
||||
data = database.execute("SELECT table_history_movie.action, table_movies.title, table_history_movie.timestamp, "
|
||||
"table_history_movie.description, table_history_movie.radarrId, "
|
||||
"table_history_movie.video_path, table_history_movie.language, "
|
||||
"table_history_movie.score FROM table_history_movie LEFT JOIN table_movies on "
|
||||
"table_movies.radarrId = table_history_movie.radarrId ORDER BY timestamp DESC")
|
||||
|
||||
for item in data:
|
||||
# Mark movies as upgradable or not
|
||||
if {"video_path": str(item['video_path']), "timestamp": float(item['timestamp']), "score": str(item['score'])} in upgradable_movies_not_perfect:
|
||||
item.update({"upgradable": True})
|
||||
else:
|
||||
item.update({"upgradable": False})
|
||||
|
||||
# Parse language
|
||||
if item['language'] and item['language'] != 'None':
|
||||
splitted_language = item['language'].split(':')
|
||||
item['language'] = {"name": language_from_alpha2(splitted_language[0]),
|
||||
"code2": splitted_language[0],
|
||||
"code3": alpha3_from_alpha2(splitted_language[0]),
|
||||
"forced": True if len(splitted_language) > 1 else False}
|
||||
|
||||
# Make timestamp pretty
|
||||
if item['timestamp']:
|
||||
item['timestamp'] = pretty.date(int(item['timestamp']))
|
||||
|
||||
if item['video_path']:
|
||||
# Provide mapped path
|
||||
mapped_path = path_replace_movie(item['video_path'])
|
||||
item.update({"mapped_path": mapped_path})
|
||||
|
||||
# Confirm if path exist
|
||||
item.update({"exist": os.path.isfile(mapped_path)})
|
||||
else:
|
||||
item.update({"mapped_path": None})
|
||||
item.update({"exist": False})
|
||||
|
||||
return jsonify(data)
|
||||
|
||||
|
||||
class WantedSeries(Resource):
|
||||
def get(self):
|
||||
if settings.sonarr.getboolean('only_monitored'):
|
||||
monitored_only_query_string = " AND monitored='True'"
|
||||
else:
|
||||
monitored_only_query_string = ''
|
||||
|
||||
data = database.execute("SELECT table_shows.title as seriesTitle, "
|
||||
"table_episodes.season || 'x' || table_episodes.episode as episode_number, "
|
||||
"table_episodes.title as episodeTitle, table_episodes.missing_subtitles, "
|
||||
"table_episodes.sonarrSeriesId, table_episodes.path, table_shows.hearing_impaired, "
|
||||
"table_episodes.sonarrEpisodeId, table_episodes.scene_name, "
|
||||
"table_episodes.failedAttempts FROM table_episodes INNER JOIN table_shows on "
|
||||
"table_shows.sonarrSeriesId = table_episodes.sonarrSeriesId WHERE "
|
||||
"table_episodes.missing_subtitles != '[]'" + monitored_only_query_string +
|
||||
" ORDER BY table_episodes._rowid_ DESC")
|
||||
|
||||
for item in data:
|
||||
# Parse missing subtitles
|
||||
if item['missing_subtitles']:
|
||||
item.update({"missing_subtitles": ast.literal_eval(item['missing_subtitles'])})
|
||||
for i, subs in enumerate(item['missing_subtitles']):
|
||||
splitted_subs = subs.split(':')
|
||||
item['missing_subtitles'][i] = {"name": language_from_alpha2(splitted_subs[0]),
|
||||
"code2": splitted_subs[0],
|
||||
"code3": alpha3_from_alpha2(splitted_subs[0]),
|
||||
"forced": True if len(splitted_subs) > 1 else False}
|
||||
|
||||
# Provide mapped path
|
||||
mapped_path = path_replace(item['path'])
|
||||
item.update({"mapped_path": mapped_path})
|
||||
|
||||
# Confirm if path exist
|
||||
item.update({"exist": os.path.isfile(mapped_path)})
|
||||
|
||||
return jsonify(data)
|
||||
|
||||
|
||||
class WantedMovies(Resource):
|
||||
def get(self):
|
||||
if settings.radarr.getboolean('only_monitored'):
|
||||
monitored_only_query_string = " AND monitored='True'"
|
||||
else:
|
||||
monitored_only_query_string = ''
|
||||
|
||||
data = database.execute("SELECT title, missing_subtitles, radarrId, path, hearing_impaired, sceneName, "
|
||||
"failedAttempts FROM table_movies WHERE missing_subtitles != '[]'" +
|
||||
monitored_only_query_string + " ORDER BY _rowid_ DESC")
|
||||
|
||||
for item in data:
|
||||
# Parse missing subtitles
|
||||
if item['missing_subtitles']:
|
||||
item.update({"missing_subtitles": ast.literal_eval(item['missing_subtitles'])})
|
||||
for i, subs in enumerate(item['missing_subtitles']):
|
||||
splitted_subs = subs.split(':')
|
||||
item['missing_subtitles'][i] = {"name": language_from_alpha2(splitted_subs[0]),
|
||||
"code2": splitted_subs[0],
|
||||
"code3": alpha3_from_alpha2(splitted_subs[0]),
|
||||
"forced": True if len(splitted_subs) > 1 else False}
|
||||
|
||||
# Provide mapped path
|
||||
mapped_path = path_replace_movie(item['path'])
|
||||
item.update({"mapped_path": mapped_path})
|
||||
|
||||
# Confirm if path exist
|
||||
item.update({"exist": os.path.isfile(mapped_path)})
|
||||
|
||||
return jsonify(data)
|
||||
|
||||
|
||||
api.add_resource(Badges, '/api/badges')
|
||||
api.add_resource(Series, '/api/series')
|
||||
api.add_resource(Episodes, '/api/episodes')
|
||||
api.add_resource(Movies, '/api/movies')
|
||||
api.add_resource(HistorySeries, '/api/history_series')
|
||||
api.add_resource(HistoryMovies, '/api/history_movies')
|
||||
api.add_resource(WantedSeries, '/api/wanted_series')
|
||||
api.add_resource(WantedMovies, '/api/wanted_movies')
|
||||
|
||||
if __name__ == '__main__':
|
||||
app.run(debug=True)
|
||||
|
|
|
|||
Loading…
Reference in New Issue