Merge branch 'confirm' into master

This commit is contained in:
Devaev Maxim
2020-10-15 09:56:52 +03:00
10 changed files with 24 additions and 7 deletions

View File

@@ -104,6 +104,7 @@ def init(
prog: Optional[str]=None,
description: Optional[str]=None,
add_help: bool=True,
check_run: bool=False,
argv: Optional[List[str]]=None,
**load: bool,
) -> Tuple[argparse.ArgumentParser, List[str], Section]:
@@ -118,6 +119,9 @@ def init(
help="Override config options list (like sec/sub/opt=value)")
args_parser.add_argument("-m", "--dump-config", dest="dump_config", action="store_true",
help="View current configuration (include all overrides)")
if check_run:
args_parser.add_argument("--run", dest="run", action="store_true",
help="Run the service")
(options, remaining) = args_parser.parse_known_args(argv)
if options.dump_config:
@@ -135,6 +139,14 @@ def init(
logging.captureWarnings(True)
logging.config.dictConfig(config.logging)
if check_run and not options.run:
raise SystemExit(
"To prevent accidental startup, you must specify the --run option to start.\n"
"Try the --help option to find out what this service does.\n"
"Make sure you understand exactly what you are doing!"
)
return (args_parser, remaining, config)

View File

@@ -78,6 +78,7 @@ def main(argv: Optional[List[str]]=None) -> None:
config = init(
prog="kvmd-cleanup",
description="Kill KVMD and clear resources",
check_run=True,
argv=argv,
)[2].kvmd

View File

@@ -38,6 +38,7 @@ def main(argv: Optional[List[str]]=None) -> None:
config = init(
prog="kvmd-ipmi",
description="IPMI to KVMD proxy",
check_run=True,
argv=argv,
)[2].ipmi

View File

@@ -47,6 +47,7 @@ def main(argv: Optional[List[str]]=None) -> None:
prog="kvmd",
description="The main Pi-KVM daemon",
argv=argv,
check_run=True,
load_auth=True,
load_hid=True,
load_atx=True,

View File

@@ -39,6 +39,7 @@ def main(argv: Optional[List[str]]=None) -> None:
config = init(
prog="kvmd-vnc",
description="VNC to KVMD proxy",
check_run=True,
argv=argv,
)[2].vnc