mirror of
https://github.com/morpheus65535/bazarr.git
synced 2024-11-12 02:22:56 +08:00
221 lines
7.7 KiB
Python
221 lines
7.7 KiB
Python
|
import os
|
||
|
|
import sqlite3
|
||
|
|
import logging
|
||
|
|
|
||
|
|
from configparser import ConfigParser
|
||
|
|
from get_argv import config_dir
|
||
|
|
|
||
|
|
# Check if config_dir exist
|
||
|
|
if os.path.exists(config_dir) is False:
|
||
|
|
# Create config_dir directory tree
|
||
|
|
try:
|
||
|
|
os.mkdir(os.path.join(config_dir))
|
||
|
|
logging.debug("Created data directory")
|
||
|
|
except OSError:
|
||
|
|
logging.exception("The configuration directory doesn't exist and Bazarr cannot create it (permission issue?).")
|
||
|
|
exit(2)
|
||
|
|
|
||
|
|
if os.path.exists(os.path.join(config_dir, 'config')) is False:
|
||
|
|
os.mkdir(os.path.join(config_dir, 'config'))
|
||
|
|
logging.debug("Created config folder")
|
||
|
|
if os.path.exists(os.path.join(config_dir, 'db')) is False:
|
||
|
|
os.mkdir(os.path.join(config_dir, 'db'))
|
||
|
|
logging.debug("Created db folder")
|
||
|
|
if os.path.exists(os.path.join(config_dir, 'log')) is False:
|
||
|
|
os.mkdir(os.path.join(config_dir, 'log'))
|
||
|
|
logging.debug("Created log folder")
|
||
|
|
|
||
|
|
config_file = os.path.normpath(os.path.join(config_dir, 'config/config.ini'))
|
||
|
|
|
||
|
|
# if os.path.exists(os.path.join(config_dir, 'db/bazarr.db')) is True and os.path.exists(config_file) is False:
|
||
|
|
try:
|
||
|
|
# Open database connection
|
||
|
|
db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
|
||
|
|
c = db.cursor()
|
||
|
|
|
||
|
|
# Get general settings from database table
|
||
|
|
c.execute("SELECT * FROM table_settings_general")
|
||
|
|
general_settings = c.fetchone()
|
||
|
|
c.execute('''SELECT * FROM table_settings_auth''')
|
||
|
|
auth_settings = c.fetchone()
|
||
|
|
c.execute('''SELECT * FROM table_settings_sonarr''')
|
||
|
|
sonarr_settings = c.fetchone()
|
||
|
|
c.execute('''SELECT * FROM table_settings_radarr''')
|
||
|
|
radarr_settings = c.fetchone()
|
||
|
|
|
||
|
|
c.execute("DROP TABLE table_settings_general")
|
||
|
|
c.execute("DROP TABLE table_settings_auth")
|
||
|
|
c.execute("DROP TABLE table_settings_sonarr")
|
||
|
|
c.execute("DROP TABLE table_settings_radarr")
|
||
|
|
|
||
|
|
# Close database connection
|
||
|
|
db.close()
|
||
|
|
|
||
|
|
cfg = ConfigParser()
|
||
|
|
|
||
|
|
section = 'general'
|
||
|
|
|
||
|
|
if not cfg.has_section(section):
|
||
|
|
cfg.add_section(section)
|
||
|
|
|
||
|
|
cfg.set(section, 'ip', str(general_settings[0]))
|
||
|
|
cfg.set(section, 'port', str(general_settings[1]))
|
||
|
|
cfg.set(section, 'base_url', general_settings[2])
|
||
|
|
cfg.set(section, 'path_mappings', general_settings[3])
|
||
|
|
cfg.set(section, 'log_level', general_settings[4])
|
||
|
|
cfg.set(section, 'branch', general_settings[5])
|
||
|
|
cfg.set(section, 'auto_update', general_settings[6])
|
||
|
|
cfg.set(section, 'single_language', general_settings[9])
|
||
|
|
cfg.set(section, 'minimum_score', general_settings[10])
|
||
|
|
cfg.set(section, 'use_scenename', general_settings[11])
|
||
|
|
cfg.set(section, 'use_postprocessing', general_settings[12])
|
||
|
|
cfg.set(section, 'postprocessing_cmd', general_settings[13])
|
||
|
|
cfg.set(section, 'use_sonarr', general_settings[14])
|
||
|
|
cfg.set(section, 'use_radarr', general_settings[15])
|
||
|
|
cfg.set(section, 'path_mappings_movie', general_settings[16])
|
||
|
|
cfg.set(section, 'serie_default_enabled', general_settings[17])
|
||
|
|
cfg.set(section, 'serie_default_language', general_settings[18])
|
||
|
|
cfg.set(section, 'serie_default_hi', general_settings[19])
|
||
|
|
cfg.set(section, 'movie_default_enabled', general_settings[20])
|
||
|
|
cfg.set(section, 'movie_default_language', general_settings[21])
|
||
|
|
cfg.set(section, 'movie_default_hi', general_settings[22])
|
||
|
|
cfg.set(section, 'page_size', general_settings[23])
|
||
|
|
cfg.set(section, 'minimum_score_movie', general_settings[24])
|
||
|
|
cfg.set(section, 'use_embedded_subs', general_settings[25])
|
||
|
|
cfg.set(section, 'only_monitored', general_settings[26])
|
||
|
|
|
||
|
|
section = 'auth'
|
||
|
|
|
||
|
|
if not cfg.has_section(section):
|
||
|
|
cfg.add_section(section)
|
||
|
|
|
||
|
|
cfg.set(section, 'enabled', auth_settings[0])
|
||
|
|
cfg.set(section, 'username', auth_settings[1])
|
||
|
|
cfg.set(section, 'password', auth_settings[2])
|
||
|
|
|
||
|
|
section = 'sonarr'
|
||
|
|
|
||
|
|
if not cfg.has_section(section):
|
||
|
|
cfg.add_section(section)
|
||
|
|
|
||
|
|
cfg.set(section, 'ip', sonarr_settings[0])
|
||
|
|
cfg.set(section, 'port', str(sonarr_settings[1]))
|
||
|
|
cfg.set(section, 'base_url', sonarr_settings[2])
|
||
|
|
cfg.set(section, 'ssl', sonarr_settings[3])
|
||
|
|
cfg.set(section, 'apikey', sonarr_settings[4])
|
||
|
|
cfg.set(section, 'full_update', sonarr_settings[5])
|
||
|
|
|
||
|
|
section = 'radarr'
|
||
|
|
|
||
|
|
if not cfg.has_section(section):
|
||
|
|
cfg.add_section(section)
|
||
|
|
|
||
|
|
cfg.set(section, 'ip', radarr_settings[0])
|
||
|
|
cfg.set(section, 'port', str(radarr_settings[1]))
|
||
|
|
cfg.set(section, 'base_url', radarr_settings[2])
|
||
|
|
cfg.set(section, 'ssl', radarr_settings[3])
|
||
|
|
cfg.set(section, 'apikey', radarr_settings[4])
|
||
|
|
cfg.set(section, 'full_update', radarr_settings[5])
|
||
|
|
|
||
|
|
with open(config_file, 'w+') as configfile:
|
||
|
|
cfg.write(configfile)
|
||
|
|
|
||
|
|
|
||
|
|
logging.info('Config file succesfully migrated from database')
|
||
|
|
|
||
|
|
except sqlite3.OperationalError:
|
||
|
|
# if not os.path.exists(config_file):
|
||
|
|
if os.path.exists(config_file) is False:
|
||
|
|
cfg = ConfigParser()
|
||
|
|
|
||
|
|
section = 'general'
|
||
|
|
|
||
|
|
if not cfg.has_section(section):
|
||
|
|
cfg.add_section(section)
|
||
|
|
|
||
|
|
cfg.set(section, 'ip', "0.0.0.0")
|
||
|
|
cfg.set(section, 'port', "6767")
|
||
|
|
cfg.set(section, 'base_url', "/")
|
||
|
|
cfg.set(section, 'path_mappings', '[]')
|
||
|
|
cfg.set(section, 'log_level', "INFO")
|
||
|
|
cfg.set(section, 'branch', "master")
|
||
|
|
cfg.set(section, 'auto_update', "True")
|
||
|
|
cfg.set(section, 'single_language', "False")
|
||
|
|
cfg.set(section, 'minimum_score', "100")
|
||
|
|
cfg.set(section, 'use_scenename', "False")
|
||
|
|
cfg.set(section, 'use_postprocessing', "False")
|
||
|
|
cfg.set(section, 'postprocessing_cmd', "False")
|
||
|
|
cfg.set(section, 'use_sonarr', "False")
|
||
|
|
cfg.set(section, 'use_radarr', "False")
|
||
|
|
cfg.set(section, 'path_mappings_movie', '[]')
|
||
|
|
cfg.set(section, 'serie_default_enabled', "False")
|
||
|
|
cfg.set(section, 'serie_default_language', '')
|
||
|
|
cfg.set(section, 'serie_default_hi', "False")
|
||
|
|
cfg.set(section, 'movie_default_enabled', "False")
|
||
|
|
cfg.set(section, 'movie_default_language', '')
|
||
|
|
cfg.set(section, 'movie_default_hi', "False")
|
||
|
|
cfg.set(section, 'page_size', "25")
|
||
|
|
cfg.set(section, 'minimum_score_movie', "100")
|
||
|
|
cfg.set(section, 'use_embedded_subs', "False")
|
||
|
|
cfg.set(section, 'only_monitored', "False")
|
||
|
|
|
||
|
|
section = 'auth'
|
||
|
|
|
||
|
|
if not cfg.has_section(section):
|
||
|
|
cfg.add_section(section)
|
||
|
|
|
||
|
|
cfg.set(section, 'enabled', "False")
|
||
|
|
cfg.set(section, 'username', "")
|
||
|
|
cfg.set(section, 'password', "")
|
||
|
|
|
||
|
|
section = 'sonarr'
|
||
|
|
|
||
|
|
if not cfg.has_section(section):
|
||
|
|
cfg.add_section(section)
|
||
|
|
|
||
|
|
cfg.set(section, 'ip', "127.0.0.1")
|
||
|
|
cfg.set(section, 'port', "8989")
|
||
|
|
cfg.set(section, 'base_url', "/")
|
||
|
|
cfg.set(section, 'ssl', "False")
|
||
|
|
cfg.set(section, 'apikey', "")
|
||
|
|
cfg.set(section, 'full_update', "Daily")
|
||
|
|
|
||
|
|
section = 'radarr'
|
||
|
|
|
||
|
|
if not cfg.has_section(section):
|
||
|
|
cfg.add_section(section)
|
||
|
|
|
||
|
|
cfg.set(section, 'ip', "127.0.0.1")
|
||
|
|
cfg.set(section, 'port', "7878")
|
||
|
|
cfg.set(section, 'base_url', "/")
|
||
|
|
cfg.set(section, 'ssl', "False")
|
||
|
|
cfg.set(section, 'apikey', "")
|
||
|
|
cfg.set(section, 'full_update', "Daily")
|
||
|
|
|
||
|
|
|
||
|
|
with open(config_file, 'w+') as configfile:
|
||
|
|
cfg.write(configfile)
|
||
|
|
|
||
|
|
logging.info('Config file created successfully')
|
||
|
|
try:
|
||
|
|
# Get SQL script from file
|
||
|
|
fd = open(os.path.join(os.path.dirname(__file__), 'create_db.sql'), 'r')
|
||
|
|
script = fd.read()
|
||
|
|
|
||
|
|
# Close SQL script file
|
||
|
|
fd.close()
|
||
|
|
|
||
|
|
# Open database connection
|
||
|
|
db = sqlite3.connect(os.path.join(config_dir, 'db/bazarr.db'), timeout=30)
|
||
|
|
c = db.cursor()
|
||
|
|
|
||
|
|
# Execute script and commit change to database
|
||
|
|
c.executescript(script)
|
||
|
|
|
||
|
|
# Close database connection
|
||
|
|
db.close()
|
||
|
|
|
||
|
|
logging.info('Database created successfully')
|
||
|
|
except:
|
||
|
|
pass
|