This commit is contained in:
Jonas Zeunert
2020-03-09 20:23:20 +01:00
parent 634cfa4106
commit 71cab5b474

View File

@@ -6,9 +6,9 @@ class DisplayFactory:
self.displayNames = displayNames
def createSpecialDisplay(self):
name = next(displayName for displayName in self.displayNames if "Special" in displayName)
name = next(displayName.name for displayName in self.displayNames if "Special" in displayName.name)
return SpecialDisplay(name)
def createPlayerDisplay(self, id):
name = next(displayName for displayName in self.displayNames if "Player" in displayName and str(id) in displayName)
name = next(displayName.name for displayName in self.displayNames if "Player" in displayName.name and str(id) in displayName.name)
return PlayerDisplay(name)