mirror of https://github.com/sgoudham/Enso-Bot.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
2.0 KiB
Python
82 lines
2.0 KiB
Python
from __future__ import absolute_import, unicode_literals
|
|
|
|
import logging
|
|
import os
|
|
|
|
|
|
class TypeData(object):
|
|
def __init__(self, default_type, as_type):
|
|
self.default_type = default_type
|
|
self.as_type = as_type
|
|
|
|
def __repr__(self):
|
|
return "{}(base={}, as={})".format(self.__class__.__name__, self.default_type, self.as_type)
|
|
|
|
def convert(self, value):
|
|
return self.default_type(value)
|
|
|
|
|
|
class BoolType(TypeData):
|
|
BOOLEAN_STATES = {
|
|
"1": True,
|
|
"yes": True,
|
|
"true": True,
|
|
"on": True,
|
|
"0": False,
|
|
"no": False,
|
|
"false": False,
|
|
"off": False,
|
|
}
|
|
|
|
def convert(self, value):
|
|
if value.lower() not in self.BOOLEAN_STATES:
|
|
raise ValueError("Not a boolean: %s" % value)
|
|
return self.BOOLEAN_STATES[value.lower()]
|
|
|
|
|
|
class NoneType(TypeData):
|
|
def convert(self, value):
|
|
if not value:
|
|
return None
|
|
return str(value)
|
|
|
|
|
|
class ListType(TypeData):
|
|
def _validate(self):
|
|
""""""
|
|
|
|
def convert(self, value, flatten=True):
|
|
if isinstance(value, (str, bytes)):
|
|
value = filter(None, [x.strip() for x in value.splitlines()])
|
|
values = list(value)
|
|
result = []
|
|
for value in values:
|
|
sub_values = value.split(os.pathsep)
|
|
result.extend(sub_values)
|
|
converted = [self.as_type(i) for i in result]
|
|
return converted
|
|
|
|
|
|
def convert(value, as_type, source):
|
|
"""Convert the value as a given type where the value comes from the given source"""
|
|
try:
|
|
return as_type.convert(value)
|
|
except Exception as exception:
|
|
logging.warning("%s failed to convert %r as %r because %r", source, value, as_type, exception)
|
|
raise
|
|
|
|
|
|
_CONVERT = {bool: BoolType, type(None): NoneType, list: ListType}
|
|
|
|
|
|
def get_type(action):
|
|
default_type = type(action.default)
|
|
as_type = default_type if action.type is None else action.type
|
|
return _CONVERT.get(default_type, TypeData)(default_type, as_type)
|
|
|
|
|
|
__all__ = (
|
|
"convert",
|
|
"get_type",
|
|
)
|