|
|
|
@ -2,10 +2,10 @@
|
|
|
|
|
|
|
|
|
|
import apprise
|
|
|
|
|
import os
|
|
|
|
|
import sqlite3
|
|
|
|
|
import logging
|
|
|
|
|
|
|
|
|
|
from get_args import args
|
|
|
|
|
from database import TableSettingsNotifier, TableShows, TableEpisodes, TableMovies
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def update_notifier():
|
|
|
|
@ -18,64 +18,75 @@ def update_notifier():
|
|
|
|
|
notifiers_new = []
|
|
|
|
|
notifiers_old = []
|
|
|
|
|
|
|
|
|
|
conn_db = sqlite3.connect(os.path.join(args.config_dir, 'db', 'bazarr.db'), timeout=30)
|
|
|
|
|
c_db = conn_db.cursor()
|
|
|
|
|
notifiers_current = c_db.execute('SELECT name FROM table_settings_notifier').fetchall()
|
|
|
|
|
notifiers_current = TableSettingsNotifier.select(
|
|
|
|
|
TableSettingsNotifier.name
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
for x in results['schemas']:
|
|
|
|
|
if x['service_name'] not in str(notifiers_current):
|
|
|
|
|
notifiers_new.append(x['service_name'])
|
|
|
|
|
logging.debug('Adding new notifier agent: ' + x['service_name'])
|
|
|
|
|
else:
|
|
|
|
|
notifiers_old.append(x['service_name'])
|
|
|
|
|
notifier_current = [i[0] for i in notifiers_current]
|
|
|
|
|
notifier_current = [i.name for i in notifiers_current]
|
|
|
|
|
|
|
|
|
|
notifiers_to_delete = list(set(notifier_current) - set(notifiers_old))
|
|
|
|
|
|
|
|
|
|
for notifier_new in notifiers_new:
|
|
|
|
|
c_db.execute('INSERT INTO `table_settings_notifier` (name, enabled) VALUES (?, ?);', (notifier_new, '0'))
|
|
|
|
|
TableSettingsNotifier.insert(
|
|
|
|
|
{
|
|
|
|
|
TableSettingsNotifier.name: notifier_new,
|
|
|
|
|
TableSettingsNotifier.enabled: 0
|
|
|
|
|
}
|
|
|
|
|
).execute()
|
|
|
|
|
|
|
|
|
|
for notifier_to_delete in notifiers_to_delete:
|
|
|
|
|
c_db.execute('DELETE FROM `table_settings_notifier` WHERE name=?', (notifier_to_delete,))
|
|
|
|
|
|
|
|
|
|
conn_db.commit()
|
|
|
|
|
c_db.close()
|
|
|
|
|
TableSettingsNotifier.delete().where(
|
|
|
|
|
TableSettingsNotifier.Name == notifier_to_delete
|
|
|
|
|
).execute()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def get_notifier_providers():
|
|
|
|
|
conn_db = sqlite3.connect(os.path.join(args.config_dir, 'db', 'bazarr.db'), timeout=30)
|
|
|
|
|
c_db = conn_db.cursor()
|
|
|
|
|
providers = c_db.execute('SELECT name, url FROM table_settings_notifier WHERE enabled = 1').fetchall()
|
|
|
|
|
c_db.close()
|
|
|
|
|
providers = TableSettingsNotifier.select(
|
|
|
|
|
TableSettingsNotifier.name,
|
|
|
|
|
TableSettingsNotifier.url
|
|
|
|
|
).where(
|
|
|
|
|
TableSettingsNotifier.enabled == 1
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
return providers
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def get_series_name(sonarrSeriesId):
|
|
|
|
|
conn_db = sqlite3.connect(os.path.join(args.config_dir, 'db', 'bazarr.db'), timeout=30)
|
|
|
|
|
c_db = conn_db.cursor()
|
|
|
|
|
data = c_db.execute('SELECT title FROM table_shows WHERE sonarrSeriesId = ?', (sonarrSeriesId,)).fetchone()
|
|
|
|
|
c_db.close()
|
|
|
|
|
data = TableShows.select(
|
|
|
|
|
TableShows.title
|
|
|
|
|
).where(
|
|
|
|
|
TableShows.sonarr_series_id == sonarrSeriesId
|
|
|
|
|
).first()
|
|
|
|
|
|
|
|
|
|
return data[0]
|
|
|
|
|
return data.title
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def get_episode_name(sonarrEpisodeId):
|
|
|
|
|
conn_db = sqlite3.connect(os.path.join(args.config_dir, 'db', 'bazarr.db'), timeout=30)
|
|
|
|
|
c_db = conn_db.cursor()
|
|
|
|
|
data = c_db.execute('SELECT title, season, episode FROM table_episodes WHERE sonarrEpisodeId = ?',
|
|
|
|
|
(sonarrEpisodeId,)).fetchone()
|
|
|
|
|
c_db.close()
|
|
|
|
|
data = TableEpisodes.select(
|
|
|
|
|
TableEpisodes.title,
|
|
|
|
|
TableEpisodes.season,
|
|
|
|
|
TableEpisodes.episode
|
|
|
|
|
).where(
|
|
|
|
|
TableEpisodes.sonarr_episode_id == sonarrEpisodeId
|
|
|
|
|
).first()
|
|
|
|
|
|
|
|
|
|
return data[0], data[1], data[2]
|
|
|
|
|
return data.title, data.season, data.episode
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def get_movies_name(radarrId):
|
|
|
|
|
conn_db = sqlite3.connect(os.path.join(args.config_dir, 'db', 'bazarr.db'), timeout=30)
|
|
|
|
|
c_db = conn_db.cursor()
|
|
|
|
|
data = c_db.execute('SELECT title FROM table_movies WHERE radarrId = ?', (radarrId,)).fetchone()
|
|
|
|
|
c_db.close()
|
|
|
|
|
data = TableMovies.select(
|
|
|
|
|
TableMovies.title
|
|
|
|
|
).where(
|
|
|
|
|
TableMovies.radarr_id == radarrId
|
|
|
|
|
).first()
|
|
|
|
|
|
|
|
|
|
return data[0]
|
|
|
|
|
return data.title
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def send_notifications(sonarrSeriesId, sonarrEpisodeId, message):
|
|
|
|
|