Depluralised buffer_tokens

bambu_buffer/__init__.py

@@ -31,7 +31,7 @@ class BufferThread(Thread):
 
 def post(item, author, **kwargs):
     try:
-        token = author.buffer_tokens.get()
+        token = author.buffer_token
     except BufferToken.DoesNotExist:
         return
 

bambu_buffer/migrations/0001_initial.py

@@ -57,7 +57,7 @@ class Migration(migrations.Migration):
             fields=[
                 ('id', models.AutoField(verbose_name='ID', serialize=False, auto_created=True, primary_key=True)),
                 ('token', models.CharField(max_length=36)),
-                ('user', models.OneToOneField(related_name='buffer_tokens', to=settings.AUTH_USER_MODEL)),
+                ('user', models.OneToOneField(related_name='buffer_token', to=settings.AUTH_USER_MODEL)),
             ],
             options={
                 'db_table': 'buffer_token',

bambu_buffer/models.py

@@ -6,7 +6,7 @@ from bambu_buffer.receivers import *
 import requests, json
 
 class BufferToken(models.Model):
-    user = models.OneToOneField('auth.User', related_name = 'buffer_tokens')
+    user = models.OneToOneField('auth.User', related_name = 'buffer_token')
     token = models.CharField(max_length = 36)
 
     def __unicode__(self):

bambu_buffer/views.py

@@ -59,8 +59,13 @@ def callback(request):
         token = data.get('access_token')
 
         with transaction.atomic():
-            request.user.buffer_tokens.all().delete()
-            request.user.buffer_tokens.create(
+            try:
+                request.user.buffer_token.delete()
+            except BufferToken.DoesNotExist:
+                pass
+
+            BufferToken.objects.create(
+                user = user,
                 token = token
             )
 
@@ -76,7 +81,7 @@ def callback(request):
 @login_required
 def profiles(request):
     try:
-        token = request.user.buffer_tokens.get()
+        token = request.user.buffer_token
     except BufferToken.DoesNotExist:
         return HttpResponseRedirect(
             reverse('buffer_auth')
@@ -120,7 +125,7 @@ def profiles(request):
 @login_required
 def refresh(request):
     try:
-        token = request.user.buffer_tokens.get()
+        token = request.user.buffer_token
     except BufferToken.DoesNotExist:
         return HttpResponseRedirect(
             reverse('buffer_auth')