6cb62f2697
Please note that this is a squshed commit and the work of: Symbolist, macdiesel, nedbat, doctoryes, muzaffaryousaf and muhammad-ammar
39 lines
1.6 KiB
Python
39 lines
1.6 KiB
Python
"""
|
|
Monkey patch implementation for a python_social_auth Django ORM method that is not Django 1.8-compatible.
|
|
Remove once the module fully supports Django 1.8!
|
|
"""
|
|
|
|
from django.db import transaction
|
|
from social.storage.django_orm import DjangoUserMixin
|
|
from social.apps.django_app.default.models import (
|
|
UserSocialAuth, Nonce, Association, Code
|
|
)
|
|
|
|
|
|
def patch():
|
|
"""
|
|
Monkey-patch the DjangoUserMixin class.
|
|
"""
|
|
def create_social_auth_wrapper(wrapped_func):
|
|
# pylint: disable=missing-docstring
|
|
wrapped_func = wrapped_func.__func__
|
|
|
|
def _create_social_auth(*args, **kwargs):
|
|
# The entire reason for this monkey-patch is to wrap the create_social_auth call
|
|
# in an atomic transaction. The call can sometime raise an IntegrityError, which is
|
|
# caught and dealt with by python_social_auth - but not inside of an atomic transaction.
|
|
# In Django 1.8, unless the exception is raised in an atomic transaction, the transaction
|
|
# becomes unusable after the IntegrityError exception is raised.
|
|
with transaction.atomic():
|
|
return wrapped_func(*args, **kwargs)
|
|
return classmethod(_create_social_auth)
|
|
|
|
DjangoUserMixin.create_social_auth = create_social_auth_wrapper(DjangoUserMixin.create_social_auth)
|
|
|
|
# Monkey-patch some social auth models' Meta class to squelch Django19 warnings.
|
|
# pylint: disable=protected-access
|
|
UserSocialAuth._meta.app_label = "default"
|
|
Nonce._meta.app_label = "default"
|
|
Association._meta.app_label = "default"
|
|
Code._meta.app_label = "default"
|