Files
edx-platform/common/djangoapps/third_party_auth/api/utils.py
Matt Hughes acce8baca4 Add management command for retroactively adding sso IdV for IdP
We sometimes update preexisting SAML SSO providers to configure them
to automatically create SSO identity verification (IdV) records when a
learner links an account via that provider. Turning that configuration
from off to on does make it such that when learners log back in via
their linked account, a new IdV record will be created for them. But
it's possible we'd want this process to happen more automatically and
seamlessly, for which this management command will be helpful.

Note that this does not help with removing SSO verification records
for a provider for which this configuration has been turned off.

JIRA:EDUCATOR-4947
2020-03-17 16:04:24 -04:00

31 lines
1.1 KiB
Python

"""
Shareable utilities for third party auth api functions
"""
def filter_user_social_auth_queryset_by_provider(query_set, provider):
"""
Filter a query set by the given TPA provider
Params:
query_set: QuerySet[UserSocialAuth]
provider: common.djangoapps.third_party_auth.models.ProviderConfig
Returns:
QuerySet[UserSocialAuth]
"""
# Note: When using multi-IdP backend, the provider column isn't
# enough to identify a specific backend
filtered_query_set = query_set.filter(provider=provider.backend_name)
# Test if the current provider has a slug which it appends to
# uids; these can be used to identify the backend more
# specifically than the provider's backend
fake_uid = 'uid'
uid = provider.get_social_auth_uid(fake_uid)
if uid != fake_uid:
# if yes, we add a filter for the slug on uid column
# carve off the fake_uid from the end, so we get just the prepended slug
filtered_query_set = filtered_query_set.filter(uid__startswith=uid[:-len(fake_uid)])
return filtered_query_set