change filter
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user