diff options
Diffstat (limited to 'node_classes.py')
| -rw-r--r-- | node_classes.py | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/node_classes.py b/node_classes.py index a53d0a34..ef42194c 100644 --- a/node_classes.py +++ b/node_classes.py @@ -457,10 +457,20 @@ class Raise(StmtMixIn, NodeNG): class Return(StmtMixIn, NodeNG): """class representing a Return node""" +def _filter_none(node): + """transform Const(None) to None""" + if isinstance(node, Const) and node.value is None: + return None + else: + return node class Slice(NodeNG): """class representing a Slice node""" - + def __init__(self, lower, upper, step, lineno): + self.lower = _filter_none(lower) + self.upper = _filter_none(upper) + self.step = _filter_none(step) + self.lineno = lineno class Subscript(NodeNG): """class representing a Subscript node""" |
