-a = emit graph of entire dungeon
-d = emit graoh of mazw all different
+-f = emit graph of forest locations
-m = emit graph of maze all alike
--s = emit graph of surface locations
--v = include internal sy,no;s in room labels
+-s = emit graph of non-forest surface locations
+-v = include internal symbols in room labels
"""
# Copyright (c) 2017 by Eric S. Raymond
# SPDX-License-Identifier: BSD-2-clause
+# pylint: disable=consider-using-f-string
+
import sys, getopt, yaml
def allalike(loc):
"Select out loci related to the Maze All Alike"
- return ("ALIKE" in loc) or (loc == "LOC_PITBRINK") or ("MAZEEND" in loc) or ("STALACTITE" in loc)
+ return location_lookup[loc]["conditions"].get("ALLALIKE")
def alldifferent(loc):
"Select out loci related to the Maze All Alike"
- return ("DIFFERENT" in loc) or (loc == "LOC_DEADEND13")
+ return location_lookup[loc]["conditions"].get("ALLDIFFERENT")
def surface(loc):
"Select out surface locations"
- attrs = location_lookup[loc]
- if ("ABOVE" in attrs["conditions"]) and attrs["conditions"]["ABOVE"]:
- return True
- if ("FOREST" in attrs["conditions"]) and attrs["conditions"]["FOREST"]:
- return True
- return False
+ return location_lookup[loc]["conditions"].get("ABOVE")
+
+def forest(loc):
+ return location_lookup[loc]["conditions"].get("FOREST")
def abbreviate(d):
m = {"NORTH":"N", "EAST":"E", "SOUTH":"S", "WEST":"W", "UPWAR":"U", "DOWN":"D"}
return not obj.get("immovable")
if __name__ == "__main__":
- with open("adventure.yaml", "r") as f:
+ with open("adventure.yaml", "r", encoding='ascii', errors='surrogateescape') as f:
db = yaml.safe_load(f)
location_lookup = dict(db["locations"])
object_lookup = dict(db["objects"])
try:
- (options, arguments) = getopt.getopt(sys.argv[1:], "admsv")
+ (options, arguments) = getopt.getopt(sys.argv[1:], "adfmsv")
except getopt.GetoptError as e:
print(e)
sys.exit(1)
subset = lambda loc: True
elif switch == '-d':
subset = alldifferent
+ elif switch == '-f':
+ subset = forest
elif switch == '-m':
subset = allalike
elif switch == '-s':