From d49bb105344a037fe98d557c9ce30f2f1da7edd6 Mon Sep 17 00:00:00 2001 From: Fabian Dill Date: Tue, 3 Mar 2020 00:17:36 +0100 Subject: [PATCH] remove unused checks (as indicated by new type annotations) --- BaseClasses.py | 17 ++++------------- EntranceShuffle.py | 2 +- 2 files changed, 5 insertions(+), 14 deletions(-) diff --git a/BaseClasses.py b/BaseClasses.py index 1c64f7be..43cf1da9 100644 --- a/BaseClasses.py +++ b/BaseClasses.py @@ -116,9 +116,7 @@ class World(object): def get_regions(self, player=None): return self.regions if player is None else self._region_cache[player].values() - def get_region(self, regionname, player: int) -> Region: - if isinstance(regionname, Region): - return regionname + def get_region(self, regionname: str, player: int) -> Region: try: return self._region_cache[player][regionname] except KeyError: @@ -128,9 +126,7 @@ class World(object): return region raise RuntimeError('No such region %s for player %d' % (regionname, player)) - def get_entrance(self, entrance, player: int) -> Entrance: - if isinstance(entrance, Entrance): - return entrance + def get_entrance(self, entrance: str, player: int) -> Entrance: try: return self._entrance_cache[(entrance, player)] except KeyError: @@ -141,9 +137,7 @@ class World(object): return exit raise RuntimeError('No such entrance %s for player %d' % (entrance, player)) - def get_location(self, location, player: int) -> Location: - if isinstance(location, Location): - return location + def get_location(self, location: str, player: int) -> Location: try: return self._location_cache[(location, player)] except KeyError: @@ -154,10 +148,7 @@ class World(object): return r_location raise RuntimeError('No such location %s for player %d' % (location, player)) - def get_dungeon(self, dungeonname, player: int) -> Dungeon: - if isinstance(dungeonname, Dungeon): - return dungeonname - + def get_dungeon(self, dungeonname: str, player: int) -> Dungeon: for dungeon in self.dungeons: if dungeon.name == dungeonname and dungeon.player == player: return dungeon diff --git a/EntranceShuffle.py b/EntranceShuffle.py index 49339ab2..aa4538f1 100644 --- a/EntranceShuffle.py +++ b/EntranceShuffle.py @@ -1766,7 +1766,7 @@ def connect_simple(world, exitname, regionname, player): world.get_entrance(exitname, player).connect(world.get_region(regionname, player)) -def connect_entrance(world, entrancename, exitname, player): +def connect_entrance(world, entrancename: str, exitname: str, player: int): entrance = world.get_entrance(entrancename, player) # check if we got an entrance or a region to connect to try: