X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/fdd62169ba22c4c1be2f2306b5339eadd74ffb6d..3b0d1718ec6c9dddcee4a4620384a83e30ff6f2f:/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