yet another fix for beta

[redakcja.git] / apps / apiclient / models.py

diff --git a/apps/apiclient/models.py b/apps/apiclient/models.py
index d3c8f62..64130bc 100644 (file)
--- 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