Skip to content

Commit

Permalink
Merge pull request #565 from fomars/release
Browse files Browse the repository at this point in the history
fix converter: cast empty strings to null
  • Loading branch information
fomars committed Apr 18, 2018
2 parents f9cb66d + 47db702 commit 89b46f5
Showing 1 changed file with 13 additions and 4 deletions.
17 changes: 13 additions & 4 deletions yandextank/config_converter/converter.py
Original file line number Diff line number Diff line change
Expand Up @@ -162,12 +162,21 @@ class UnknownOption(ConversionError):
pass


def empty_to_none(func):
def new_func(k, v):
if v in '':
return {k: None}
else:
return func(k, v)
return new_func


class Option(object):
TYPE_CASTERS = {
'boolean': lambda k, v: {k: to_bool(v)},
'integer': lambda k, v: {k: int(v)},
'list': lambda k, v: {k: [_.strip() for _ in v.strip().split()]},
'float': lambda k, v: {k: float(v)}
'boolean': empty_to_none(lambda k, v: {k: to_bool(v)}),
'integer': empty_to_none(lambda k, v: {k: int(v)}),
'list': empty_to_none(lambda k, v: {k: [_.strip() for _ in v.strip().split()]}),
'float': empty_to_none(lambda k, v: {k: float(v)})
}

SPECIAL_CONVERTERS = {
Expand Down

0 comments on commit 89b46f5

Please sign in to comment.