From ccf693455f0ff914ec98687cf5858e142e250ce7 Mon Sep 17 00:00:00 2001 From: Sergey Vartanov Date: Sun, 12 Sep 2021 21:27:17 +0300 Subject: [PATCH] Add `all` value to --level argument. --- map_machine/constructor.py | 19 +++++++++---------- 1 file changed, 9 insertions(+), 10 deletions(-) diff --git a/map_machine/constructor.py b/map_machine/constructor.py index 2f0c212..688ad89 100644 --- a/map_machine/constructor.py +++ b/map_machine/constructor.py @@ -165,17 +165,16 @@ class Constructor: self.extractor: ShapeExtractor = extractor self.configuration: MapConfiguration = configuration - if self.configuration.level: - if self.configuration.level == "overground": - self.check_level = check_level_overground - elif self.configuration.level == "underground": - self.check_level = lambda x: not check_level_overground(x) - else: - self.check_level = lambda x: not check_level_number( - x, float(self.configuration.level) - ) - else: + if self.configuration.level == "all": self.check_level = lambda x: True + elif self.configuration.level == "overground": + self.check_level = check_level_overground + elif self.configuration.level == "underground": + self.check_level = lambda x: not check_level_overground(x) + else: + self.check_level = lambda x: not check_level_number( + x, float(self.configuration.level) + ) self.points: list[Point] = [] self.figures: list[StyledFigure] = []