· 7 years ago · Dec 11, 2017, 06:28 PM
1class Achievements(models.Model):
2 name = models.CharField(max_length = 40)
3 points = models.IntegerField(blank=True, null = True)
4 date = models.DateField(auto_now_add=True)
5 description = models.CharField(max_length = 140)
6 def __unicode__(self):
7 return self.name
8
9 class Meta:
10 verbose_name_plural = "Achievements"
11
12class ProloggerUser(models.Model):
13 user = models.ForeignKey(User, unique=True)
14 oauthtoken = models.CharField(max_length = 50)
15 achievements = models.ManyToManyField(Achievements, blank=True)
16 objects = ProloggerUserManager()
17
18 def __unicode__(self):
19 return u"Prologger information for %s" % self.user
20 class Meta:
21 verbose_name_plural = "ProloggerUsers"
22
23 def add_achievement(name):
24 achievement = Achievements.objects.get(name=name)
25 self.achievements.add(achievement)
26 self.save()