sima/utils/startopt.py

   1 # -*- coding: utf-8 -*-
   2
   3 # Copyright (c) 2009-2015, 2021 kaliko <kaliko@azylum.org>
   4 #
   5 #  This file is part of sima
   6 #
   7 #  sima is free software: you can redistribute it and/or modify
   8 #  it under the terms of the GNU General Public License as published by
   9 #  the Free Software Foundation, either version 3 of the License, or
  10 #  (at your option) any later version.
  11 #
  12 #  sima is distributed in the hope that it will be useful,
  13 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 #  GNU General Public License for more details.
  16 #
  17 #  You should have received a copy of the GNU General Public License
  18 #  along with sima.  If not, see <http://www.gnu.org/licenses/>.
  19 #
  20 #
  21
  22 from argparse import ArgumentParser, RawDescriptionHelpFormatter, SUPPRESS
  23
  24
  25 from .utils import Wfile, Rfile, Wdir
  26
  27 DESCRIPTION = """
  28 MPD_sima automagicaly queue new tracks in MPD playlist.
  29
  30 Command line options override their equivalent in configuration file.
  31 If a positional arguments is provided MPD_sima execute the command and returns.
  32 Commands available:
  33 {}
  34 """
  35
  36
  37 def clean_dict(to_clean):
  38     """Remove items which values are set to None/False"""
  39     for k in list(to_clean.keys()):
  40         if not to_clean.get(k):
  41             to_clean.pop(k)
  42
  43 # COMMANDS LIST
  44 CMDS = {'config-test': 'Test configuration (MPD connection and Tags plugin only)',
  45         'create-db': 'Create the database',
  46         'generate-config': 'Generate a configuration file to stdout',
  47         'purge-history': 'Remove play history'
  48         }
  49 # OPTIONS LIST
  50 # pop out 'sw' value before creating Parser object.
  51 # PAY ATTENTION:
  52 #   If an option has to override its dual in conf file, the destination
  53 #   identifier "dest" is to be named after that option in the conf file.
  54 #   The supersedes_config_with_cmd_line_options method in ConfMan() (config.py)
  55 #   is looking for command line option names identical to config file option
  56 #   name it is meant to override.
  57 OPTS = [
  58     {
  59         'sw': ['-l', '--log'],
  60         'type': str,
  61         'dest': 'logfile',
  62         'action': Wfile,
  63         'metavar': 'LOG',
  64         'help': 'file to log message to, default is stdout/stderr'},
  65     {
  66         'sw': ['-v', '--log-level'],
  67         'type': str,
  68         'dest': 'verbosity',
  69         'choices': ['debug', 'info', 'warning', 'error'],
  70         'help': 'log messages verbosity, default is info'},
  71     {
  72         'sw': ['-p', '--pid'],
  73         'dest': 'pidfile',
  74         'action': Wfile,
  75         'metavar': 'FILE',
  76         'help': 'file to save PID to, default is not to store pid'},
  77     {
  78         'sw': ['-d', '--daemon'],
  79         'dest': 'daemon',
  80         'action': 'store_true',
  81         'help': 'daemonize process'},
  82     {
  83         'sw': ['-S', '--host'],
  84         'dest': 'host',
  85         'help': 'host MPD in running on (IP or FQDN)'},
  86     {
  87         'sw': ['-P', '--port'],
  88         'type': int,
  89         'dest': 'port',
  90         'help': 'port MPD in listening on'},
  91     {
  92         'sw': ['-c', '--config'],
  93         'dest': 'conf_file',
  94         'action': Rfile,
  95         'metavar': 'CONFIG',
  96         'help': 'configuration file to load'},
  97     {
  98         'sw': ['--var-dir', '--var_dir'],
  99         'dest': 'var_dir',
 100         'action': Wdir,
 101         'help': 'directory to store var content (ie. database, cache)'},
 102     {
 103         'sw': ['command'],
 104         'nargs': '?',
 105         'choices': CMDS.keys(),
 106         'help': 'command to run (cf. description or unix manual for more)'},
 107 ]
 108
 109
 110 class StartOpt:
 111     """Command line management.
 112     """
 113
 114     def __init__(self, script_info,):
 115         self.parser = None
 116         self.info = dict(script_info)
 117         self.options = dict()
 118         self.main()
 119
 120     def declare_opts(self):
 121         """
 122         Declare options in ArgumentParser object.
 123         """
 124         cmds = '\n'.join([f'    * {k}: {v}' for k, v in CMDS.items()])
 125         self.parser = ArgumentParser(description=DESCRIPTION.format(cmds),
 126                                      prog=self.info.get('prog'),
 127                                      epilog='Happy Listening',
 128                                      formatter_class=RawDescriptionHelpFormatter,
 129                                      )
 130         self.parser.add_argument('--version', action='version',
 131                         version='%(prog)s {version}'.format(**self.info))
 132         # Add all options declare in OPTS
 133         for opt in OPTS:
 134             opt_names = opt.pop('sw')
 135             self.parser.add_argument(*opt_names, **opt)
 136
 137     def main(self):
 138         """
 139         Look for env. var and parse command line.
 140         """
 141         self.declare_opts()
 142         options = vars(self.parser.parse_args())
 143         # Set log file to os.devnull in daemon mode to avoid logging to
 144         # std(out|err).
 145         # TODO: Probably useless. To be checked
 146         #if options.__dict__.get('daemon', False) and \
 147         #        not options.__dict__.get('logfile', False):
 148         #    options.__dict__['logfile'] = devnull
 149         self.options.update(options)
 150         clean_dict(self.options)
 151
 152
 153 # VIM MODLINE
 154 # vim: ai ts=4 sw=4 sts=4 expandtab