diff --git a/factories/DisplayFactory.py b/factories/DisplayFactory.py index a55c2f3..14412f5 100644 --- a/factories/DisplayFactory.py +++ b/factories/DisplayFactory.py @@ -6,9 +6,9 @@ class DisplayFactory: self.displayNames = displayNames def createSpecialDisplay(self): - name = next(displayName.name for displayName in self.displayNames if "Special" in displayName.name) + name = next(displayName['name'] for displayName in self.displayNames if "Special" in displayName['name']) return SpecialDisplay(name) def createPlayerDisplay(self, id): - name = next(displayName.name for displayName in self.displayNames if "Player" in displayName.name and str(id) in displayName.name) + name = next(displayName['name'] for displayName in self.displayNames if "Player" in displayName['name']' and str(id) in displayName['name']) return PlayerDisplay(name) diff --git a/factories/GameFactory.py b/factories/GameFactory.py index 725d9dd..597a472 100644 --- a/factories/GameFactory.py +++ b/factories/GameFactory.py @@ -14,7 +14,7 @@ from Game import Game class GameFactory: def __init__(self): - displayNames = Networking().getDisplays().displays + displayNames = Networking().getDisplays()['displays'] self.displayFactory = DisplayFactory(displayNames) specialDisplay = self.displayFactory.createSpecialDisplay()