最新版本

WebAdmin/signals.py

@@ -5,33 +5,63 @@ from .models import User, UserSource
 
 @receiver(user_logged_in)
 def user_logged_in_handler(request, user, **kwargs):
-    print('1111111111111111111111111')
+    print('信号处理开始: user_logged_in_handler 被触发')
+    
+    # 防止信号被重复处理
+    if hasattr(request, '_google_user_logged_in_handled'):
+        print('重复信号，跳过处理')
+        return
+    request._google_user_logged_in_handled = True
+    print('处理信号标志设置完成')
+
     try:
         social_account = SocialAccount.objects.get(user=user, provider='google')
-        print(social_account.extra_data)  # 打印 extra_data 内容以进行调试
-
-        google_id = social_account.extra_data['sub']  # 使用 'sub' 字段
-        email = social_account.extra_data['email']
-        username = social_account.extra_data['name']
+        print(f'找到 social_account，用户ID: {user.id}')
+        google_id = social_account.extra_data.get('sub')
+        email = social_account.extra_data.get('email')
+        username = social_account.extra_data.get('name')
+        print(f'从 social_account 获取的 google_id: {google_id}, email: {email}, username: {username}')
+    
+        if not google_id:
+            # 如果 google_id 为空，不要继续创建用户，记录日志或处理错误
+            print('google_id 为空，终止处理')
+            return
 
+        # 优化：根据 google_id 和 email 双重检查用户是否已存在
         try:
             custom_user = User.objects.get(google_id=google_id)
+            print(f'根据 google_id 找到用户: {custom_user.username}')
         except User.DoesNotExist:
-            custom_user = User.objects.create(
-                username=username,
-                email=email,
-                google_id=google_id,
-                password_hash='',  # Google 登录不需要密码
-            )
-            UserSource.objects.create(user=custom_user, source='Google')
-
-        # 更新登录次数和最后登录IP
+            print(f'没有找到 google_id 为 {google_id} 的用户，检查 email: {email}')
+            # 进一步检查是否有相同的 email 用户存在
+            if User.objects.filter(email=email).exists():
+                custom_user = User.objects.get(email=email)
+                print(f'找到 email 为 {email} 的用户: {custom_user.username}')
+                # 如果用户存在但 google_id 不同，更新 google_id
+                custom_user.google_id = google_id
+                custom_user.save()
+                print(f'更新用户的 google_id 为: {google_id}')
+            else:
+                custom_user = User.objects.create(
+                    username=username,
+                    email=email,
+                    google_id=google_id,
+                    # password_hash='',  # Google 登录不需要密码
+                )
+                print(f'创建新用户: {custom_user.username}，google_id: {google_id}')
+                UserSource.objects.create(user=custom_user, source='Google')
+                print(f'创建 UserSource 记录，来源为 Google')
+                
+        x_forwarded_for = request.META.get('HTTP_X_FORWARDED_FOR')
+        if x_forwarded_for:
+            ip = x_forwarded_for.split(',')[0]  # 可能有多个 IP 地址，取第一个
+        else:
+            ip = request.META.get('REMOTE_ADDR')  # 使用 REMOTE_ADDR 作为备用
+        # 更新登录次数和 IP 地址
         custom_user.login_count += 1
-        custom_user.last_login_ip = request.META.get('REMOTE_ADDR')
+        custom_user.last_login_ip = ip
         custom_user.save()
+        print(f'更新用户信息: 登录次数 {custom_user.login_count}, IP 地址 {custom_user.last_login_ip}')
 
-        # 确保 Django 用户同步
-        user.username = custom_user.username
-        user.save()
     except SocialAccount.DoesNotExist:
-        pass
+        print('未找到与用户关联的 Google social_account')