mirror of https://github.com/mvt-project/mvt.git
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
# Mobile Verification Toolkit (MVT)
|
|
# Copyright (c) 2021 MVT Project Developers.
|
|
# See the file 'LICENSE' for usage and copying permissions, or find a copy at
|
|
# https://github.com/mvt-project/mvt/blob/main/LICENSE
|
|
|
|
# From: https://gist.github.com/stanchan/bce1c2d030c76fe9223b5ff6ad0f03db
|
|
|
|
from click import command, option, 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 = kwargs.get("help", "")
|
|
if self.mutually_exclusive:
|
|
ex_str = ", ".join(self.mutually_exclusive)
|
|
kwargs["help"] = help + (
|
|
" NOTE: This argument is mutually exclusive with "
|
|
"arguments: [" + ex_str + "]."
|
|
)
|
|
super(MutuallyExclusiveOption, self).__init__(*args, **kwargs)
|
|
|
|
def handle_parse_result(self, ctx, opts, args):
|
|
if self.mutually_exclusive.intersection(opts) and self.name in opts:
|
|
raise UsageError(
|
|
"Illegal usage: `{}` is mutually exclusive with "
|
|
"arguments `{}`.".format(
|
|
self.name,
|
|
", ".join(self.mutually_exclusive)
|
|
)
|
|
)
|
|
|
|
return super(MutuallyExclusiveOption, self).handle_parse_result(
|
|
ctx,
|
|
opts,
|
|
args
|
|
)
|