X-Git-Url: https://git.stderr.nl/gitweb?p=matthijs%2Fprojects%2Fxerxes.git;a=blobdiff_plain;f=influences%2Fmodels.py;h=2c626266fde1312aa9b6445bd66ac647c6e4b112;hp=c513c1a5c307e3f29b9365c12252fac42e8acc78;hb=eb2b5fc938fcc2fdc290da0b4a94ddbd9ed5cc76;hpb=06b512bb8370c00e2f5f28096ec0d6e7d591a989 diff --git a/influences/models.py b/influences/models.py index c513c1a..2c62626 100644 --- a/influences/models.py +++ b/influences/models.py @@ -101,6 +101,23 @@ class Influence(models.Model): if (self.other_contacts): chars.extend(map(strip,self.other_contacts.split(','))) return chars + + @property + def related_players(self): + """ Returns all players to this Influence (ie, the players of the + initiator or involved characters). Returns a dict where the + players (User objects) are keys and a list of Character objects + for which this player is related is the value. + """ + players = {self.initiator.player : [self.initiator]} + for char in self.other_characters.all(): + # Add this character to the player's list of characters for + # this Influence, creating a new list if this is the first + # character. + chars = players.get(char.player, []) + chars.append(char) + players[char.player] = chars + return players class Meta: verbose_name = _("Influence")