Add all value to --level argument.

This commit is contained in:
Sergey Vartanov 2021-09-12 21:27:17 +03:00
parent b8999a7d07
commit ccf693455f

View file

@ -165,8 +165,9 @@ class Constructor:
self.extractor: ShapeExtractor = extractor self.extractor: ShapeExtractor = extractor
self.configuration: MapConfiguration = configuration self.configuration: MapConfiguration = configuration
if self.configuration.level: if self.configuration.level == "all":
if self.configuration.level == "overground": self.check_level = lambda x: True
elif self.configuration.level == "overground":
self.check_level = check_level_overground self.check_level = check_level_overground
elif self.configuration.level == "underground": elif self.configuration.level == "underground":
self.check_level = lambda x: not check_level_overground(x) self.check_level = lambda x: not check_level_overground(x)
@ -174,8 +175,6 @@ class Constructor:
self.check_level = lambda x: not check_level_number( self.check_level = lambda x: not check_level_number(
x, float(self.configuration.level) x, float(self.configuration.level)
) )
else:
self.check_level = lambda x: True
self.points: list[Point] = [] self.points: list[Point] = []
self.figures: list[StyledFigure] = [] self.figures: list[StyledFigure] = []