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
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)