]> kaliko git repositories - mpd-sima.git/blob - sima/utils/startopt.py
Use os.getenv instead of os.environ.get
[mpd-sima.git] / 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