0001-Model-Data-migrations-of-LibertyServiceProvider-to-L.patch

Josué Kouka, 21 décembre 2015 22:42

Voir les différences: en ligne côte à côte

Subject: [PATCH] Model/Data migrations of LibertyServiceProvider to
 LibertyProvider (#8826)

 src/authentic2/idp/saml/backend.py                 |  4 +-
 src/authentic2/idp/saml/saml2_endpoints.py         |  6 +--
 src/authentic2/saml/admin.py                       | 20 +++++----
 src/authentic2/saml/common.py                      | 17 +++-----
 src/authentic2/saml/forms.py                       | 13 ++----
 .../saml/management/commands/sync-metadata.py      |  8 +---
 .../saml/migrations/0017_auto_20151208_1537.py     | 45 +++++++++++++++++++
 .../saml/migrations/0018_auto_20151208_1542.py     | 28 ++++++++++++
 .../saml/migrations/0019_libertyfederation_nsp.py  | 20 +++++++++
 .../saml/migrations/0020_auto_20151221_1108.py     | 21 +++++++++
 .../saml/migrations/0021_auto_20151221_1149.py     | 35 +++++++++++++++
 src/authentic2/saml/models.py                      | 50 +++++++---------------
 tests/test_idp_saml2.py                            |  4 --
 13 files changed, 192 insertions(+), 79 deletions(-)
 create mode 100644 src/authentic2/saml/migrations/0017_auto_20151208_1537.py
 create mode 100644 src/authentic2/saml/migrations/0018_auto_20151208_1542.py
 create mode 100644 src/authentic2/saml/migrations/0019_libertyfederation_nsp.py
 create mode 100644 src/authentic2/saml/migrations/0020_auto_20151221_1108.py
 create mode 100644 src/authentic2/saml/migrations/0021_auto_20151221_1149.py

             self.logger = logging.getLogger(__name__)
         def service_list(self, request):
             q = models.LibertyServiceProvider.objects.filter(enabled = True) \
             q = models.LibertyProvider.objects.filter(enabled = True) \
                     .select_related()
             ls = []
             sessions = models.LibertySession.objects.filter(
-...
                     sp_options_policy__idp_initiated_sso=True))
                 queries.append(q.filter(sp_options_policy__enabled=True,
                     sp_options_policy__accept_slo=True,
                     liberty_provider__entity_id__in=sessions_eids))
                     entity_id__in=sessions_eids))
                 if default_policy and default_policy.idp_initiated_sso:
                     queries.append(q.filter(sp_options_policy__isnull=True))
                 if default_policy and default_policy.accept_slo:

         LibertySession, LibertyFederation,
         nameid2kwargs, saml2_urn_to_nidformat,
         nidformat_to_saml2_urn, save_key_values, get_and_delete_key_values,
         LibertyProvider, LibertyServiceProvider, SAMLAttribute, NAME_ID_FORMATS)
         LibertyProvider, SAMLAttribute, NAME_ID_FORMATS)
     from authentic2.saml.common import redirect_next, asynchronous_bindings, \
         soap_bindings, load_provider, get_saml2_request_message, \
         error_page, set_saml2_response_responder_status_code, \
-...
                 kwargs['name_id_qualifier'] = AUTHENTIC_SAME_ID_SENTINEL
             if kwargs.get('name_id_sp_name_qualifier') == login.remoteProviderId:
                 kwargs['name_id_sp_name_qualifier'] = AUTHENTIC_SAME_ID_SENTINEL
             service_provider = LibertyServiceProvider.objects \
                     .get(liberty_provider__entity_id=login.remoteProviderId)
             service_provider = LibertyProvider.objects \
                     .get(entity_id=login.remoteProviderId)
             federation, new = LibertyFederation.objects.get_or_create(
                     sp=service_provider,
                     user=request.user, **kwargs)

     except ImportError:
         from django.contrib.contenttypes.generic import GenericTabularInline
     from authentic2.saml.models import (LibertyProvider, LibertyServiceProvider,
                                         SPOptionsIdPPolicy, LibertyFederation,
                                         KeyValue, LibertySession, SAMLAttribute)
     from authentic2.saml.models import (LibertyProvider, SPOptionsIdPPolicy,
                                         LibertyFederation, KeyValue,
                                         LibertySession, SAMLAttribute)
     from authentic2.decorators import to_iter
     from authentic2.attributes_ng.engine import get_attribute_names
-...
     logger = logging.getLogger(__name__)
     class LibertyServiceProviderInline(admin.StackedInline):
         model = LibertyServiceProvider
     #class LibertyServiceProviderInline(admin.StackedInline):
     #    model = LibertyServiceProvider
     class TextAndFileWidget(forms.widgets.MultiWidget):
         def __init__(self, attrs=None):
-...
                 (_('Metadata files'), {
                     'fields': ('metadata_url', 'metadata', 'public_key', 'ssl_certificate', 'ca_cert_chain')
                 }),
                 (_('SAML service provider'), {
                     'fields': ('enabled', 'enable_following_sp_options_policy', 'sp_options_policy',
                         'users_can_manage_federations')
                 }),
+        )
         inlines = [
                 LibertyServiceProviderInline,
                 #LibertyServiceProviderInline,
                 SAMLAttributeInlineAdmin,
+        ]
         actions = [ update_metadata ]
         prepopulated_fields = {'slug': ('name',)}
         list_filter = (
                 'service_provider__sp_options_policy',
                 'service_provider__enabled',
                 'sp_options_policy',
                 'enabled',
+        )
         def get_urls(self):

     from django.core.exceptions import ValidationError
     from authentic2.saml.models import (LibertyFederation, LibertyProvider,
                                         LibertyServiceProvider, SPOptionsIdPPolicy)
                                         SPOptionsIdPPolicy)
     from authentic2.saml import models
     from authentic2.saml import saml2utils
-...
             return None
         p.save()
         logger.debug('%s saved', p)
         s = LibertyServiceProvider(liberty_provider=p, enabled=True)
         s.save()
         return p
-...
                     return False
             else:
                 return False
         try:
             service_provider = liberty_provider.service_provider
         except LibertyServiceProvider.DoesNotExist:
             return False
         if not service_provider.enabled:
         if not liberty_provider.enabled:
             return False
         if server:
             server.addProviderFromBuffer(lasso.PROVIDER_ROLE_SP,
-...
         policy = get_sp_options_policy_all()
         if policy:
             return policy
         if provider.service_provider.enable_following_sp_options_policy:
             policy = provider.service_provider.sp_options_policy
         if provider.enable_following_sp_options_policy:
             policy = provider.sp_options_policy
             if policy and policy.enabled:
                 return provider.service_provider.sp_options_policy
                 return provider.sp_options_policy
         return get_sp_options_policy_default()

     from django.core.exceptions import ValidationError
     from django.utils.translation import ugettext_lazy as _
     from .models import LibertyProvider, LibertyServiceProvider
     from .models import LibertyProvider
     from authentic2.a2_rbac.utils import get_default_ou
-...
             url = cleaned_data.get('url')
             ou = cleaned_data.get('ou')
             self.instance = None
             self.childs = []
             if name and slug and url:
                 try:
                     content = urllib2.urlopen(url).read().decode('utf-8')
-...
                         slug=slug, metadata=content, metadata_url=url, ou=ou)
                     liberty_provider.full_clean(exclude=
                             ('entity_id', 'protocol_conformance'))
                     self.childs.append(LibertyServiceProvider(
                         liberty_provider=liberty_provider,
                         enabled=True))
                 except ValidationError, e:
                     raise
                 except Exception, e:
                     raise
                     raise ValidationError('unsupported error: %s' % e)
                 self.instance = liberty_provider
             return cleaned_data
         def save(self):
             if not self.instance is None:
                 self.instance.save()
                 for child in self.childs:
                     child.liberty_provider = self.instance
                     child.save()
             self.instance.save()
             return self.instance

             provider.save()
             options['count'] = options.get('count', 0) + 1
             if sp:
                 service_provider, created = LibertyServiceProvider.objects.get_or_create(
                         liberty_provider=provider,
                         defaults={'enabled': not options['create-disabled']})
                 if sp_policy:
                     service_provider.sp_options_policy = sp_policy
                 service_provider.save()
                 pks = []
                 if options['load_attribute_consuming_service']:
                     load_acs(tree, provider, pks, verbosity)
-...
                         SAMLAttribute.objects.for_generic_object(provider).exclude(pk__in=pks).delete()
     class Command(BaseCommand):
         '''Load SAMLv2 metadata file into the LibertyProvider, LibertyServiceProvider
         '''Load SAMLv2 metadata file into the LibertyProvider,
         and LibertyIdentityProvider files'''
         can_import_django_settings = True
         output_transaction = True

     # -*- coding: utf-8 -*-
     from __future__ import unicode_literals
     from django.db import models, migrations
     import django.db.models.deletion
     class Migration(migrations.Migration):
         dependencies = [
             ('saml', '0016_auto_20150915_2041'),
+        ]
         operations = [
             migrations.AddField(
                 model_name='libertyprovider',
                 name='enable_following_sp_options_policy',
                 field=models.BooleanField(default=False, verbose_name='The following options policy will apply except if a policy for all service provider is defined.'),
                 preserve_default=True,
             ),
             migrations.AddField(
                 model_name='libertyprovider',
                 name='enabled',
                 field=models.BooleanField(default=False, db_index=True, verbose_name='Enabled'),
                 preserve_default=True,
             ),
             migrations.AddField(
                 model_name='libertyprovider',
                 name='sp_options_policy',
                 field=models.ForeignKey(related_name='sp_options_policy', on_delete=django.db.models.deletion.SET_NULL, verbose_name='service provider options policy', blank=True, to='saml.SPOptionsIdPPolicy', null=True),
                 preserve_default=True,
             ),
             migrations.AddField(
                 model_name='libertyprovider',
                 name='users_can_manage_federations',
                 field=models.BooleanField(default=True, db_index=True, verbose_name='users can manage federation'),
                 preserve_default=True,
             ),
             migrations.AlterField(
                 model_name='libertyserviceprovider',
                 name='sp_options_policy',
                 field=models.ForeignKey(related_name='old_isp_options_policy', on_delete=django.db.models.deletion.SET_NULL, verbose_name='service provider options policy', blank=True, to='saml.SPOptionsIdPPolicy', null=True),
                 preserve_default=True,
             ),
+        ]

     # -*- coding: utf-8 -*-
     from __future__ import unicode_literals
     from django.db import models, migrations
     def liberty_service_provider_data_to_liberty_provider(apps, schema_editor):
         LibertyProvider = apps.get_model('saml','LibertyProvider')
         LibertyServiceProvider = apps.get_model('saml','LibertyServiceProvider')
         for lsp in LibertyServiceProvider.objects.all():
             lp = lsp.liberty_provider
             lp.enabled = lsp.enabled
             lp.enable_following_sp_options_policy = lsp.enable_following_sp_options_policy
             lp.sp_options_policy = lsp.sp_options_policy
             lp.users_can_manage_federations = lsp.users_can_manage_federations
             lp.save()
     class Migration(migrations.Migration):
         dependencies = [
             ('saml', '0017_auto_20151208_1537'),
+        ]
         operations = [
                 migrations.RunPython(liberty_service_provider_data_to_liberty_provider),
+        ]

     # -*- coding: utf-8 -*-
     from __future__ import unicode_literals
     from django.db import models, migrations
     class Migration(migrations.Migration):
         dependencies = [
             ('saml', '0018_auto_20151208_1542'),
+        ]
         operations = [
             migrations.AddField(
                 model_name='libertyfederation',
                 name='nsp',
                 field=models.ForeignKey(blank=True, to='saml.LibertyProvider', null=True),
                 preserve_default=True,
             ),
+        ]

     # -*- coding: utf-8 -*-
     from __future__ import unicode_literals
     from django.db import models, migrations
     def liberty_federation_sp_to_nsp(apps, schema_editor):
         LibertyFederation = apps.get_model('saml','LibertyFederation')
         for liberty_federation in LibertyFederation.objects.all():
             liberty_federation.nsp = liberty_federation.sp
             liberty_federation.save()
     class Migration(migrations.Migration):
         dependencies = [
             ('saml', '0019_libertyfederation_nsp'),
+        ]
         operations = [
+        ]

     # -*- coding: utf-8 -*-
     from __future__ import unicode_literals
     from django.db import models, migrations
     class Migration(migrations.Migration):
         dependencies = [
             ('saml', '0020_auto_20151221_1108'),
+        ]
         operations = [
             migrations.RemoveField(
                 model_name='libertyserviceprovider',
                 name='liberty_provider',
             ),
             migrations.RemoveField(
                 model_name='libertyserviceprovider',
                 name='sp_options_policy',
             ),
             migrations.RemoveField(
                 model_name='libertyfederation',
                 name='nsp',
             ),
             migrations.AlterField(
                 model_name='libertyfederation',
                 name='sp',
                 field=models.ForeignKey(blank=True, to='saml.LibertyProvider', null=True),
                 preserve_default=True,
             ),
             migrations.DeleteModel(
                 name='LibertyServiceProvider',
             ),
+        ]

         ssl_certificate = models.TextField(blank=True)
         ca_cert_chain = models.TextField(blank=True)
         federation_source = models.CharField(max_length=64, blank=True, null=True)
         enabled = models.BooleanField(verbose_name = _('Enabled'),
                 default=False, db_index=True)
         enable_following_sp_options_policy = models.BooleanField(verbose_name = \
             _('The following options policy will apply except if a policy for all service provider is defined.'),
             default=False)
         sp_options_policy = models.ForeignKey(SPOptionsIdPPolicy,
                 related_name="sp_options_policy",
                 verbose_name=_('service provider options policy'), blank=True,
                 null=True,
                 on_delete=models.SET_NULL)
         users_can_manage_federations = models.BooleanField(
                 verbose_name=_('users can manage federation'),
                 default=True,
                 blank=True,
                 db_index=True)
         attributes = GenericRelation(SAMLAttribute)
-...
         except ObjectDoesNotExist:
             raise RuntimeError('Default %s is missing' % model)
     # TODO: The IdP must look to the preferred binding order for sso in the SP metadata (AssertionConsumerService)
     # expect if the protocol for response is defined in the request (ProtocolBinding attribute)
     class LibertyServiceProvider(models.Model):
         liberty_provider = models.OneToOneField(LibertyProvider,
                 primary_key = True, related_name = 'service_provider')
         enabled = models.BooleanField(verbose_name = _('Enabled'),
                 default=False, db_index=True)
         enable_following_sp_options_policy = models.BooleanField(verbose_name = \
             _('The following options policy will apply except if a policy for all service provider is defined.'),
             default=False)
         sp_options_policy = models.ForeignKey(SPOptionsIdPPolicy,
                 related_name="sp_options_policy",
                 verbose_name=_('service provider options policy'), blank=True,
                 null=True,
                 on_delete=models.SET_NULL)
         users_can_manage_federations = models.BooleanField(
                 verbose_name=_('users can manage federation'),
                 default=True,
                 blank=True,
                 db_index=True)
         objects = managers.GetByLibertyProviderManager()
         def natural_key(self):
             return (self.liberty_provider.slug,)
         def __unicode__(self):
             return unicode(self.liberty_provider)
         class Meta:
             verbose_name = _('SAML service provider')
             verbose_name_plural = _('SAML service providers')
     LIBERTY_SESSION_DUMP_KIND_SP = 0
     LIBERTY_SESSION_DUMP_KIND_IDP = 1
-...
            it IdP or SP"""
         user = models.ForeignKey(settings.AUTH_USER_MODEL, null=True, blank=True,
                 on_delete=models.SET_NULL)
         sp = models.ForeignKey('LibertyServiceProvider', null=True, blank=True)
         sp = models.ForeignKey('LibertyProvider', null=True, blank=True)
         name_id_format = models.CharField(max_length = 100,
                 verbose_name = "NameIDFormat", blank=True, null=True)
         name_id_content = models.CharField(max_length = 100,

                 metadata=sp_meta)
             self.liberty_provider.clean()
             self.liberty_provider.save()
             self.liberty_service_provider = saml_models.LibertyServiceProvider \
                 .objects.create(
                     liberty_provider=self.liberty_provider,
                     enabled=True)
             self.default_sp_options_idp_policy = saml_models.SPOptionsIdPPolicy \
                 .objects.create(
                     name='Default',
+    -

Projet

Général

Profil

Produits Entr'ouvert » Authentic 2

0001-Model-Data-migrations-of-LibertyServiceProvider-to-L.patch