mirror of https://github.com/mvt-project/mvt.git
34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
# Mobile Verification Toolkit (MVT)
|
|
# Copyright (c) 2021-2023 The MVT Authors.
|
|
# Use of this software is governed by the MVT License 1.1 that can be found at
|
|
# https://license.mvt.re/1.1/
|
|
|
|
# From: https://gist.github.com/stanchan/bce1c2d030c76fe9223b5ff6ad0f03db
|
|
|
|
from click import Option, UsageError
|
|
|
|
|
|
class MutuallyExclusiveOption(Option):
|
|
"""This class extends click to support mutually exclusive options."""
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self.mutually_exclusive = set(kwargs.pop("mutually_exclusive", []))
|
|
help_msg = kwargs.get("help", "")
|
|
if self.mutually_exclusive:
|
|
ex_str = ", ".join(self.mutually_exclusive)
|
|
kwargs["help"] = (
|
|
f"{help_msg} NOTE: This argument is mutually exclusive with arguments"
|
|
f"[{ex_str}]."
|
|
)
|
|
|
|
super().__init__(*args, **kwargs)
|
|
|
|
def handle_parse_result(self, ctx, opts, args):
|
|
if self.mutually_exclusive.intersection(opts) and self.name in opts:
|
|
raise UsageError(
|
|
f"Illegal usage: `{self.name}` is mutually exclusive "
|
|
f"with arguments `{', '.join(self.mutually_exclusive)}`."
|
|
)
|
|
|
|
return super().handle_parse_result(ctx, opts, args)
|