"""

Tautulli application entry point. Parses arguments, setups encoding and

initializes the application.

"""

# Fixed paths to Tautulli

if hasattr(sys, 'frozen'):

plexpy.FULL_PATH = os.path.abspath(sys.executable)

else:

plexpy.FULL_PATH = os.path.abspath(__file__)

plexpy.PROG_DIR = os.path.dirname(plexpy.FULL_PATH)

plexpy.ARGS = sys.argv[1:]

# From sickbeard

plexpy.SYS_PLATFORM = sys.platform

plexpy.SYS_ENCODING = None

try:

locale.setlocale(locale.LC_ALL, "")

plexpy.SYS_LANGUAGE, plexpy.SYS_ENCODING = locale.getdefaultlocale()

except (locale.Error, IOError):

pass

# for OSes that are poorly configured I'll just force UTF-8

if not plexpy.SYS_ENCODING or plexpy.SYS_ENCODING in ('ANSI_X3.4-1968', 'US-ASCII', 'ASCII'):

plexpy.SYS_ENCODING = 'UTF-8'

# Set up and gather command line arguments

parser = argparse.ArgumentParser(

description='A Python based monitoring and tracking tool for Plex Media Server.')

parser.add_argument(

'-v', '--verbose', action='store_true', help='Increase console logging verbosity')

parser.add_argument(

'-q', '--quiet', action='store_true', help='Turn off console logging')

parser.add_argument(

'-d', '--daemon', action='store_true', help='Run as a daemon')

parser.add_argument(

'-p', '--port', type=int, help='Force Tautulli to run on a specified port')

parser.add_argument(

'--dev', action='store_true', help='Start Tautulli in the development environment')

parser.add_argument(

'--datadir', help='Specify a directory where to store your data files')

parser.add_argument(

'--config', help='Specify a config file to use')

parser.add_argument(

'--nolaunch', action='store_true', help='Prevent browser from launching on startup')

parser.add_argument(

'--pidfile', help='Create a pid file (only relevant when running as a daemon)')

parser.add_argument(

'--nofork', action='store_true', help='Start Tautulli as a service, do not fork when restarting')

args = parser.parse_args()

if args.verbose:

plexpy.VERBOSE = True

if args.quiet:

plexpy.QUIET = True

# Do an intial setup of the logger.

logger.initLogger(console=not plexpy.QUIET, log_dir=False,

verbose=plexpy.VERBOSE)

try:

plexpy.SYS_TIMEZONE = tzlocal.get_localzone()

except (pytz.UnknownTimeZoneError, LookupError, ValueError) as e:

logger.error("Could not determine system timezone: %s" % e)

plexpy.SYS_TIMEZONE = pytz.UTC

plexpy.SYS_UTC_OFFSET = datetime.datetime.now(plexpy.SYS_TIMEZONE).strftime('%z')

if os.getenv('TAUTULLI_DOCKER', False) == 'True':

plexpy.DOCKER = True

if args.dev:

plexpy.DEV = True

logger.debug(u"Tautulli is running in the dev environment.")

if args.daemon:

if sys.platform == 'win32':

sys.stderr.write(

"Daemonizing not supported under Windows, starting normally\n")

else:

plexpy.DAEMON = True

plexpy.QUIET = True

if args.nofork:

plexpy.NOFORK = True

logger.info("Tautulli is running as a service, it will not fork when restarted.")

if args.pidfile:

plexpy.PIDFILE = str(args.pidfile)

# If the pidfile already exists, plexpy may still be running, so

# exit

if os.path.exists(plexpy.PIDFILE):

try:

with open(plexpy.PIDFILE, 'r') as fp:

pid = int(fp.read())

os.kill(pid, 0)

except IOError as e:

raise SystemExit("Unable to read PID file: %s", e)

except OSError:

logger.warn("PID file '%s' already exists, but PID %d is " \

"not running. Ignoring PID file." %

(plexpy.PIDFILE, pid))

else:

# The pidfile exists and points to a live PID. plexpy may

# still be running, so exit.

raise SystemExit("PID file '%s' already exists. Exiting." %

plexpy.PIDFILE)

# The pidfile is only useful in daemon mode, make sure we can write the

# file properly

if plexpy.DAEMON:

plexpy.CREATEPID = True

try:

with open(plexpy.PIDFILE, 'w') as fp:

fp.write("pid\n")

except IOError as e:

raise SystemExit("Unable to write PID file: %s", e)

else:

logger.warn("Not running in daemon mode. PID file creation " \

"disabled.")

# Determine which data directory and config file to use

if args.datadir:

plexpy.DATA_DIR = args.datadir

else:

plexpy.DATA_DIR = plexpy.PROG_DIR

if args.config:

config_file = args.config

else:

config_file = os.path.join(plexpy.DATA_DIR, config.FILENAME)

# Try to create the DATA_DIR if it doesn't exist

if not os.path.exists(plexpy.DATA_DIR):

try:

os.makedirs(plexpy.DATA_DIR)

except OSError:

raise SystemExit(

'Could not create data directory: ' + plexpy.DATA_DIR + '. Exiting....')

# Make sure the DATA_DIR is writeable

if not os.access(plexpy.DATA_DIR, os.W_OK):

raise SystemExit(

'Cannot write to the data directory: ' + plexpy.DATA_DIR + '. Exiting...')

# Put the database in the DATA_DIR

plexpy.DB_FILE = os.path.join(plexpy.DATA_DIR, database.FILENAME)

# Move 'plexpy.db' to 'tautulli.db'

if os.path.isfile(os.path.join(plexpy.DATA_DIR, 'plexpy.db')) and \

not os.path.isfile(os.path.join(plexpy.DATA_DIR, plexpy.DB_FILE)):

try:

os.rename(os.path.join(plexpy.DATA_DIR, 'plexpy.db'), plexpy.DB_FILE)

except OSError as e:

raise SystemExit("Unable to rename plexpy.db to tautulli.db: %s", e)

if plexpy.DAEMON:

plexpy.daemonize()

# Read config and start logging

plexpy.initialize(config_file)

# Start the background threads

plexpy.start()

# Force the http port if neccessary

if args.port:

plexpy.HTTP_PORT = args.port

logger.info('Using forced web server port: %i', plexpy.HTTP_PORT)

else:

plexpy.HTTP_PORT = int(plexpy.CONFIG.HTTP_PORT)

# Check if pyOpenSSL is installed. It is required for certificate generation

# and for CherryPy.

if plexpy.CONFIG.ENABLE_HTTPS:

try:

import OpenSSL

except ImportError:

logger.warn("The pyOpenSSL module is missing. Install this " \

"module to enable HTTPS. HTTPS will be disabled.")

plexpy.CONFIG.ENABLE_HTTPS = False

# Try to start the server. Will exit here is address is already in use.

webstart.start()

# Windows system tray icon

if os.name == 'nt' and plexpy.CONFIG.WIN_SYS_TRAY:

plexpy.win_system_tray()

logger.info("Tautulli is ready!")

# Open webbrowser

if plexpy.CONFIG.LAUNCH_BROWSER and not args.nolaunch and not plexpy.DEV:

plexpy.launch_browser(plexpy.CONFIG.HTTP_HOST, plexpy.HTTP_PORT,

plexpy.HTTP_ROOT)

# Wait endlessy for a signal to happen

while True:

if not plexpy.SIGNAL:

try:

time.sleep(1)

except KeyboardInterrupt:

plexpy.SIGNAL = 'shutdown'

else:

logger.info('Received signal: %s', plexpy.SIGNAL)

if plexpy.SIGNAL == 'shutdown':

plexpy.shutdown()

elif plexpy.SIGNAL == 'restart':

plexpy.shutdown(restart=True)

elif plexpy.SIGNAL == 'checkout':

plexpy.shutdown(restart=True, checkout=True)

else:

plexpy.shutdown(restart=True, update=True)