452 lines
15 KiB
Python
452 lines
15 KiB
Python
import os
|
|
import sys
|
|
import stat
|
|
|
|
from ._compat import open_stream, text_type, filename_to_ui, get_streerror
|
|
from .exceptions import BadParameter
|
|
from .utils import safecall, LazyFile
|
|
|
|
|
|
class ParamType(object):
|
|
"""Helper for converting values through types. The following is
|
|
necessary for a valid type:
|
|
|
|
* it needs a name
|
|
* it needs to pass through None unchanged
|
|
* it needs to convert from a string
|
|
* it needs to convert its result type through unchanged
|
|
(eg: needs to be idempotent)
|
|
* it needs to be able to deal with param and context being `None`.
|
|
This can be the case when the object is used with prompt
|
|
inputs.
|
|
"""
|
|
|
|
#: the descriptive name of this type
|
|
name = None
|
|
|
|
#: if a list of this type is expected and the value is pulled from a
|
|
#: string environment variable, this is what splits it up. `None`
|
|
#: means any whitespace. For all parameters the general rule is that
|
|
#: whitespace splits them up. The exception are paths and files which
|
|
#: are split by ``os.path.pathsep`` by default (":" on Unix and ";" on
|
|
#: Windows).
|
|
envvar_list_splitter = None
|
|
|
|
def __call__(self, value, param=None, ctx=None):
|
|
if value is not None:
|
|
return self.convert(value, param, ctx)
|
|
|
|
def get_metavar(self, param):
|
|
"""Returns the metavar default for this param if it provides one."""
|
|
|
|
def get_missing_message(self, param):
|
|
"""Optionally might return extra information about a missing
|
|
parameter.
|
|
|
|
.. versionadded:: 2.0
|
|
"""
|
|
|
|
def convert(self, value, param, ctx):
|
|
"""Converts the value. This is not invoked for values that are
|
|
`None` (the missing value).
|
|
"""
|
|
return value
|
|
|
|
def split_envvar_value(self, rv):
|
|
"""Given a value from an environment variable this splits it up
|
|
into small chunks depending on the defined envvar list splitter.
|
|
|
|
If the splitter is set to `None`, which means that whitespace splits,
|
|
then leading and trailing whitespace is ignored. Otherwise, leading
|
|
and trailing splitters usually lead to empty items being included.
|
|
"""
|
|
return (rv or '').split(self.envvar_list_splitter)
|
|
|
|
def fail(self, message, param=None, ctx=None):
|
|
"""Helper method to fail with an invalid value message."""
|
|
raise BadParameter(message, ctx=ctx, param=param)
|
|
|
|
|
|
class FuncParamType(ParamType):
|
|
|
|
def __init__(self, func):
|
|
self.name = func.__name__
|
|
self.func = func
|
|
|
|
def convert(self, value, param, ctx):
|
|
try:
|
|
return self.func(value)
|
|
except ValueError:
|
|
try:
|
|
value = text_type(value)
|
|
except UnicodeError:
|
|
value = str(value).decode('utf-8', 'replace')
|
|
self.fail(value, param, ctx)
|
|
|
|
|
|
class StringParamType(ParamType):
|
|
name = 'text'
|
|
|
|
def convert(self, value, param, ctx):
|
|
if isinstance(value, bytes):
|
|
try:
|
|
enc = getattr(sys.stdin, 'encoding', None)
|
|
if enc is not None:
|
|
value = value.decode(enc)
|
|
except UnicodeError:
|
|
try:
|
|
value = value.decode(sys.getfilesystemencoding())
|
|
except UnicodeError:
|
|
value = value.decode('utf-8', 'replace')
|
|
return value
|
|
return value
|
|
|
|
def __repr__(self):
|
|
return 'STRING'
|
|
|
|
|
|
class Choice(ParamType):
|
|
"""The choice type allows a value to checked against a fixed set of
|
|
supported values. All of these values have to be strings.
|
|
|
|
See :ref:`choice-opts` for an example.
|
|
"""
|
|
name = 'choice'
|
|
|
|
def __init__(self, choices):
|
|
self.choices = choices
|
|
|
|
def get_metavar(self, param):
|
|
return '[%s]' % '|'.join(self.choices)
|
|
|
|
def get_missing_message(self, param):
|
|
return 'Choose from %s.' % ', '.join(self.choices)
|
|
|
|
def convert(self, value, param, ctx):
|
|
# Exact match
|
|
if value in self.choices:
|
|
return value
|
|
|
|
# Match through normalization
|
|
if ctx is not None and \
|
|
ctx.token_normalize_func is not None:
|
|
value = ctx.token_normalize_func(value)
|
|
for choice in self.choices:
|
|
if ctx.token_normalize_func(choice) == value:
|
|
return choice
|
|
|
|
self.fail('invalid choice: %s. (choose from %s)' %
|
|
(value, ', '.join(self.choices)), param, ctx)
|
|
|
|
def __repr__(self):
|
|
return 'Choice(%r)' % list(self.choices)
|
|
|
|
|
|
class IntParamType(ParamType):
|
|
name = 'integer'
|
|
|
|
def convert(self, value, param, ctx):
|
|
try:
|
|
return int(value)
|
|
except ValueError:
|
|
self.fail('%s is not a valid integer' % value, param, ctx)
|
|
|
|
def __repr__(self):
|
|
return 'INT'
|
|
|
|
|
|
class IntRange(IntParamType):
|
|
"""A parameter that works similar to :data:`click.INT` but restricts
|
|
the value to fit into a range. The default behavior is to fail if the
|
|
value falls outside the range, but it can also be silently clamped
|
|
between the two edges.
|
|
|
|
See :ref:`ranges` for an example.
|
|
"""
|
|
name = 'integer range'
|
|
|
|
def __init__(self, min=None, max=None, clamp=False):
|
|
self.min = min
|
|
self.max = max
|
|
self.clamp = clamp
|
|
|
|
def convert(self, value, param, ctx):
|
|
rv = IntParamType.convert(self, value, param, ctx)
|
|
if self.clamp:
|
|
if self.min is not None and rv < self.min:
|
|
return self.min
|
|
if self.max is not None and rv > self.max:
|
|
return self.max
|
|
if self.min is not None and rv < self.min or \
|
|
self.max is not None and rv > self.max:
|
|
if self.min is None:
|
|
self.fail('%s is bigger than the maximum valid value '
|
|
'%s.' % (rv, self.max), param, ctx)
|
|
elif self.max is None:
|
|
self.fail('%s is smaller than the minimum valid value '
|
|
'%s.' % (rv, self.min), param, ctx)
|
|
else:
|
|
self.fail('%s is not in the valid range of %s to %s.'
|
|
% (rv, self.min, self.max), param, ctx)
|
|
return rv
|
|
|
|
def __repr__(self):
|
|
return 'IntRange(%r, %r)' % (self.min, self.max)
|
|
|
|
|
|
class BoolParamType(ParamType):
|
|
name = 'boolean'
|
|
|
|
def convert(self, value, param, ctx):
|
|
if isinstance(value, bool):
|
|
return bool(value)
|
|
value = value.lower()
|
|
if value in ('true', '1', 'yes', 'y'):
|
|
return True
|
|
elif value in ('false', '0', 'no', 'n'):
|
|
return False
|
|
self.fail('%s is not a valid boolean' % value, param, ctx)
|
|
|
|
def __repr__(self):
|
|
return 'BOOL'
|
|
|
|
|
|
class FloatParamType(ParamType):
|
|
name = 'float'
|
|
|
|
def convert(self, value, param, ctx):
|
|
try:
|
|
return float(value)
|
|
except ValueError:
|
|
self.fail('%s is not a valid floating point value' %
|
|
value, param, ctx)
|
|
|
|
def __repr__(self):
|
|
return 'FLOAT'
|
|
|
|
|
|
class UUIDParameterType(ParamType):
|
|
name = 'uuid'
|
|
|
|
def convert(self, value, param, ctx):
|
|
import uuid
|
|
try:
|
|
return uuid.UUID(value)
|
|
except ValueError:
|
|
self.fail('%s is not a valid UUID value' % value, param, ctx)
|
|
|
|
def __repr__(self):
|
|
return 'UUID'
|
|
|
|
|
|
class File(ParamType):
|
|
"""Declares a parameter to be a file for reading or writing. The file
|
|
is automatically closed once the context tears down (after the command
|
|
finished working).
|
|
|
|
Files can be opened for reading or writing. The special value ``-``
|
|
indicates stdin or stdout depending on the mode.
|
|
|
|
By default, the file is opened for reading text data, but it can also be
|
|
opened in binary mode or for writing. The encoding parameter can be used
|
|
to force a specific encoding.
|
|
|
|
The `lazy` flag controls if the file should be opened immediately or
|
|
upon first IO. The default is to be non lazy for standard input and
|
|
output streams as well as files opened for reading, lazy otherwise.
|
|
|
|
Starting with Click 2.0, files can also be opened atomically in which
|
|
case all writes go into a separate file in the same folder and upon
|
|
completion the file will be moved over to the original location. This
|
|
is useful if a file regularly read by other users is modified.
|
|
|
|
See :ref:`file-args` for more information.
|
|
"""
|
|
name = 'filename'
|
|
envvar_list_splitter = os.path.pathsep
|
|
|
|
def __init__(self, mode='r', encoding=None, errors='strict', lazy=None,
|
|
atomic=False):
|
|
self.mode = mode
|
|
self.encoding = encoding
|
|
self.errors = errors
|
|
self.lazy = lazy
|
|
self.atomic = atomic
|
|
|
|
def resolve_lazy_flag(self, value):
|
|
if self.lazy is not None:
|
|
return self.lazy
|
|
if value == '-':
|
|
return False
|
|
elif 'w' in self.mode:
|
|
return True
|
|
return False
|
|
|
|
def convert(self, value, param, ctx):
|
|
try:
|
|
if hasattr(value, 'read') or hasattr(value, 'write'):
|
|
return value
|
|
|
|
lazy = self.resolve_lazy_flag(value)
|
|
|
|
if lazy:
|
|
f = LazyFile(value, self.mode, self.encoding, self.errors,
|
|
atomic=self.atomic)
|
|
if ctx is not None:
|
|
ctx.call_on_close(f.close_intelligently)
|
|
return f
|
|
|
|
f, should_close = open_stream(value, self.mode,
|
|
self.encoding, self.errors,
|
|
atomic=self.atomic)
|
|
# If a context is provided, we automatically close the file
|
|
# at the end of the context execution (or flush out). If a
|
|
# context does not exist, it's the caller's responsibility to
|
|
# properly close the file. This for instance happens when the
|
|
# type is used with prompts.
|
|
if ctx is not None:
|
|
if should_close:
|
|
ctx.call_on_close(safecall(f.close))
|
|
else:
|
|
ctx.call_on_close(safecall(f.flush))
|
|
return f
|
|
except (IOError, OSError) as e:
|
|
self.fail('Could not open file: %s: %s' % (
|
|
filename_to_ui(value),
|
|
get_streerror(e),
|
|
), param, ctx)
|
|
|
|
|
|
class Path(ParamType):
|
|
"""The path type is similar to the :class:`File` type but it performs
|
|
different checks. First of all, instead of returning a open file
|
|
handle it returns just the filename. Secondly, it can perform various
|
|
basic checks about what the file or directory should be.
|
|
|
|
:param exists: if set to true, the file or directory needs to exist for
|
|
this value to be valid. If this is not required and a
|
|
file does indeed not exist, then all further checks are
|
|
silently skipped.
|
|
:param file_okay: controls if a file is a possible value.
|
|
:param dir_okay: controls if a directory is a possible value.
|
|
:param writable: if true, a writable check is performed.
|
|
:param readable: if true, a readable check is performed.
|
|
:param resolve_path: if this is true, then the path is fully resolved
|
|
before the value is passed onwards. This means
|
|
that it's absolute and symlinks are resolved.
|
|
"""
|
|
envvar_list_splitter = os.path.pathsep
|
|
|
|
def __init__(self, exists=False, file_okay=True, dir_okay=True,
|
|
writable=False, readable=True, resolve_path=False):
|
|
self.exists = exists
|
|
self.file_okay = file_okay
|
|
self.dir_okay = dir_okay
|
|
self.writable = writable
|
|
self.readable = readable
|
|
self.resolve_path = resolve_path
|
|
|
|
if self.file_okay and not self.dir_okay:
|
|
self.name = 'file'
|
|
self.path_type = 'File'
|
|
if self.dir_okay and not self.file_okay:
|
|
self.name = 'directory'
|
|
self.path_type = 'Directory'
|
|
else:
|
|
self.name = 'path'
|
|
self.path_type = 'Path'
|
|
|
|
def convert(self, value, param, ctx):
|
|
rv = value
|
|
if self.resolve_path:
|
|
rv = os.path.realpath(rv)
|
|
|
|
try:
|
|
st = os.stat(rv)
|
|
except OSError:
|
|
if not self.exists:
|
|
return rv
|
|
self.fail('%s "%s" does not exist.' % (
|
|
self.path_type,
|
|
filename_to_ui(value)
|
|
), param, ctx)
|
|
|
|
if not self.file_okay and stat.S_ISREG(st.st_mode):
|
|
self.fail('%s "%s" is a file.' % (
|
|
self.path_type,
|
|
filename_to_ui(value)
|
|
), param, ctx)
|
|
if not self.dir_okay and stat.S_ISDIR(st.st_mode):
|
|
self.fail('%s "%s" is a directory.' % (
|
|
self.path_type,
|
|
filename_to_ui(value)
|
|
), param, ctx)
|
|
if self.writable and not os.access(value, os.W_OK):
|
|
self.fail('%s "%s" is not writable.' % (
|
|
self.path_type,
|
|
filename_to_ui(value)
|
|
), param, ctx)
|
|
if self.readable and not os.access(value, os.R_OK):
|
|
self.fail('%s "%s" is not readable.' % (
|
|
self.path_type,
|
|
filename_to_ui(value)
|
|
), param, ctx)
|
|
|
|
return rv
|
|
|
|
|
|
def convert_type(ty, default=None):
|
|
"""Converts a callable or python ty into the most appropriate param
|
|
ty.
|
|
"""
|
|
if isinstance(ty, ParamType):
|
|
return ty
|
|
guessed_type = False
|
|
if ty is None and default is not None:
|
|
ty = type(default)
|
|
guessed_type = True
|
|
if ty is text_type or ty is str or ty is None:
|
|
return STRING
|
|
if ty is int:
|
|
return INT
|
|
# Booleans are only okay if not guessed. This is done because for
|
|
# flags the default value is actually a bit of a lie in that it
|
|
# indicates which of the flags is the one we want. See get_default()
|
|
# for more information.
|
|
if ty is bool and not guessed_type:
|
|
return BOOL
|
|
if ty is float:
|
|
return FLOAT
|
|
if guessed_type:
|
|
return STRING
|
|
|
|
# Catch a common mistake
|
|
if __debug__:
|
|
try:
|
|
if issubclass(ty, ParamType):
|
|
raise AssertionError('Attempted to use an uninstantiated '
|
|
'parameter type (%s).' % ty)
|
|
except TypeError:
|
|
pass
|
|
return FuncParamType(ty)
|
|
|
|
|
|
#: A unicode string parameter type which is the implicit default. This
|
|
#: can also be selected by using ``str`` as type.
|
|
STRING = StringParamType()
|
|
|
|
#: An integer parameter. This can also be selected by using ``int`` as
|
|
#: type.
|
|
INT = IntParamType()
|
|
|
|
#: A floating point value parameter. This can also be selected by using
|
|
#: ``float`` as type.
|
|
FLOAT = FloatParamType()
|
|
|
|
#: A boolean parameter. This is the default for boolean flags. This can
|
|
#: also be selected by using ``bool`` as a type.
|
|
BOOL = BoolParamType()
|
|
|
|
#: A UUID parameter.
|
|
UUID = UUIDParameterType()
|