change filter

This commit is contained in:
Jonas Zeunert
2020-03-09 20:16:51 +01:00
parent 9b675e5b58
commit 634cfa4106

View File

@@ -6,9 +6,9 @@ class DisplayFactory:
self.displayNames = displayNames self.displayNames = displayNames
def createSpecialDisplay(self): 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) return SpecialDisplay(name)
def createPlayerDisplay(self, id): 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) return PlayerDisplay(name)