diff --git a/plasma/node.py b/plasma/node.py index 825aaf0..7ab7633 100644 --- a/plasma/node.py +++ b/plasma/node.py @@ -17,7 +17,8 @@ class Orient(Flag): HORIZONTAL = 0 VERTICAL = 1 -HORIZONTAL, VERTICAL = Orient +HORIZONTAL = Orient.HORIZONTAL +VERTICAL = Orient.VERTICAL class Direction(Enum): UP = auto() @@ -33,7 +34,10 @@ def orient(self): def offset(self): return 1 if self in [self.RIGHT, self.DOWN] else -1 -UP, DOWN, LEFT, RIGHT = Direction +UP = Direction.UP +DOWN = Direction.DOWN +LEFT = Direction.LEFT +RIGHT = Direction.RIGHT class AddMode(Flag): HORIZONTAL = 0