Как прокинуть данные при Oauth2 авторизации с помощью django-allauth?

Question

[Максим Тарабрин]

Друзья, у меня имеется авторизация по oauth2 с помощью django-allauth. Написан кастомный provider. Не пойму как прокинуть данные при авторизации в модель User?

provider

class ProjectProvider(OAuth2Provider):
    id = 'project'
    name = 'Project'
    account_class = ProjectAccount

    def extract_uid(self, data):
        try:
            return data['user_id']
        except KeyError:
            raise AuthExtractError('Error on extract user_id from response userinfo')

    def extract_common_fields(self, data):
        try:
            return dict(
                email=data['email'],
                username=data['username'],
                user_id=data['user_id'], # Не работает...
            )
        except KeyError:
            raise AuthExtractError('Error on extract user_id from response userinfo')

    def get_default_scope(self):
        scope = ["read write"]
        return scope

adapter

class ProjectAdapter(OAuth2Adapter):
    provider_id = ProjectProvider.id
    provider_base_url = settings.PROVIDER_AUTH_HOST

    @property
    def access_token_url(self):
        return '{0}/oauth2/access_token'.format(self.provider_base_url)

    @property
    def authorize_url(self):
        return '{0}/oauth2/authorize'.format(self.provider_base_url)

    @property
    def profile_url(self):
        return '{0}/api/userinfo'.format(self.provider_base_url)

    @backoff.on_exception(backoff.expo,
                          requests.exceptions.RequestException,
                          max_tries=5,
                          )
    def complete_login(self, request, app, token, response):
        response_profile = requests.get(self.profile_url, params={
            'access_token': token.token
        })
        if response_profile.status_code == 200:
            return self.get_provider().sociallogin_from_response(
                request,
                response_profile.json()
            )
        else:
            raise AuthLoginError('Error on request to api userinfo')


oauth2_login = OAuth2LoginView.adapter_view(ProjectAdapter)
oauth2_callback = OAuth2CallbackView.adapter_view(ProjectAdapter)

По результату я хочу, чтобы при авторизации через сторонний сервис я мог прокинуть данные в модель User и уже использовать эти данные.

Comments

[Anton Kuzmichev]

Вы всё по документации сделали? Класс провайдера зарегистрировали?

user_id=data['user_id'], # Не работает...

Не работает, простите, что? Получение значения ключа user_id из словаря data? Значит этих данных там нет, а почему их там нет - вопрос уже выходящий за рамки кода, который вы предоставили.
В массиве data есть нужная информация? Точно ли в этом методе надо id вытаскивать, для него вроде отдельный метод есть, у вас он выше.

[Максим Тарабрин]

Anton Kuzmichev, я авторизуюсь с помощью другого сервиса и хочу id из того сервиса сохранить в модель User отдельным полем user_id(пример). В data имеется user_id(это int).

[Максим Тарабрин]

Максим Тарабрин, то есть при авторизации сделать user.user_id = data['user_id'].

[Максим Тарабрин]

Anton Kuzmichev, Класс провайдера зарегистрирован. В файле provider.py: provider_classes = [ProjectProvider]

[Anton Kuzmichev]

Максим Тарабрин,

Не работает, простите, что? Получение значения ключа user_id из словаря data? Значит этих данных там нет [..]

Что у вас в data?
метод extract_uid() отрабатывает норм?
Вы поотлаживайте, посмотрите что приходит в данных-то.

[Максим Тарабрин]

Anton Kuzmichev, я же описал. Мне приходит словарь, в котором имеется user_id. Он имеет тип int. В модели User описано поле user_id с типом: user_id = models.BigIntegerField(null=True). С точки зрения данных всё Ок, но сохранение данных в модели User не отрабатывает. Логин в системе заводится, сессия поднимается. В модель User попадает email и username(он берется из email перед собакой).

[Максим Тарабрин]

Anton Kuzmichev, extract_uid мне не нужен, он отрабатывает верно и сохраняется в модель SocialAccount в uid. А мне нужно сохранить один из параметров response profile data сохранить в модель User при авторизации.

[Anton Kuzmichev]

Максим Тарабрин, вот этот метод адаптера DefaultSocialAccountAdapter.populate_user делает объект пользователя из данных от провайдера. Но у вас Oauth2, там адаптер не отнаследован от базового адаптера.. Поизучайте исходники, что происходит в данными из extract_common_fields, где происходит обновление этих данных в инстанс модели User.
Кстати, у вас видимо кастомная модель пользователя - она же зарегена в settings.py?) Или у вас всё работает норм, кроме сохранения значения этого поля user_id?

[Максим Тарабрин]

Anton Kuzmichev, Да, кастомная модель User. Прописана в settings. Только сейчас я решил переписать метод populate_user, но он записывает user_id при первом входе пользователя, когда создается User. А потом не обновляет. Хотелось бы, чтобы обновлял(

[Максим Тарабрин]

Например к нам приходит nickname с того сервиса и нам нужно каждый раз его переписывать в модели User.

[Anton Kuzmichev]

Максим Тарабрин, изучайте исходники, ищите нужные методы. Очень вероятно, что есть что то вроде хуков. Задача типичная, в общем-то, вряд ли вы первый ее решаете.

[Максим Тарабрин]

Anton Kuzmichev, искал, но пока что то не найду) Буду благодарен, если ткнёшь)

[Anton Kuzmichev]

Максим Тарабрин, что насчёт https://django-allauth.readthedocs.io/en/latest/ad... ?