SubscribeUserInfo

Kimi.Huang 5 jaren geleden
bovenliggende
commit
887cfbd85a
1 gewijzigde bestanden met toevoegingen van 23 en 20 verwijderingen
  1. 23 20
      commands/management/commands/subscribe.py

+ 23 - 20
commands/management/commands/subscribe.py

@@ -55,25 +55,28 @@ class Command(CompatibilityBaseCommand):
55 55
             close_old_connections()
56 56
 
57 57
             with transaction.atomic():
58
-                unionid, openid = userinfo.get('unionid', ''), userinfo.get('openid', '')
59
-                SubscribeUserInfo.objects.update_or_create(openid=openid, defaults={
60
-                    'unionid': unionid,
61
-                    'nickname': userinfo.get('nickname', ''),
62
-                    'sex': userinfo.get('sex', ''),
63
-                    'headimgurl': userinfo.get('headimgurl', ''),
64
-                    'country': userinfo.get('country', ''),
65
-                    'province': userinfo.get('province', ''),
66
-                    'city': userinfo.get('city', ''),
67
-                    'subscribe': userinfo.get('subscribe', ''),
68
-                    'subscribe_time': userinfo.get('subscribe_time', ''),
69
-                    'subscribe_scene': userinfo.get('subscribe_scene', ''),
70
-                    'groupid': userinfo.get('groupid', ''),
71
-                    'tagid_list': userinfo.get('tagid_list', ''),
72
-                    'qr_scene': userinfo.get('qr_scene', ''),
73
-                    'qr_scene_str': userinfo.get('qr_scene_str', ''),
74
-                    'language': userinfo.get('language', ''),
75
-                    'remark': userinfo.get('remark', ''),
76
-                })
77
-                UserInfo.objects.filter(unionid=unionid).update(openid=openid)
58
+                subscribe, unionid, openid = userinfo.get('subscribe', ''), userinfo.get('unionid', ''), userinfo.get('openid', '')
59
+                if subscribe:
60
+                    SubscribeUserInfo.objects.update_or_create(openid=openid, defaults={
61
+                        'unionid': unionid,
62
+                        'nickname': userinfo.get('nickname', ''),
63
+                        'sex': userinfo.get('sex', ''),
64
+                        'headimgurl': userinfo.get('headimgurl', ''),
65
+                        'country': userinfo.get('country', ''),
66
+                        'province': userinfo.get('province', ''),
67
+                        'city': userinfo.get('city', ''),
68
+                        'subscribe': userinfo.get('subscribe', ''),
69
+                        'subscribe_time': userinfo.get('subscribe_time', ''),
70
+                        'subscribe_scene': userinfo.get('subscribe_scene', ''),
71
+                        'groupid': userinfo.get('groupid', ''),
72
+                        'tagid_list': userinfo.get('tagid_list', ''),
73
+                        'qr_scene': userinfo.get('qr_scene', ''),
74
+                        'qr_scene_str': userinfo.get('qr_scene_str', ''),
75
+                        'language': userinfo.get('language', ''),
76
+                        'remark': userinfo.get('remark', ''),
77
+                    })
78
+                    UserInfo.objects.filter(unionid=unionid).update(openid=openid)
79
+                else:
80
+                    SubscribeUserInfo.objects.filter(openid=openid).update(subscribe=subscribe)
78 81
 
79 82
             close_old_connections()