1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
#! /usr/bin/env python3
# SPDX-License-Identifier: GPL-2.0-or-later
# depthcharge-tools depthchargectl config subcommand
# Copyright (C) 2021-2022 Alper Nebi Yasak <alpernebiyasak@gmail.com>
# See COPYRIGHT and LICENSE files for full copyright information.
import argparse
import logging
from depthcharge_tools import __version__
from depthcharge_tools.utils.argparse import (
Command,
Argument,
Group,
CommandExit,
)
from depthcharge_tools.depthchargectl import depthchargectl
@depthchargectl.subcommand("config")
class depthchargectl_config(
depthchargectl,
prog="depthchargectl config",
usage="%(prog)s [options] KEY",
add_help=False,
):
"""Get depthchargectl configuration values."""
_logger = depthchargectl._logger.getChild("config")
config_section = "depthchargectl/config"
@depthchargectl.board.copy()
def board(self, codename=""):
"""Assume we're running on the specified board"""
# We can query configs without knowing the board.
try:
return super().board
except Exception as err:
self.logger.warning(err)
return None
@Group
def positionals(self):
"""Positional arguments"""
@positionals.add
@Argument
def key(self, key):
"""Config key to get value of."""
return key
@Group
def options(self):
"""Options"""
@options.add
@Argument("--section", nargs=1)
def section(self, section=None):
"""Config section to work on."""
parser = self.config.parser
if section is None:
section = self.config.name
if section not in parser.sections():
if section != parser.default_section:
parser.add_section(section)
return parser[section]
@options.add
@Argument("--default", nargs=1)
def default(self, default=None):
"""Value to return if key doesn't exist in section."""
return default
def __call__(self):
if self.key not in self.section:
if self.default is not None:
return self.default
else:
raise KeyError(
"Key '{}' not found in section '{}'."
.format(self.key, self.section.name)
)
return self.section[self.key]
global_options = depthchargectl.global_options
config_options = depthchargectl.config_options
|
