X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/4c48c47d0b84e516114c16ee191359166e93a51c..4d9632481149b2f7792505d521bdfdbae5245cc4:/apps/apiclient/models.py diff --git a/apps/apiclient/models.py b/apps/apiclient/models.py index d3c8f620..64130bc1 100644 --- a/apps/apiclient/models.py +++ b/apps/apiclient/models.py @@ -3,17 +3,18 @@ from django.contrib.auth.models import User class OAuthConnection(models.Model): - user = models.OneToOneField(User) + user = models.ForeignKey(User) access = models.BooleanField(default=False) token = models.CharField(max_length=64, null=True, blank=True) token_secret = models.CharField(max_length=64, null=True, blank=True) + beta = models.BooleanField(default=False) @classmethod - def get(cls, user): + def get(cls, user, beta=False): try: - return cls.objects.get(user=user) + return cls.objects.get(user=user, beta=beta) except cls.DoesNotExist: - o = cls(user=user) + o = cls(user=user, beta=beta) o.save() return o