From 779deff9ccb19bcca970af2b61f68181d44aa2f1 Mon Sep 17 00:00:00 2001 From: wuhao <15392746632@qq.com> Date: Wed, 16 Sep 2020 14:11:34 +0800 Subject: [PATCH] =?UTF-8?q?=E5=BB=BA=E7=AB=8B=E5=8F=91=E8=A1=8C=E6=B8=A0?= =?UTF-8?q?=E9=81=93=E8=A1=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- update_data/update_user.py | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/update_data/update_user.py b/update_data/update_user.py index 30760fa..5233ed6 100644 --- a/update_data/update_user.py +++ b/update_data/update_user.py @@ -13,6 +13,7 @@ class LoadUserData(LoadData): self.s_db_df = kwargs.get('s_db_df') self.gm_mysql = None self.channel_set = set() + self.owner_channel_set = set() async def get_data(self) -> list: """ @@ -53,7 +54,8 @@ class LoadUserData(LoadData): for user in data_list: user['lasttime'] = user['logintime'] user['vip'] = user.get('vip', 0) - self.channel_set.add((self.game_name, user.get('ext_channel', 'unkonwn'), user.get('ext_owner', 'unkonwn'))) + self.channel_set.add((self.game_name, user.get('ext_channel', 'unkonwn'))) + self.owner_channel_set.add((self.game_name, user.get('ext_channel', 'unkonwn'), user.get('ext_owner', 'unkonwn'))) user_info_data.extend(data_list) logger.info(f'db:{db}') logger.info(f'区服id:{sid}') @@ -130,13 +132,22 @@ class LoadUserData(LoadData): await conn.commit() # 记录渠道 - update_channel_sql = """insert into `gm_game_channel_list` (game,channel,owner) values (%s,%s,%s) + update_channel_sql = """insert into `gm_game_channel_list` (game,channel) values (%s,%s) + on duplicate key update + game=values(game), + channel=values(channel) + """ + await cur.executemany(update_channel_sql, self.channel_set) + await conn.commit() + + # 记录发行和渠道 + update_owner_channel_sql = """insert into `gm_owner_channel` (game,channel,owner) values (%s,%s,%s) on duplicate key update game=values(game), channel=values(channel), owner=values(owner) """ - await cur.executemany(update_channel_sql, self.channel_set) + await cur.executemany(update_owner_channel_sql, self.owner_channel_set) await conn.commit() # 记录登录