from django.db import models
from django.contrib.auth.models import User
from django.contrib.contenttypes.models import ContentType
from django.contrib.contenttypes import generic
from django.utils.translation import ugettext as _
from authentic2.attribute_aggregator.mapping import ATTRIBUTE_MAPPING, \
ATTRIBUTE_NAMESPACES
from authentic2.attribute_aggregator.models import AttributeSource
ATTRIBUTES = [(key, key) \
for key in sorted(ATTRIBUTE_MAPPING.iterkeys())]
ATTRIBUTES_NS = [('Default', 'Default')] \
+ [(ns, ns) for ns in ATTRIBUTE_NAMESPACES]
ATTRIBUTE_VALUE_FORMATS = (
(lasso.SAML2_ATTRIBUTE_NAME_FORMAT_URI, 'SAMLv2 URI'),
(lasso.SAML2_ATTRIBUTE_NAME_FORMAT_BASIC, 'SAMLv2 BASIC'))
def set_user_consent_attributes(user, provider, attributes):
if not user or not provider:
return None
return UserConsentAttributes.objects.get_or_create(user=user,
object_id=provider.id,
content_type=ContentType.objects.get_for_model(provider))
