Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class TavernHero(models.Model):
- """ available heroes in a perticulair Tavern """
- tavern = models.ForeignKey(Tavern)
- #classes
- warrior = models.ForeignKey('heroes.Warrior', null=True)
- trader = models.ForeignKey('heroes.Trader', null=True)
- crafter = models.ForeignKey('heroes.Crafter', null=True)
- def __unicode__(self):
- return "Tavern Hero"
- #delete related warrior/trader/crafter object
- def delete_hero(self):
- klasses = ['warrior', 'trader', 'crafter']
- for klass in klasses:
- obj = getattr(klass)
- if obj != None:
- obj.delete()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement