Partial pipeline DB storage implementation

CHANGELOG.md

@@ -7,6 +7,9 @@ and this project adheres to [Semantic Versioning](http://semver.org/).
 
 ## [Unreleased](https://github.com/python-social-auth/social-app-django/commits/master)
 
+### Added
+- Partial pipeline DB storage implementation
+
 ### Changed
 - Monkey patch BaseAuth to load the current strategy to workaround django load_backend() call
 - Remove usage of set/get current strategy methods

social_django/migrations/0006_partial.py

@@ -0,0 +1,31 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.10.4 on 2017-01-02 11:54
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+import social_django.fields
+import social_django.storage
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('social_django', '0005_auto_20160727_2333'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Partial',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('token', models.CharField(db_index=True, max_length=32)),
+                ('next_step', models.PositiveSmallIntegerField(default=0)),
+                ('backend', models.CharField(max_length=32)),
+                ('data', social_django.fields.JSONField(default={})),
+            ],
+            options={
+                'db_table': 'social_auth_partial',
+            },
+            bases=(models.Model, social_django.storage.DjangoPartialMixin),
+        ),
+    ]

social_django/models.py

@@ -9,7 +9,7 @@
 
 from .storage import DjangoUserMixin, DjangoAssociationMixin, \
                      DjangoNonceMixin, DjangoCodeMixin, \
-                     BaseDjangoStorage
+                     DjangoPartialMixin, BaseDjangoStorage
 from .fields import JSONField
 from .managers import UserSocialAuthManager
 
@@ -111,11 +111,22 @@ class Meta:
         unique_together = ('email', 'code')
 
 
+class Partial(models.Model, DjangoPartialMixin):
+    token = models.CharField(max_length=32, db_index=True)
+    next_step = models.PositiveSmallIntegerField(default=0)
+    backend = models.CharField(max_length=32)
+    data = JSONField()
+
+    class Meta:
+        db_table = 'social_auth_partial'
+
+
 class DjangoStorage(BaseDjangoStorage):
     user = UserSocialAuth
     nonce = Nonce
     association = Association
     code = Code
+    partial = Partial
 
     @classmethod
     def is_integrity_error(cls, exception):

social_django/storage.py

@@ -6,7 +6,7 @@
 from django.db.utils import IntegrityError
 
 from social_core.storage import UserMixin, AssociationMixin, NonceMixin, \
-                                CodeMixin, BaseStorage
+                                CodeMixin, PartialMixin, BaseStorage
 
 
 class DjangoUserMixin(UserMixin):
@@ -168,6 +168,21 @@ def get_code(cls, code):
             return None
 
 
+class DjangoPartialMixin(PartialMixin):
+    @classmethod
+    def load(cls, token):
+        try:
+            return cls.objects.get(token=token)
+        except cls.DoesNotExist:
+            return None
+
+    @classmethod
+    def destroy(cls, token):
+        partial = cls.load(token)
+        if partial:
+            partial.delete()
+
+
 class BaseDjangoStorage(BaseStorage):
     user = DjangoUserMixin
     nonce = DjangoNonceMixin