Louis Vézina
5 years ago
parent
b1a0e5e0ea
commit
98e3ff14a7
@ -1,224 +0,0 @@
|
||||
import os
|
||||
import atexit
|
||||
|
||||
from get_args import args
|
||||
from peewee import *
|
||||
from playhouse.sqliteq import SqliteQueueDatabase
|
||||
from playhouse.migrate import *
|
||||
|
||||
from helper import path_replace, path_replace_movie, path_replace_reverse, path_replace_reverse_movie
|
||||
|
||||
database = SqliteQueueDatabase(
|
||||
None,
|
||||
use_gevent=False,
|
||||
autostart=False,
|
||||
queue_max_size=256, # Max. # of pending writes that can accumulate.
|
||||
results_timeout=30.0 # Max. time to wait for query to be executed.
|
||||
)
|
||||
|
||||
|
||||
@database.func('path_substitution')
|
||||
def path_substitution(path):
|
||||
return path_replace(path)
|
||||
|
||||
|
||||
@database.func('path_substitution_movie')
|
||||
def path_substitution_movie(path):
|
||||
return path_replace_movie(path)
|
||||
|
||||
|
||||
class UnknownField(object):
|
||||
def __init__(self, *_, **__): pass
|
||||
|
||||
class BaseModel(Model):
|
||||
class Meta:
|
||||
database = database
|
||||
|
||||
|
||||
class System(BaseModel):
|
||||
configured = TextField(null=True)
|
||||
updated = TextField(null=True)
|
||||
|
||||
class Meta:
|
||||
table_name = 'system'
|
||||
primary_key = False
|
||||
|
||||
|
||||
class TableShows(BaseModel):
|
||||
alternate_titles = TextField(column_name='alternateTitles', null=True)
|
||||
audio_language = TextField(null=True)
|
||||
fanart = TextField(null=True)
|
||||
forced = TextField(null=True, constraints=[SQL('DEFAULT "False"')])
|
||||
hearing_impaired = TextField(null=True)
|
||||
languages = TextField(null=True)
|
||||
overview = TextField(null=True)
|
||||
path = TextField(null=False, unique=True)
|
||||
poster = TextField(null=True)
|
||||
sonarr_series_id = IntegerField(column_name='sonarrSeriesId', null=True, unique=True)
|
||||
sort_title = TextField(column_name='sortTitle', null=True)
|
||||
title = TextField(null=True)
|
||||
tvdb_id = IntegerField(column_name='tvdbId', null=True, unique=True, primary_key=True)
|
||||
year = TextField(null=True)
|
||||
|
||||
class Meta:
|
||||
table_name = 'table_shows'
|
||||
|
||||
|
||||
class TableEpisodes(BaseModel):
|
||||
rowid = IntegerField()
|
||||
audio_codec = TextField(null=True)
|
||||
episode = IntegerField(null=False)
|
||||
failed_attempts = TextField(column_name='failedAttempts', null=True)
|
||||
format = TextField(null=True)
|
||||
missing_subtitles = TextField(null=True)
|
||||
monitored = TextField(null=True)
|
||||
path = TextField(null=False)
|
||||
resolution = TextField(null=True)
|
||||
scene_name = TextField(null=True)
|
||||
season = IntegerField(null=False)
|
||||
sonarr_episode_id = IntegerField(column_name='sonarrEpisodeId', unique=True, null=False)
|
||||
sonarr_series_id = ForeignKeyField(TableShows, field='sonarr_series_id', column_name='sonarrSeriesId', null=False)
|
||||
subtitles = TextField(null=True)
|
||||
title = TextField(null=True)
|
||||
video_codec = TextField(null=True)
|
||||
episode_file_id = IntegerField(null=True)
|
||||
|
||||
class Meta:
|
||||
table_name = 'table_episodes'
|
||||
primary_key = False
|
||||
|
||||
|
||||
class TableMovies(BaseModel):
|
||||
rowid = IntegerField()
|
||||
alternative_titles = TextField(column_name='alternativeTitles', null=True)
|
||||
audio_codec = TextField(null=True)
|
||||
audio_language = TextField(null=True)
|
||||
failed_attempts = TextField(column_name='failedAttempts', null=True)
|
||||
fanart = TextField(null=True)
|
||||
forced = TextField(null=True, constraints=[SQL('DEFAULT "False"')])
|
||||
format = TextField(null=True)
|
||||
hearing_impaired = TextField(null=True)
|
||||
imdb_id = TextField(column_name='imdbId', null=True)
|
||||
languages = TextField(null=True)
|
||||
missing_subtitles = TextField(null=True)
|
||||
monitored = TextField(null=True)
|
||||
overview = TextField(null=True)
|
||||
path = TextField(unique=True)
|
||||
poster = TextField(null=True)
|
||||
radarr_id = IntegerField(column_name='radarrId', null=False, unique=True)
|
||||
resolution = TextField(null=True)
|
||||
scene_name = TextField(column_name='sceneName', null=True)
|
||||
sort_title = TextField(column_name='sortTitle', null=True)
|
||||
subtitles = TextField(null=True)
|
||||
title = TextField(null=False)
|
||||
tmdb_id = TextField(column_name='tmdbId', primary_key=True, null=False)
|
||||
video_codec = TextField(null=True)
|
||||
year = TextField(null=True)
|
||||
movie_file_id = IntegerField(null=True)
|
||||
|
||||
class Meta:
|
||||
table_name = 'table_movies'
|
||||
|
||||
|
||||
class TableHistory(BaseModel):
|
||||
id = PrimaryKeyField(null=False)
|
||||
action = IntegerField(null=False)
|
||||
description = TextField(null=False)
|
||||
language = TextField(null=True)
|
||||
provider = TextField(null=True)
|
||||
score = TextField(null=True)
|
||||
sonarr_episode_id = ForeignKeyField(TableEpisodes, field='sonarr_episode_id', column_name='sonarrEpisodeId', null=False)
|
||||
sonarr_series_id = ForeignKeyField(TableShows, field='sonarr_series_id', column_name='sonarrSeriesId', null=False)
|
||||
timestamp = IntegerField(null=False)
|
||||
video_path = TextField(null=True)
|
||||
|
||||
class Meta:
|
||||
table_name = 'table_history'
|
||||
|
||||
|
||||
class TableHistoryMovie(BaseModel):
|
||||
id = PrimaryKeyField(null=False)
|
||||
action = IntegerField(null=False)
|
||||
description = TextField(null=False)
|
||||
language = TextField(null=True)
|
||||
provider = TextField(null=True)
|
||||
radarr_id = ForeignKeyField(TableMovies, field='radarr_id', column_name='radarrId', null=False)
|
||||
score = TextField(null=True)
|
||||
timestamp = IntegerField(null=False)
|
||||
video_path = TextField(null=True)
|
||||
|
||||
class Meta:
|
||||
table_name = 'table_history_movie'
|
||||
|
||||
|
||||
class TableSettingsLanguages(BaseModel):
|
||||
code2 = TextField(null=False)
|
||||
code3 = TextField(null=False, unique=True, primary_key=True)
|
||||
code3b = TextField(null=True)
|
||||
enabled = IntegerField(null=True)
|
||||
name = TextField(null=False)
|
||||
|
||||
class Meta:
|
||||
table_name = 'table_settings_languages'
|
||||
|
||||
|
||||
class TableSettingsNotifier(BaseModel):
|
||||
enabled = IntegerField(null=False)
|
||||
name = TextField(null=False, primary_key=True)
|
||||
url = TextField(null=True)
|
||||
|
||||
class Meta:
|
||||
table_name = 'table_settings_notifier'
|
||||
|
||||
|
||||
def database_init():
|
||||
database.init(os.path.join(args.config_dir, 'db', 'bazarr.db'))
|
||||
database.start()
|
||||
database.connect()
|
||||
|
||||
database.pragma('wal_checkpoint', 'TRUNCATE') # Run a checkpoint and merge remaining wal-journal.
|
||||
database.cache_size = -1024 # Number of KB of cache for wal-journal.
|
||||
# Must be negative because positive means number of pages.
|
||||
database.wal_autocheckpoint = 50 # Run an automatic checkpoint every 50 write transactions.
|
||||
|
||||
# Database tables creation if they don't exists
|
||||
models_list = [TableShows, TableEpisodes, TableMovies, TableHistory, TableHistoryMovie, TableSettingsLanguages,
|
||||
TableSettingsNotifier, System]
|
||||
|
||||
database.create_tables(models_list, safe=True)
|
||||
|
||||
# Upgrade DB schema
|
||||
database_upgrade()
|
||||
|
||||
|
||||
def database_upgrade():
|
||||
# Database migration
|
||||
migrator = SqliteMigrator(database)
|
||||
|
||||
# TableShows migration
|
||||
table_shows_columns = []
|
||||
for column in database.get_columns('table_shows'):
|
||||
table_shows_columns.append(column.name)
|
||||
if 'forced' not in table_shows_columns:
|
||||
migrate(migrator.add_column('table_shows', 'forced', TableShows.forced))
|
||||
|
||||
# TableEpisodes migration
|
||||
table_episodes_columns = []
|
||||
for column in database.get_columns('table_episodes'):
|
||||
table_episodes_columns.append(column.name)
|
||||
if 'episode_file_id' not in table_episodes_columns:
|
||||
migrate(migrator.add_column('table_episodes', 'episode_file_id', TableEpisodes.episode_file_id))
|
||||
|
||||
# TableMovies migration
|
||||
table_movies_columns = []
|
||||
for column in database.get_columns('table_movies'):
|
||||
table_movies_columns.append(column.name)
|
||||
if 'forced' not in table_movies_columns:
|
||||
migrate(migrator.add_column('table_movies', 'forced', TableMovies.forced))
|
||||
if 'movie_file_id' not in table_movies_columns:
|
||||
migrate(migrator.add_column('table_movies', 'movie_file_id', TableMovies.movie_file_id))
|
||||
|
||||
|
||||
def wal_cleaning():
|
||||
database.pragma('wal_checkpoint', 'TRUNCATE') # Run a checkpoint and merge remaining wal-journal.
|
||||
database.wal_autocheckpoint = 50 # Run an automatic checkpoint every 50 write transactions.
|
||||
Loading…
Reference in new issue