在social-auth-app-django中通过自定义字段实现社交账户关联

import uuid
from django.contrib.auth.models import AbstractUser
from django.db import models

class Profile(AbstractUser):
    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    # ... 其他字段
    telegram_id = models.BigIntegerField(verbose_name='телеграм ID', unique=True, blank=True, null=True)
    # ...

SOCIAL_AUTH_PIPELINE = (
    'social_core.pipeline.social_auth.social_details',
    'social_core.pipeline.social_auth.social_uid',
    'social_core.pipeline.social_auth.social_user',
    'social_core.pipeline.user.get_username',
    'social_core.pipeline.social_auth.associate_by_email', # 默认通过邮箱关联
    'social_core.pipeline.user.create_user',
    'social_core.pipeline.social_auth.associate_user',
    'social_core.pipeline.social_auth.load_extra_data',
    'social_core.pipeline.user.user_details',
)

from django.contrib.auth import get_user_model
from social_core.exceptions import AuthException

def associate_by_telegram_id(backend, details, user=None, *args, **kwargs):
    """
    自定义管道函数，用于通过Telegram ID关联现有用户。
    """
    # 仅针对Telegram后端执行此逻辑
    if backend.name == 'telegram':
        # 如果用户已经登录或已通过之前的管道步骤关联，则跳过
        if user:
            return None

        # 从kwargs中获取社交平台提供的UID（通常是Telegram ID）
        tgid_str = kwargs.get('uid')
        if not tgid_str:
            return None # 如果没有UID，则无法进行关联

        try:
            tgid = int(tgid_str)
        except (ValueError, TypeError):
            # UID不是有效的整数，可能是数据异常，记录日志或抛出异常
            return None # 或者 raise AuthException(backend, "Invalid Telegram ID format")

        UserModel = get_user_model()
        # 尝试查找与该Telegram ID匹配的用户
        users = list(UserModel.objects.filter(telegram_id=tgid))

        if len(users) == 0:
            # 没有找到匹配的用户，让后续管道步骤（如create_user）处理
            return None
        elif len(users) > 1:
            # 发现多个用户绑定了同一个Telegram ID，这通常是数据异常，应抛出错误
            raise AuthException(
                backend, "The given Telegram ID is associated with multiple accounts"
            )
        else:
            # 找到一个匹配的用户，将其返回以供后续管道步骤关联
            return {"user": users[0], "is_new": False}
    return None # 对于其他后端，不执行任何操作

# settings.py

SOCIAL_AUTH_PIPELINE = (
    'social_core.pipeline.social_auth.social_details',
    'social_core.pipeline.social_auth.social_uid',
    'your_app_name.pipeline.associate_by_telegram_id', # 插入你的自定义管道函数
    'social_core.pipeline.social_auth.associate_by_email',
    'social_core.pipeline.user.get_username', # 注意：get_username可以放在这里或更早，根据你的需求
    'social_core.pipeline.user.create_user',
    'social_core.pipeline.social_auth.associate_user',
    'social_core.pipeline.social_auth.load_extra_data',
    'social_core.pipeline.user.user_details',
)