diff --git a/factories/DisplayFactory.py b/factories/DisplayFactory.py index dd60017..c140238 100644 --- a/factories/DisplayFactory.py +++ b/factories/DisplayFactory.py @@ -6,9 +6,9 @@ class DisplayFactory: self.displayNames = displayNames def createSpecialDisplay(self): - name = filter(lambda displayName: "Special" in displayName, self.displayNames) + name = next(displayName for displayName in self.displayNames if "Special" in displayName) return SpecialDisplay(name) def createPlayerDisplay(self, id): - name = filter(lambda displayName: "Player" in displayName and str(id) in displayName, self.displayNames) + name = next(displayName for displayName in self.displayNames if "Player" in displayName and str(id) in displayName) return PlayerDisplay(name)