bazarr/bazarr/init.py

# coding=utf-8

import os
import io
import rarfile
import json
import hashlib
import sys
import subprocess

from config import settings, configure_captcha_func
from get_args import args
from logger import configure_logging
from helper import path_mappings

from dogpile.cache.region import register_backend as register_cache_backend
import subliminal
import datetime

# set subliminal_patch user agent
os.environ["SZ_USER_AGENT"] = "Bazarr/1"

# set anti-captcha provider and key
configure_captcha_func()

# Check if args.config_dir exist
if not os.path.exists(args.config_dir):
    # Create config_dir directory tree
    try:
        os.mkdir(os.path.join(args.config_dir))
    except OSError:
        print("BAZARR The configuration directory doesn't exist and Bazarr cannot create it (permission issue?).")
        exit(2)

if not os.path.exists(os.path.join(args.config_dir, 'config')):
    os.mkdir(os.path.join(args.config_dir, 'config'))
if not os.path.exists(os.path.join(args.config_dir, 'db')):
    os.mkdir(os.path.join(args.config_dir, 'db'))
if not os.path.exists(os.path.join(args.config_dir, 'log')):
    os.mkdir(os.path.join(args.config_dir, 'log'))
if not os.path.exists(os.path.join(args.config_dir, 'cache')):
    os.mkdir(os.path.join(args.config_dir, 'cache'))

configure_logging(settings.general.getboolean('debug') or args.debug)
import logging

# deploy requirements.txt
if not args.no_update:
    try:
        import lxml, numpy, webrtcvad
    except ImportError:
        try:
            import pip
        except ImportError:
            logging.info('BAZARR unable to install requirements (pip not installed).')
        else:
            logging.info('BAZARR installing requirements...')
            subprocess.call([sys.executable, '-m', 'pip', 'install', '--user', '-r',
                             os.path.join(os.path.dirname(__file__), '..', 'requirements.txt')],
                            stdout=subprocess.PIPE, stderr=subprocess.PIPE)
            logging.info('BAZARR requirements installed.')
            try:
                restart_file = io.open(os.path.join(args.config_dir, "bazarr.restart"), "w", encoding='UTF-8')
            except Exception as e:
                logging.error('BAZARR Cannot create bazarr.restart file: ' + repr(e))
            else:
                logging.info('Bazarr is being restarted...')
                restart_file.write(str(''))
                restart_file.close()
                os._exit(0)

# create random api_key if there's none in config.ini
if not settings.auth.apikey or settings.auth.apikey.startswith("b'"):
    from binascii import hexlify
    settings.auth.apikey = hexlify(os.urandom(16)).decode()
    with open(os.path.join(args.config_dir, 'config', 'config.ini'), 'w+') as handle:
        settings.write(handle)

# create random Flask secret_key if there's none in config.ini
if not settings.general.flask_secret_key:
    from binascii import hexlify
    settings.general.flask_secret_key = hexlify(os.urandom(16)).decode()
    with open(os.path.join(args.config_dir, 'config', 'config.ini'), 'w+') as handle:
        settings.write(handle)

# change default base_url to ''
if settings.general.base_url == '/':
    settings.general.base_url = ''
    with open(os.path.join(args.config_dir, 'config', 'config.ini'), 'w+') as handle:
        settings.write(handle)

# create database file
if not os.path.exists(os.path.join(args.config_dir, 'db', 'bazarr.db')):
    import sqlite3
    # 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(args.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('BAZARR Database created successfully')

# upgrade database schema
from database import db_upgrade
db_upgrade()

# Configure dogpile file caching for Subliminal request
register_cache_backend("subzero.cache.file", "subzero.cache_backends.file", "SZFileBackend")
subliminal.region.configure('subzero.cache.file', expiration_time=datetime.timedelta(days=30),
                            arguments={'appname': "sz_cache", 'app_cache_dir': args.config_dir})
subliminal.region.backend.sync()

if not os.path.exists(os.path.join(args.config_dir, 'config', 'releases.txt')):
    from check_update import check_releases
    check_releases()
    logging.debug("BAZARR Created releases file")

config_file = os.path.normpath(os.path.join(args.config_dir, 'config', 'config.ini'))

# Reset form login password for Bazarr after migration from 0.8.x to 0.9. Password will be equal to username.
if settings.auth.type == 'form' and \
        os.path.exists(os.path.normpath(os.path.join(args.config_dir, 'config', 'users.json'))):
    username = False
    with open(os.path.normpath(os.path.join(args.config_dir, 'config', 'users.json'))) as json_file:
        try:
            data = json.load(json_file)
            username = next(iter(data))
        except:
            logging.error('BAZARR is unable to migrate credentials. You should disable login by modifying config.ini '
                          'file and settings [auth]-->type = None')
    if username:
        settings.auth.username = username
        settings.auth.password = hashlib.md5(username.encode('utf-8')).hexdigest()
        with open(os.path.join(args.config_dir, 'config', 'config.ini'), 'w+') as handle:
            settings.write(handle)
        os.remove(os.path.normpath(os.path.join(args.config_dir, 'config', 'users.json')))
        os.remove(os.path.normpath(os.path.join(args.config_dir, 'config', 'roles.json')))
        os.remove(os.path.normpath(os.path.join(args.config_dir, 'config', 'register.json')))
        logging.info('BAZARR your login credentials have been migrated successfully and your password is now equal '
                     'to your username. Please change it as soon as possible in settings.')
else:
    if os.path.exists(os.path.normpath(os.path.join(args.config_dir, 'config', 'users.json'))):
        try:
            os.remove(os.path.normpath(os.path.join(args.config_dir, 'config', 'users.json')))
            os.remove(os.path.normpath(os.path.join(args.config_dir, 'config', 'roles.json')))
            os.remove(os.path.normpath(os.path.join(args.config_dir, 'config', 'register.json')))
        except:
            logging.error("BAZARR cannot delete those file. Please do it manually: users.json, roles.json, "
                          "register.json")


def init_binaries():
    from utils import get_binary
    exe = get_binary("unrar")
    
    rarfile.UNRAR_TOOL = exe
    rarfile.ORIG_UNRAR_TOOL = exe
    try:
        rarfile.custom_check([rarfile.UNRAR_TOOL], True)
    except:
        logging.debug("custom check failed for: %s", exe)
    
    rarfile.OPEN_ARGS = rarfile.ORIG_OPEN_ARGS
    rarfile.EXTRACT_ARGS = rarfile.ORIG_EXTRACT_ARGS
    rarfile.TEST_ARGS = rarfile.ORIG_TEST_ARGS
    logging.debug("Using UnRAR from: %s", exe)
    unrar = exe
    
    return unrar


init_binaries()
path_mappings.update()