bazarr/bazarr.py

804 lines
36 KiB
Python
Raw Normal View History

2018-02-16 03:18:28 +00:00
bazarr_version = '0.3.3'
2017-11-16 02:25:14 +00:00
import os
import sys
2018-01-17 14:41:27 +00:00
sys.path.insert(0,os.path.join(os.path.dirname(__file__), 'libs/'))
2017-12-22 00:49:39 +00:00
from bottle import route, run, template, static_file, request, redirect, response
2017-09-16 00:49:46 +00:00
import bottle
2017-11-20 21:25:34 +00:00
bottle.debug(True)
bottle.TEMPLATES.clear()
2017-10-22 01:09:04 +00:00
2017-10-22 00:43:32 +00:00
bottle.TEMPLATE_PATH.insert(0,os.path.join(os.path.dirname(__file__), 'views/'))
2017-09-16 00:49:46 +00:00
import sqlite3
2018-01-30 20:25:49 +00:00
from json import dumps
2018-01-01 17:46:53 +00:00
import itertools
import operator
import requests
import pycountry
import pretty
2018-01-08 02:47:40 +00:00
from datetime import datetime, timedelta
2018-01-01 17:46:53 +00:00
from PIL import Image
from io import BytesIO
from fdsend import send_file
import urllib
import math
2018-01-07 17:29:11 +00:00
import ast
2018-01-01 17:46:53 +00:00
from init_db import *
2018-01-05 04:24:11 +00:00
from update_db import *
from update_modules import *
2017-09-16 00:49:46 +00:00
2017-10-16 23:27:19 +00:00
import logging
from logging.handlers import TimedRotatingFileHandler
2017-11-16 02:17:51 +00:00
2017-10-16 23:27:19 +00:00
logger = logging.getLogger('waitress')
2017-12-05 00:01:10 +00:00
db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
2017-10-16 23:27:19 +00:00
c = db.cursor()
c.execute("SELECT log_level FROM table_settings_general")
log_level = c.fetchone()
2017-11-30 19:35:03 +00:00
c.close()
2017-10-16 23:27:19 +00:00
log_level = log_level[0]
if log_level is None:
log_level = "INFO"
2017-10-16 23:27:19 +00:00
log_level = getattr(logging, log_level)
2017-10-16 23:27:19 +00:00
class OneLineExceptionFormatter(logging.Formatter):
def formatException(self, exc_info):
"""
Format an exception so that it prints on a single line.
"""
result = super(OneLineExceptionFormatter, self).formatException(exc_info)
return repr(result) # or format into one line however you want to
def format(self, record):
s = super(OneLineExceptionFormatter, self).format(record)
if record.exc_text:
s = s.replace('\n', '') + '|'
return s
def configure_logging():
2017-12-13 17:42:09 +00:00
global fh
2017-10-20 12:59:21 +00:00
fh = TimedRotatingFileHandler(os.path.join(os.path.dirname(__file__), 'data/log/bazarr.log'), when="midnight", interval=1, backupCount=7)
2017-10-16 23:27:19 +00:00
f = OneLineExceptionFormatter('%(asctime)s|%(levelname)s|%(message)s|',
'%d/%m/%Y %H:%M:%S')
fh.setFormatter(f)
2017-12-05 00:01:10 +00:00
logging.getLogger("enzyme").setLevel(logging.CRITICAL)
2017-11-21 21:03:29 +00:00
logging.getLogger("apscheduler").setLevel(logging.WARNING)
logging.getLogger("subliminal").setLevel(logging.CRITICAL)
2017-10-16 23:27:19 +00:00
root = logging.getLogger()
root.setLevel(log_level)
root.addHandler(fh)
configure_logging()
2017-12-31 21:28:06 +00:00
from get_languages import *
from get_providers import *
from get_series import *
from get_episodes import *
from get_general_settings import *
from get_sonarr_settings import *
from check_update import *
from list_subtitles import *
from get_subtitle import *
from utils import *
from scheduler import *
from notifier import send_notifications
2017-12-13 17:42:09 +00:00
2018-01-01 17:46:53 +00:00
# Reset restart required warning on start
conn = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
c = conn.cursor()
c.execute("UPDATE table_settings_general SET configured = 0, updated = 0")
conn.commit()
c.close()
2017-11-09 21:13:14 +00:00
@route('/')
def redirect_root():
redirect (base_url)
@route(base_url + 'static/:path#.+#', name='static')
2017-09-16 00:49:46 +00:00
def static(path):
2017-10-22 01:25:31 +00:00
return static_file(path, root=os.path.join(os.path.dirname(__file__), 'static'))
2017-09-16 00:49:46 +00:00
2017-12-13 17:42:09 +00:00
@route(base_url + 'emptylog')
def emptylog():
ref = request.environ['HTTP_REFERER']
fh.doRollover()
logging.info('Log file emptied')
redirect(ref)
@route(base_url + 'bazarr.log')
def download_log():
2017-12-13 17:55:03 +00:00
return static_file('bazarr.log', root=os.path.join(os.path.dirname(__file__), 'data/log/'), download='bazarr.log')
2017-12-13 17:42:09 +00:00
2017-11-09 21:13:14 +00:00
@route(base_url + 'image_proxy/<url:path>', method='GET')
2017-09-16 00:49:46 +00:00
def image_proxy(url):
from get_sonarr_settings import get_sonarr_settings
url_sonarr_short = get_sonarr_settings()[1]
2017-09-16 00:49:46 +00:00
img_pil = Image.open(BytesIO(requests.get(url_sonarr_short + '/' + url).content))
img_buffer = BytesIO()
img_pil.tobytes()
img_pil.save(img_buffer, img_pil.format)
img_buffer.seek(0)
return send_file(img_buffer, ctype=img_pil.format)
2017-11-09 21:13:14 +00:00
@route(base_url)
2017-09-16 00:49:46 +00:00
def series():
import update_db
2018-01-10 16:44:47 +00:00
single_language = get_general_settings()[7]
2017-12-05 00:01:10 +00:00
db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
2017-09-16 00:49:46 +00:00
db.create_function("path_substitution", 1, path_replace)
c = db.cursor()
2017-12-06 02:05:49 +00:00
c.execute("SELECT COUNT(*) FROM table_shows")
missing_count = c.fetchone()
missing_count = missing_count[0]
page = request.GET.page
if page == "":
page = "1"
offset = (int(page) - 1) * 15
max_page = int(math.ceil(missing_count / 15.0))
2017-12-06 02:05:49 +00:00
c.execute("SELECT tvdbId, title, path_substitution(path), languages, hearing_impaired, sonarrSeriesId, poster, audio_language FROM table_shows ORDER BY title ASC LIMIT 15 OFFSET ?", (offset,))
2017-09-16 00:49:46 +00:00
data = c.fetchall()
c.execute("SELECT code2, name FROM table_settings_languages WHERE enabled = 1")
languages = c.fetchall()
c.close()
2018-01-10 16:44:47 +00:00
output = template('series', __file__=__file__, bazarr_version=bazarr_version, rows=data, languages=languages, missing_count=missing_count, page=page, max_page=max_page, base_url=base_url, single_language=single_language)
2017-09-16 00:49:46 +00:00
return output
2018-01-07 17:29:11 +00:00
@route(base_url + 'serieseditor')
def serieseditor():
2018-01-10 16:44:47 +00:00
single_language = get_general_settings()[7]
2018-01-07 17:29:11 +00:00
db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
db.create_function("path_substitution", 1, path_replace)
c = db.cursor()
c.execute("SELECT COUNT(*) FROM table_shows")
missing_count = c.fetchone()
missing_count = missing_count[0]
c.execute("SELECT tvdbId, title, path_substitution(path), languages, hearing_impaired, sonarrSeriesId, poster, audio_language FROM table_shows ORDER BY title ASC")
data = c.fetchall()
c.execute("SELECT code2, name FROM table_settings_languages WHERE enabled = 1")
languages = c.fetchall()
c.close()
2018-01-10 16:44:47 +00:00
output = template('serieseditor', __file__=__file__, bazarr_version=bazarr_version, rows=data, languages=languages, missing_count=missing_count, base_url=base_url, single_language=single_language)
2018-01-07 17:29:11 +00:00
return output
2017-12-22 00:49:39 +00:00
@route(base_url + 'series_json/<query>', method='GET')
def series_json(query):
db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
c = db.cursor()
c.execute("SELECT title, sonarrSeriesId FROM table_shows WHERE title LIKE ? ORDER BY title", ('%'+query+'%',))
data = c.fetchall()
series_list = []
for serie in data:
series_list.append(dict([('name', serie[0]), ('url', base_url + 'episodes/' + str(serie[1]))]))
response.content_type = 'application/json'
return dict(items=series_list)
2017-11-09 21:13:14 +00:00
@route(base_url + 'edit_series/<no:int>', method='POST')
2017-09-16 00:49:46 +00:00
def edit_series(no):
2017-10-18 18:54:24 +00:00
ref = request.environ['HTTP_REFERER']
2017-09-16 00:49:46 +00:00
lang = request.forms.getall('languages')
if len(lang) > 0:
2018-01-07 17:29:11 +00:00
pass
2017-09-16 00:49:46 +00:00
else:
2018-01-07 17:29:11 +00:00
lang = 'None'
2017-09-16 00:49:46 +00:00
hi = request.forms.get('hearing_impaired')
if hi == "on":
hi = "True"
else:
hi = "False"
2017-12-05 00:01:10 +00:00
conn = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
2017-09-16 00:49:46 +00:00
c = conn.cursor()
2018-01-07 17:29:11 +00:00
c.execute("UPDATE table_shows SET languages = ?, hearing_impaired = ? WHERE sonarrSeriesId LIKE ?", (str(lang), hi, no))
2017-09-16 00:49:46 +00:00
conn.commit()
c.close()
2017-10-16 23:27:19 +00:00
list_missing_subtitles(no)
2017-10-18 18:54:24 +00:00
redirect(ref)
2017-09-16 00:49:46 +00:00
2018-01-07 17:29:11 +00:00
@route(base_url + 'edit_serieseditor', method='POST')
def edit_serieseditor():
ref = request.environ['HTTP_REFERER']
series = request.forms.get('series')
series = ast.literal_eval(str('[' + series + ']'))
lang = request.forms.getall('languages')
hi = request.forms.get('hearing_impaired')
conn = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
c = conn.cursor()
for serie in series:
2018-01-19 02:10:16 +00:00
if str(lang) != "[]" and str(lang) != "['']":
2018-01-07 17:29:11 +00:00
if str(lang) == "['None']":
lang = 'None'
else:
lang = str(lang)
c.execute("UPDATE table_shows SET languages = ? WHERE sonarrSeriesId LIKE ?", (lang, serie))
if hi != '':
c.execute("UPDATE table_shows SET hearing_impaired = ? WHERE sonarrSeriesId LIKE ?", (hi, serie))
conn.commit()
c.close()
for serie in series:
list_missing_subtitles(serie)
redirect(ref)
2017-11-09 21:13:14 +00:00
@route(base_url + 'episodes/<no:int>', method='GET')
2017-09-16 00:49:46 +00:00
def episodes(no):
2018-01-10 16:44:47 +00:00
single_language = get_general_settings()[7]
url_sonarr_short = get_sonarr_settings()[1]
2017-12-05 00:01:10 +00:00
conn = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
2017-09-16 00:49:46 +00:00
conn.create_function("path_substitution", 1, path_replace)
c = conn.cursor()
series_details = []
2018-01-05 04:24:11 +00:00
series_details = c.execute("SELECT title, overview, poster, fanart, hearing_impaired, tvdbid, audio_language, languages, path_substitution(path) FROM table_shows WHERE sonarrSeriesId LIKE ?", (str(no),)).fetchone()
2017-11-16 14:53:12 +00:00
tvdbid = series_details[5]
2017-09-16 00:49:46 +00:00
2017-10-16 23:27:19 +00:00
episodes = c.execute("SELECT title, path_substitution(path), season, episode, subtitles, sonarrSeriesId, missing_subtitles, sonarrEpisodeId FROM table_episodes WHERE sonarrSeriesId LIKE ? ORDER BY episode ASC", (str(no),)).fetchall()
2018-01-05 04:24:11 +00:00
number = len(episodes)
languages = c.execute("SELECT code2, name FROM table_settings_languages WHERE enabled = 1").fetchall()
c.close()
2017-09-17 13:02:16 +00:00
episodes = reversed(sorted(episodes, key=operator.itemgetter(2)))
2017-09-16 00:49:46 +00:00
seasons_list = []
for key,season in itertools.groupby(episodes,operator.itemgetter(2)):
seasons_list.append(list(season))
2018-01-05 04:24:11 +00:00
return template('episodes', __file__=__file__, bazarr_version=bazarr_version, no=no, details=series_details, languages=languages, seasons=seasons_list, url_sonarr_short=url_sonarr_short, base_url=base_url, tvdbid=tvdbid, number=number)
2017-10-16 23:27:19 +00:00
2017-11-09 21:13:14 +00:00
@route(base_url + 'scan_disk/<no:int>', method='GET')
2017-10-16 23:27:19 +00:00
def scan_disk(no):
ref = request.environ['HTTP_REFERER']
series_scan_subtitles(no)
redirect(ref)
2017-11-09 21:13:14 +00:00
@route(base_url + 'search_missing_subtitles/<no:int>', method='GET')
2017-10-16 23:27:19 +00:00
def search_missing_subtitles(no):
ref = request.environ['HTTP_REFERER']
series_download_subtitles(no)
redirect(ref)
2017-09-16 00:49:46 +00:00
2017-11-09 21:13:14 +00:00
@route(base_url + 'history')
2017-09-16 00:49:46 +00:00
def history():
2017-12-05 00:01:10 +00:00
db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
2017-09-16 00:49:46 +00:00
c = db.cursor()
2017-10-03 02:59:45 +00:00
c.execute("SELECT COUNT(*) FROM table_history")
row_count = c.fetchone()
row_count = row_count[0]
page = request.GET.page
if page == "":
page = "1"
offset = (int(page) - 1) * 15
max_page = int(math.ceil(row_count / 15.0))
2017-10-03 02:59:45 +00:00
2018-01-08 02:47:40 +00:00
now = datetime.now()
today = []
thisweek = []
thisyear = []
stats = c.execute("SELECT timestamp FROM table_history WHERE action LIKE '1'").fetchall()
total = len(stats)
for stat in stats:
if now - timedelta(hours=24) <= datetime.fromtimestamp(stat[0]) <= now:
today.append(datetime.fromtimestamp(stat[0]).date())
if now - timedelta(weeks=1) <= datetime.fromtimestamp(stat[0]) <= now:
thisweek.append(datetime.fromtimestamp(stat[0]).date())
if now - timedelta(weeks=52) <= datetime.fromtimestamp(stat[0]) <= now:
thisyear.append(datetime.fromtimestamp(stat[0]).date())
stats = [len(today), len(thisweek), len(thisyear), total]
2017-10-03 02:59:45 +00:00
c.execute("SELECT table_history.action, table_shows.title, table_episodes.season || 'x' || table_episodes.episode, table_episodes.title, table_history.timestamp, table_history.description, table_history.sonarrSeriesId FROM table_history INNER JOIN table_shows on table_shows.sonarrSeriesId = table_history.sonarrSeriesId INNER JOIN table_episodes on table_episodes.sonarrEpisodeId = table_history.sonarrEpisodeId ORDER BY id DESC LIMIT 15 OFFSET ?", (offset,))
2017-09-16 00:49:46 +00:00
data = c.fetchall()
c.close()
data = reversed(sorted(data, key=operator.itemgetter(4)))
2018-01-08 02:47:40 +00:00
return template('history', __file__=__file__, bazarr_version=bazarr_version, rows=data, row_count=row_count, page=page, max_page=max_page, stats=stats, base_url=base_url)
2017-10-03 02:59:45 +00:00
2017-11-09 21:13:14 +00:00
@route(base_url + 'wanted')
2017-10-03 02:59:45 +00:00
def wanted():
2017-12-05 00:01:10 +00:00
db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
2017-10-03 02:59:45 +00:00
db.create_function("path_substitution", 1, path_replace)
c = db.cursor()
c.execute("SELECT COUNT(*) FROM table_episodes WHERE missing_subtitles != '[]'")
missing_count = c.fetchone()
missing_count = missing_count[0]
page = request.GET.page
if page == "":
page = "1"
offset = (int(page) - 1) * 15
max_page = int(math.ceil(missing_count / 15.0))
2017-10-03 02:59:45 +00:00
2017-10-16 23:27:19 +00:00
c.execute("SELECT table_shows.title, table_episodes.season || 'x' || table_episodes.episode, table_episodes.title, table_episodes.missing_subtitles, table_episodes.sonarrSeriesId, path_substitution(table_episodes.path), table_shows.hearing_impaired, table_episodes.sonarrEpisodeId FROM table_episodes INNER JOIN table_shows on table_shows.sonarrSeriesId = table_episodes.sonarrSeriesId WHERE table_episodes.missing_subtitles != '[]' ORDER BY table_episodes._rowid_ DESC LIMIT 15 OFFSET ?", (offset,))
2017-10-03 02:59:45 +00:00
data = c.fetchall()
c.close()
return template('wanted', __file__=__file__, bazarr_version=bazarr_version, rows=data, missing_count=missing_count, page=page, max_page=max_page, base_url=base_url)
2017-09-16 00:49:46 +00:00
2017-11-09 21:13:14 +00:00
@route(base_url + 'wanted_search_missing_subtitles')
2017-10-23 21:24:22 +00:00
def wanted_search_missing_subtitles_list():
2017-10-16 23:27:19 +00:00
ref = request.environ['HTTP_REFERER']
2017-10-23 03:00:11 +00:00
wanted_search_missing_subtitles()
2017-10-16 23:27:19 +00:00
redirect(ref)
2017-11-09 21:13:14 +00:00
@route(base_url + 'settings')
2017-09-16 00:49:46 +00:00
def settings():
2017-12-05 00:01:10 +00:00
db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
2017-09-16 00:49:46 +00:00
c = db.cursor()
c.execute("SELECT * FROM table_settings_general")
settings_general = c.fetchone()
2017-09-19 10:43:14 +00:00
c.execute("SELECT * FROM table_settings_languages ORDER BY name")
2017-09-16 00:49:46 +00:00
settings_languages = c.fetchall()
2017-09-19 10:43:14 +00:00
c.execute("SELECT * FROM table_settings_providers ORDER BY name")
2017-09-16 00:49:46 +00:00
settings_providers = c.fetchall()
c.execute("SELECT * FROM table_settings_sonarr")
settings_sonarr = c.fetchone()
c.execute("SELECT * FROM table_settings_notifier")
settings_notifier = c.fetchall()
2017-09-16 00:49:46 +00:00
c.close()
return template('settings', __file__=__file__, bazarr_version=bazarr_version, settings_general=settings_general, settings_languages=settings_languages, settings_providers=settings_providers, settings_sonarr=settings_sonarr, settings_notifier=settings_notifier, base_url=base_url)
2017-09-19 10:43:14 +00:00
2017-11-09 21:13:14 +00:00
@route(base_url + 'save_settings', method='POST')
2017-09-19 10:43:14 +00:00
def save_settings():
2017-10-18 18:54:24 +00:00
ref = request.environ['HTTP_REFERER']
2017-12-05 00:01:10 +00:00
conn = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
2017-10-03 02:59:45 +00:00
c = conn.cursor()
2017-09-19 10:43:14 +00:00
2017-10-03 02:59:45 +00:00
settings_general_ip = request.forms.get('settings_general_ip')
settings_general_port = request.forms.get('settings_general_port')
settings_general_baseurl = request.forms.get('settings_general_baseurl')
2017-10-16 23:27:19 +00:00
settings_general_loglevel = request.forms.get('settings_general_loglevel')
2017-10-03 02:59:45 +00:00
settings_general_sourcepath = request.forms.getall('settings_general_sourcepath')
settings_general_destpath = request.forms.getall('settings_general_destpath')
settings_general_pathmapping = []
settings_general_pathmapping.extend([list(a) for a in zip(settings_general_sourcepath, settings_general_destpath)])
2017-10-28 01:41:24 +00:00
settings_general_branch = request.forms.get('settings_general_branch')
settings_general_automatic = request.forms.get('settings_general_automatic')
2017-10-28 02:09:03 +00:00
if settings_general_automatic is None:
settings_general_automatic = 'False'
else:
settings_general_automatic = 'True'
2018-01-10 16:44:47 +00:00
settings_general_single_language = request.forms.get('settings_general_single_language')
if settings_general_single_language is None:
settings_general_single_language = 'False'
else:
settings_general_single_language = 'True'
before = c.execute("SELECT ip, port, base_url FROM table_settings_general").fetchone()
after = (unicode(settings_general_ip), int(settings_general_port), unicode(settings_general_baseurl))
2018-01-10 16:44:47 +00:00
c.execute("UPDATE table_settings_general SET ip = ?, port = ?, base_url = ?, path_mapping = ?, log_level = ?, branch=?, auto_update=?, single_language=?", (unicode(settings_general_ip), int(settings_general_port), unicode(settings_general_baseurl), unicode(settings_general_pathmapping), unicode(settings_general_loglevel), unicode(settings_general_branch), unicode(settings_general_automatic), unicode(settings_general_single_language)))
conn.commit()
if after != before:
configured()
get_general_settings()
2017-10-03 02:59:45 +00:00
settings_sonarr_ip = request.forms.get('settings_sonarr_ip')
settings_sonarr_port = request.forms.get('settings_sonarr_port')
settings_sonarr_baseurl = request.forms.get('settings_sonarr_baseurl')
settings_sonarr_ssl = request.forms.get('settings_sonarr_ssl')
if settings_sonarr_ssl is None:
settings_sonarr_ssl = 'False'
2017-09-19 10:43:14 +00:00
else:
2017-10-03 02:59:45 +00:00
settings_sonarr_ssl = 'True'
settings_sonarr_apikey = request.forms.get('settings_sonarr_apikey')
c.execute("UPDATE table_settings_sonarr SET ip = ?, port = ?, base_url = ?, ssl = ?, apikey = ?", (settings_sonarr_ip, settings_sonarr_port, settings_sonarr_baseurl, settings_sonarr_ssl, settings_sonarr_apikey))
2017-10-03 02:59:45 +00:00
settings_subliminal_providers = request.forms.getall('settings_subliminal_providers')
c.execute("UPDATE table_settings_providers SET enabled = 0")
for item in settings_subliminal_providers:
c.execute("UPDATE table_settings_providers SET enabled = '1' WHERE name = ?", (item,))
settings_addic7ed_username = request.forms.get('settings_addic7ed_username')
settings_addic7ed_password = request.forms.get('settings_addic7ed_password')
c.execute("UPDATE table_settings_providers SET username = ?, password = ? WHERE name = 'addic7ed'", (settings_addic7ed_username, settings_addic7ed_password))
settings_legendastv_username = request.forms.get('settings_legendastv_username')
settings_legendastv_password = request.forms.get('settings_legendastv_password')
c.execute("UPDATE table_settings_providers SET username = ?, password = ? WHERE name = 'legendastv'", (settings_legendastv_username, settings_legendastv_password))
settings_opensubtitles_username = request.forms.get('settings_opensubtitles_username')
settings_opensubtitles_password = request.forms.get('settings_opensubtitles_password')
c.execute("UPDATE table_settings_providers SET username = ?, password = ? WHERE name = 'opensubtitles'", (settings_opensubtitles_username, settings_opensubtitles_password))
2017-10-03 02:59:45 +00:00
settings_subliminal_languages = request.forms.getall('settings_subliminal_languages')
c.execute("UPDATE table_settings_languages SET enabled = 0")
for item in settings_subliminal_languages:
c.execute("UPDATE table_settings_languages SET enabled = '1' WHERE code2 = ?", (item,))
settings_notifier_Boxcar_enabled = request.forms.get('settings_notifier_Boxcar_enabled')
if settings_notifier_Boxcar_enabled == 'on':
settings_notifier_Boxcar_enabled = 1
else:
settings_notifier_Boxcar_enabled = 0
settings_notifier_Boxcar_url = request.forms.get('settings_notifier_Boxcar_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'Boxcar'", (settings_notifier_Boxcar_enabled, settings_notifier_Boxcar_url))
settings_notifier_Faast_enabled = request.forms.get('settings_notifier_Faast_enabled')
if settings_notifier_Faast_enabled == 'on':
settings_notifier_Faast_enabled = 1
else:
settings_notifier_Faast_enabled = 0
settings_notifier_Faast_url = request.forms.get('settings_notifier_Faast_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'Faast'", (settings_notifier_Faast_enabled, settings_notifier_Faast_url))
settings_notifier_Growl_enabled = request.forms.get('settings_notifier_Growl_enabled')
if settings_notifier_Growl_enabled == 'on':
settings_notifier_Growl_enabled = 1
else:
settings_notifier_Growl_enabled = 0
settings_notifier_Growl_url = request.forms.get('settings_notifier_Growl_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'Growl'", (settings_notifier_Growl_enabled, settings_notifier_Growl_url))
settings_notifier_Join_enabled = request.forms.get('settings_notifier_Join_enabled')
if settings_notifier_Join_enabled == 'on':
settings_notifier_Join_enabled = 1
else:
settings_notifier_Join_enabled = 0
settings_notifier_Join_url = request.forms.get('settings_notifier_Join_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'Join'", (settings_notifier_Join_enabled, settings_notifier_Join_url))
settings_notifier_KODI_enabled = request.forms.get('settings_notifier_KODI_enabled')
if settings_notifier_KODI_enabled == 'on':
settings_notifier_KODI_enabled = 1
else:
settings_notifier_KODI_enabled = 0
settings_notifier_KODI_url = request.forms.get('settings_notifier_KODI_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'KODI'", (settings_notifier_KODI_enabled, settings_notifier_KODI_url))
settings_notifier_Mattermost_enabled = request.forms.get('settings_notifier_Mattermost_enabled')
if settings_notifier_Mattermost_enabled == 'on':
settings_notifier_Mattermost_enabled = 1
else:
settings_notifier_Mattermost_enabled = 0
settings_notifier_Mattermost_url = request.forms.get('settings_notifier_Mattermost_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'Mattermost'", (settings_notifier_Mattermost_enabled, settings_notifier_Mattermost_url))
settings_notifier_NMA_enabled = request.forms.get('settings_notifier_Notify My Android_enabled')
if settings_notifier_NMA_enabled == 'on':
settings_notifier_NMA_enabled = 1
else:
settings_notifier_NMA_enabled = 0
settings_notifier_NMA_url = request.forms.get('settings_notifier_Notify My Android_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'Notify My Android'", (settings_notifier_NMA_enabled, settings_notifier_NMA_url))
settings_notifier_Prowl_enabled = request.forms.get('settings_notifier_Prowl_enabled')
if settings_notifier_Prowl_enabled == 'on':
settings_notifier_Prowl_enabled = 1
else:
settings_notifier_Prowl_enabled = 0
settings_notifier_Prowl_url = request.forms.get('settings_notifier_Prowl_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'Prowl'", (settings_notifier_Prowl_enabled, settings_notifier_Prowl_url))
settings_notifier_Pushalot_enabled = request.forms.get('settings_notifier_Pushalot_enabled')
if settings_notifier_Pushalot_enabled == 'on':
settings_notifier_Pushalot_enabled = 1
else:
settings_notifier_Pushalot_enabled = 0
settings_notifier_Pushalot_url = request.forms.get('settings_notifier_Pushalot_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'Pushalot'", (settings_notifier_Pushalot_enabled, settings_notifier_Pushalot_url))
settings_notifier_PushBullet_enabled = request.forms.get('settings_notifier_PushBullet_enabled')
if settings_notifier_PushBullet_enabled == 'on':
settings_notifier_PushBullet_enabled = 1
else:
settings_notifier_PushBullet_enabled = 0
settings_notifier_PushBullet_url = request.forms.get('settings_notifier_PushBullet_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'PushBullet'", (settings_notifier_PushBullet_enabled, settings_notifier_PushBullet_url))
settings_notifier_Pushjet_enabled = request.forms.get('settings_notifier_Pushjet_enabled')
if settings_notifier_Pushjet_enabled == 'on':
settings_notifier_Pushjet_enabled = 1
else:
settings_notifier_Pushjet_enabled = 0
settings_notifier_Pushjet_url = request.forms.get('settings_notifier_Pushjet_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'Pushjet'", (settings_notifier_Pushjet_enabled, settings_notifier_Pushjet_url))
settings_notifier_Pushover_enabled = request.forms.get('settings_notifier_Pushover_enabled')
if settings_notifier_Pushover_enabled == 'on':
settings_notifier_Pushover_enabled = 1
else:
settings_notifier_Pushover_enabled = 0
settings_notifier_Pushover_url = request.forms.get('settings_notifier_Pushover_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'Pushover'", (settings_notifier_Pushover_enabled, settings_notifier_Pushover_url))
settings_notifier_RocketChat_enabled = request.forms.get('settings_notifier_Rocket.Chat_enabled')
if settings_notifier_RocketChat_enabled == 'on':
settings_notifier_RocketChat_enabled = 1
else:
settings_notifier_RocketChat_enabled = 0
settings_notifier_RocketChat_url = request.forms.get('settings_notifier_Rocket.Chat_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'Rocket.Chat'", (settings_notifier_RocketChat_enabled, settings_notifier_RocketChat_url))
settings_notifier_Slack_enabled = request.forms.get('settings_notifier_Slack_enabled')
if settings_notifier_Slack_enabled == 'on':
settings_notifier_Slack_enabled = 1
else:
settings_notifier_Slack_enabled = 0
settings_notifier_Slack_url = request.forms.get('settings_notifier_Slack_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'Slack'", (settings_notifier_Slack_enabled, settings_notifier_Slack_url))
settings_notifier_SuperToasty_enabled = request.forms.get('settings_notifier_Super Toasty_enabled')
if settings_notifier_SuperToasty_enabled == 'on':
settings_notifier_SuperToasty_enabled = 1
else:
settings_notifier_SuperToasty_enabled = 0
settings_notifier_SuperToasty_url = request.forms.get('settings_notifier_Super Toasty_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'Super Toasty'", (settings_notifier_SuperToasty_enabled, settings_notifier_SuperToasty_url))
settings_notifier_Telegram_enabled = request.forms.get('settings_notifier_Telegram_enabled')
if settings_notifier_Telegram_enabled == 'on':
settings_notifier_Telegram_enabled = 1
else:
settings_notifier_Telegram_enabled = 0
settings_notifier_Telegram_url = request.forms.get('settings_notifier_Telegram_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'Telegram'", (settings_notifier_Telegram_enabled, settings_notifier_Telegram_url))
settings_notifier_Twitter_enabled = request.forms.get('settings_notifier_Twitter_enabled')
if settings_notifier_Twitter_enabled == 'on':
settings_notifier_Twitter_enabled = 1
else:
settings_notifier_Twitter_enabled = 0
settings_notifier_Twitter_url = request.forms.get('settings_notifier_Twitter_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'Twitter'", (settings_notifier_Twitter_enabled, settings_notifier_Twitter_url))
settings_notifier_XBMC_enabled = request.forms.get('settings_notifier_XBMC_enabled')
if settings_notifier_XBMC_enabled == 'on':
settings_notifier_XBMC_enabled = 1
else:
settings_notifier_XBMC_enabled = 0
settings_notifier_XBMC_url = request.forms.get('settings_notifier_XBMC_url')
c.execute("UPDATE table_settings_notifier SET enabled = ?, url = ? WHERE name = 'XBMC'", (settings_notifier_XBMC_enabled, settings_notifier_XBMC_url))
2017-09-19 10:43:14 +00:00
conn.commit()
c.close()
2017-12-06 04:07:37 +00:00
2017-12-31 21:28:06 +00:00
logging.info('Settings saved succesfully.')
2017-10-03 02:59:45 +00:00
2017-10-18 18:54:24 +00:00
redirect(ref)
2017-09-16 00:49:46 +00:00
2017-11-09 21:13:14 +00:00
@route(base_url + 'check_update')
2017-11-11 05:54:19 +00:00
def check_update():
2017-11-09 21:13:14 +00:00
ref = request.environ['HTTP_REFERER']
2017-12-06 04:07:37 +00:00
check_and_apply_update()
2017-11-09 21:13:14 +00:00
redirect(ref)
@route(base_url + 'system')
2017-09-16 00:49:46 +00:00
def system():
2017-11-20 03:38:06 +00:00
def get_time_from_interval(interval):
interval_clean = interval.split('[')
interval_clean = interval_clean[1][:-1]
interval_split = interval_clean.split(':')
2017-11-20 03:49:17 +00:00
hour = interval_split[0]
minute = interval_split[1].lstrip("0")
second = interval_split[2].lstrip("0")
2017-11-20 03:38:06 +00:00
text = "every "
2017-11-20 03:49:17 +00:00
if hour != "0":
text = text + hour
if hour == "1":
2017-11-20 03:38:06 +00:00
text = text + " hour"
else:
text = text + " hours"
2017-11-20 03:49:17 +00:00
if minute != "" and second != "":
2017-11-20 03:38:06 +00:00
text = text + ", "
2017-11-20 03:49:17 +00:00
elif minute == "" and second != "":
text = text + " and "
elif minute != "" and second == "":
text = text + " and "
if minute != "":
text = text + minute
if minute == "1":
2017-11-20 03:38:06 +00:00
text = text + " minute"
else:
text = text + " minutes"
2017-11-20 03:49:17 +00:00
if second != "":
2017-11-20 03:38:06 +00:00
text = text + " and "
2017-11-20 03:49:17 +00:00
if second != "":
text = text + second
if second == "1":
2017-11-20 03:38:06 +00:00
text = text + " second"
else:
text = text + " seconds"
return text
def get_time_from_cron(cron):
text = "at "
hour = str(cron[5])
minute = str(cron[6])
second = str(cron[7])
if hour != "0" and hour != "*":
text = text + hour
if hour == "0" or hour == "1":
text = text + " hour"
else:
text = text + " hours"
if minute != "*" and second != "0":
text = text + ", "
elif minute == "*" and second != "0":
text = text + " and "
elif minute != "0" and minute != "*" and second == "0":
text = text + " and "
if minute != "0" and minute != "*":
text = text + minute
if minute == "0" or minute == "1":
text = text + " minute"
else:
text = text + " minutes"
if second != "0" and second != "*":
text = text + " and "
if second != "0" and second != "*":
text = text + second
if second == "0" or second == "1":
text = text + " second"
else:
text = text + " seconds"
return text
2017-10-23 03:00:11 +00:00
task_list = []
for job in scheduler.get_jobs():
if job.trigger.__str__().startswith('interval'):
2017-11-20 03:38:06 +00:00
task_list.append([job.name, get_time_from_interval(str(job.trigger)), pretty.date(job.next_run_time.replace(tzinfo=None)), job.id])
elif job.trigger.__str__().startswith('cron'):
2017-11-20 03:38:06 +00:00
task_list.append([job.name, get_time_from_cron(job.trigger.fields), pretty.date(job.next_run_time.replace(tzinfo=None)), job.id])
2017-11-22 04:01:26 +00:00
i = 0
2017-11-22 04:01:26 +00:00
with open(os.path.join(os.path.dirname(__file__), 'data/log/bazarr.log')) as f:
for i, l in enumerate(f, 1):
pass
row_count = i
max_page = int(math.ceil(row_count / 50.0))
2017-10-16 23:27:19 +00:00
return template('system', __file__=__file__, bazarr_version=bazarr_version, base_url=base_url, task_list=task_list, row_count=row_count, max_page=max_page)
2017-11-22 04:01:26 +00:00
@route(base_url + 'logs/<page:int>')
def get_logs(page):
page_size = 50
begin = (page * page_size) - page_size
end = (page * page_size) - 1
logs_complete = []
for line in reversed(open(os.path.join(os.path.dirname(__file__), 'data/log/bazarr.log')).readlines()):
logs_complete.append(line.rstrip())
logs = logs_complete[begin:end]
return template('logs', logs=logs, base_url=base_url)
2017-10-16 23:27:19 +00:00
@route(base_url + 'execute/<taskid>')
def execute_task(taskid):
ref = request.environ['HTTP_REFERER']
execute_now(taskid)
redirect(ref)
2017-11-09 21:13:14 +00:00
@route(base_url + 'remove_subtitles', method='POST')
2017-09-16 00:49:46 +00:00
def remove_subtitles():
2017-10-16 23:27:19 +00:00
episodePath = request.forms.get('episodePath')
language = request.forms.get('language')
subtitlesPath = request.forms.get('subtitlesPath')
sonarrSeriesId = request.forms.get('sonarrSeriesId')
sonarrEpisodeId = request.forms.get('sonarrEpisodeId')
2017-11-16 14:53:12 +00:00
tvdbid = request.forms.get('tvdbid')
2017-09-16 00:49:46 +00:00
try:
os.remove(subtitlesPath)
2017-09-17 13:02:16 +00:00
result = pycountry.languages.lookup(language).name + " subtitles deleted from disk."
history_log(0, sonarrSeriesId, sonarrEpisodeId, result)
2017-09-16 00:49:46 +00:00
except OSError:
2017-09-17 00:11:47 +00:00
pass
store_subtitles(episodePath)
2017-11-21 20:42:18 +00:00
list_missing_subtitles(sonarrSeriesId)
2017-10-16 23:27:19 +00:00
2017-11-09 21:13:14 +00:00
@route(base_url + 'get_subtitle', method='POST')
2017-09-17 00:11:47 +00:00
def get_subtitle():
2017-10-03 02:59:45 +00:00
ref = request.environ['HTTP_REFERER']
2017-10-16 23:27:19 +00:00
episodePath = request.forms.get('episodePath')
language = request.forms.get('language')
hi = request.forms.get('hi')
sonarrSeriesId = request.forms.get('sonarrSeriesId')
sonarrEpisodeId = request.forms.get('sonarrEpisodeId')
2017-11-16 14:53:12 +00:00
tvdbid = request.forms.get('tvdbid')
2017-10-03 02:59:45 +00:00
2017-12-05 00:01:10 +00:00
db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
2017-10-03 02:59:45 +00:00
c = db.cursor()
c.execute("SELECT * FROM table_settings_providers WHERE enabled = 1")
enabled_providers = c.fetchall()
2017-10-03 02:59:45 +00:00
c.close()
providers_list = []
providers_auth = {}
if len(enabled_providers) > 0:
for provider in enabled_providers:
providers_list.append(provider[0])
try:
if provider[2] is not '' and provider[3] is not '':
provider_auth = providers_auth.append(provider[0])
provider_auth.update({'username':providers[2], 'password':providers[3]})
else:
providers_auth = None
except:
providers_auth = None
else:
providers_list = None
providers_auth = None
2017-09-17 00:11:47 +00:00
2017-09-16 01:33:49 +00:00
try:
result = download_subtitle(episodePath, language, hi, providers_list, providers_auth)
2017-10-03 02:59:45 +00:00
if result is not None:
history_log(1, sonarrSeriesId, sonarrEpisodeId, result)
send_notifications(sonarrSeriesId, sonarrEpisodeId, result)
2017-10-03 02:59:45 +00:00
store_subtitles(episodePath)
2017-11-21 20:42:18 +00:00
list_missing_subtitles(sonarrSeriesId)
2017-10-03 02:59:45 +00:00
redirect(ref)
2017-09-16 01:33:49 +00:00
except OSError:
2017-11-16 14:53:12 +00:00
pass
2017-09-16 01:33:49 +00:00
def configured():
conn = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
c = conn.cursor()
c.execute("UPDATE table_settings_general SET configured = 1")
conn.commit()
c.close()
2018-01-30 20:25:49 +00:00
@route(base_url + 'api/wanted')
def api_wanted():
db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
c = db.cursor()
data = c.execute("SELECT table_shows.title, table_episodes.season || 'x' || table_episodes.episode, table_episodes.title, table_episodes.missing_subtitles FROM table_episodes INNER JOIN table_shows on table_shows.sonarrSeriesId = table_episodes.sonarrSeriesId WHERE table_episodes.missing_subtitles != '[]' ORDER BY table_episodes._rowid_ DESC").fetchall()
c.close()
return dict(subtitles=data)
@route(base_url + 'api/history')
def api_history():
db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
c = db.cursor()
data = c.execute("SELECT table_shows.title, table_episodes.season || 'x' || table_episodes.episode, table_episodes.title, strftime('%Y-%m-%d', datetime(table_history.timestamp, 'unixepoch')) FROM table_history INNER JOIN table_shows on table_shows.sonarrSeriesId = table_history.sonarrSeriesId INNER JOIN table_episodes on table_episodes.sonarrEpisodeId = table_history.sonarrEpisodeId WHERE table_history.action = '1' ORDER BY id DESC").fetchall()
c.close()
return dict(subtitles=data)
2017-12-13 17:42:09 +00:00
logging.info('Bazarr is started and waiting for request on http://' + str(ip) + ':' + str(port) + str(base_url))
2017-09-28 01:55:21 +00:00
run(host=ip, port=port, server='waitress')
2017-12-13 17:42:09 +00:00
logging.info('Bazarr has been stopped.')