Librarian in regular requirements.

[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):
 
 
 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)
     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
 
     @classmethod

-    def get(cls, user):
+    def get(cls, user, beta=False):

         try:
         try:

-            return cls.objects.get(user=user)
+            return cls.objects.get(user=user, beta=beta)

         except cls.DoesNotExist:
         except cls.DoesNotExist:

-            o = cls(user=user)
+            o = cls(user=user, beta=beta)

             o.save()
             return o
 
             o.save()
             return o