more fixes for publishing on beta

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

diff --git a/apps/apiclient/models.py b/apps/apiclient/models.py
index d3c8f62..9cce9a8 100644 (file)
--- a/apps/apiclient/models.py
+++ b/apps/apiclient/models.py
@@ -7,13 +7,14 @@ class OAuthConnection(models.Model):
     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