sima/launch.py

   1 # -*- coding: utf-8 -*-
   2 # Copyright (c) 2013, 2014 Jack Kaliko <kaliko@azylum.org>
   3 #
   4 #  This file is part of sima
   5 #
   6 #  sima is free software: you can redistribute it and/or modify
   7 #  it under the terms of the GNU General Public License as published by
   8 #  the Free Software Foundation, either version 3 of the License, or
   9 #  (at your option) any later version.
  10 #
  11 #  sima is distributed in the hope that it will be useful,
  12 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 #  GNU General Public License for more details.
  15 #
  16 #  You should have received a copy of the GNU General Public License
  17 #  along with sima.  If not, see <http://www.gnu.org/licenses/>.
  18 #
  19 #
  20 """Sima
  21 """
  22
  23 # standard library import
  24 import logging
  25 import sys
  26
  27 from importlib import __import__
  28 from os.path import isfile
  29 ##
  30
  31 # third parties components
  32 ##
  33
  34 # local import
  35 from . import core, info
  36 from .lib.logger import set_logger
  37 from .lib.simadb import SimaDB
  38 from .utils.config import ConfMan
  39 from .utils.startopt import StartOpt
  40 from .utils.utils import exception_log, SigHup
  41  # core plugins
  42 from .plugins.core.history import History
  43 from .plugins.core.mpdoptions import MpdOptions
  44 ##
  45
  46
  47 def load_plugins(sima, source):
  48     """Handles internal/external plugins
  49         sima:   sima.core.Sima instance
  50         source: ['internal', 'contrib']
  51     """
  52     if not sima.config.get('sima', source ):
  53         return
  54     logger = logging.getLogger('sima')
  55     for plugin in sima.config.get('sima', source).split(','):
  56         plugin = plugin.strip(' \n')
  57         module = 'sima.plugins.{0}.{1}'.format(source, plugin.lower())
  58         try:
  59             mod_obj = __import__(module, fromlist=[plugin])
  60         except ImportError as err:
  61             logger.error('Failed to load plugin\'s module: ' +
  62                          '{0} ({1})'.format(module, err))
  63             sima.shutdown()
  64             sys.exit(1)
  65         try:
  66             plugin_obj = getattr(mod_obj, plugin)
  67         except AttributeError as err:
  68             logger.error('Failed to load plugin {0} ({1})'.format(plugin, err))
  69             sima.shutdown()
  70             sys.exit(1)
  71         logger.info('Loading {0} plugin: {name} ({doc})'.format(
  72                              source, **plugin_obj.info()))
  73         sima.register_plugin(plugin_obj)
  74
  75
  76 def start(sopt, restart=False):
  77     """starts application
  78     """
  79     # set logger
  80     verbosity = sopt.options.get('verbosity', 'warning')
  81     logfile = sopt.options.get('logfile', None)
  82     cli_loglevel = getattr(logging, verbosity.upper())
  83     set_logger(level=verbosity, logfile=logfile)
  84     logger = logging.getLogger('sima')
  85     logger.setLevel(cli_loglevel)
  86     # loads configuration
  87     config = ConfMan(logger, sopt.options).config
  88     logger.setLevel(getattr(logging,
  89                     config.get('log', 'verbosity').upper()))  # pylint: disable=E1103
  90
  91     logger.debug('Command line say: {0}'.format(sopt.options))
  92     # Create Database
  93     db_file = config.get('sima', 'db_file')
  94     if (sopt.options.get('create_db', None)
  95        or not isfile(db_file)):
  96         logger.info('Creating database in "{}"'.format(db_file))
  97         open(db_file, 'a').close()
  98         SimaDB(db_path=db_file).create_db()
  99         if sopt.options.get('create_db', None):
 100             logger.info('Done, bye...')
 101             sys.exit(0)
 102
 103     logger.info('Starting...')
 104     sima = core.Sima(config)
 105
 106     # required core plugins
 107     sima.register_plugin(History)
 108     sima.register_plugin(MpdOptions)
 109
 110     #  Loading internal plugins
 111     load_plugins(sima, 'internal')
 112
 113     #  Loading contrib plugins
 114     load_plugins(sima, 'contrib')
 115     # Run as a daemon
 116     if config.getboolean('daemon', 'daemon'):
 117         if restart:
 118             sima.run()
 119         else:
 120             logger.info('Daemonize process...')
 121             sima.start()
 122
 123     try:
 124         sima.foreground()
 125     except KeyboardInterrupt:
 126         logger.info('Caught KeyboardInterrupt, stopping')
 127         sys.exit(0)
 128
 129
 130 def run(sopt, restart=False):
 131     """
 132     Handles SigHup exception
 133     Catches Unhandled exception
 134     """
 135     # pylint: disable=broad-except
 136     try:
 137         start(sopt, restart)
 138     except SigHup:  # SigHup inherit from Exception
 139         run(sopt, True)
 140     except Exception:  # Unhandled exception
 141         exception_log()
 142
 143 # Script starts here
 144 def main():
 145     """Entry point"""
 146     nfo = dict({'version': info.__version__,
 147                  'prog': 'sima'})
 148     # StartOpt gathers options from command line call (in StartOpt().options)
 149     sopt = StartOpt(nfo)
 150     run(sopt)
 151
 152
 153 if __name__ == '__main__':
 154     main()
 155
 156 # VIM MODLINE
 157 # vim: ai ts=4 sw=4 sts=4 expandtab