coin/coin/mixins.py

# -*- coding: utf-8 -*-
from __future__ import unicode_literals

from django.db import transaction
from django.conf import settings


class CoinLdapSyncMixin(object):

    """
    Ce mixin est à utiliser lorsqu'il s'agit de définir un modèle
    à synchroniser avec le LDAP. Le modèle doit définir les méthodes :
    sync_to_ldap et delete_from_ldap qui s'occupent du transfert vers le LDAP.
    L'avantage de ce modèle est que si cette méthode échoue, la sauvegarde en
    base de données échoue à son tour et rien n'est sauvegardé afin de conserver
    la cohérence entre base de donnée et LDAP
    """

    def sync_to_ldap(self, creation, *args, **kwargs):
        raise NotImplementedError('Using CoinLdapSyncModel require '
                                  'sync_to_ldap method being implemented')

    def delete_from_ldap(self, *args, **kwargs):
        raise NotImplementedError('Using CoinLdapSyncModel require '
                                  'delete_from_ldap method being implemented')

    @transaction.atomic
    def save(self, *args, **kwargs):
        # Détermine si on est dans une création ou une mise à jour
        creation = (self.pk == None)

        # Récupère les champs mis à jour si cela est précisé
        update_fields = kwargs[
            'update_fields'] if 'update_fields' in kwargs else None

        # Sauvegarde en base de donnée (mais sans commit, cf decorator)
        super(CoinLdapSyncMixin, self).save(*args, **kwargs)

        # Sauvegarde dans le LDAP
        # Si la sauvegarde LDAP échoue, Rollback la sauvegarde en base, sinon
        # commit
        if settings.LDAP_ACTIVATE:
            try:
                self.sync_to_ldap(
                    creation=creation, update_fields=update_fields)
            except:
                raise

    @transaction.atomic
    def delete(self, *args, **kwargs):
        # Supprime de la base de donnée (mais sans commit, cf decorator)
        super(CoinLdapSyncMixin, self).delete(*args, **kwargs)

        if settings.LDAP_ACTIVATE:
            try:
                self.delete_from_ldap()
            except:
                raise