diff --git a/gather_new_user.py b/gather_new_user.py index 9800563..a907a0b 100644 --- a/gather_new_user.py +++ b/gather_new_user.py @@ -44,6 +44,19 @@ def comp_datetime(x, y): return False +def get_ad_chanllel(gameid): + sql = f"select distinct ad_channel from newuser where gameid={gameid};" + data = mydb.query(sql) + ad_channel = [] + if data: + for line in data: + try: + ad_channel.append(line[0]) + except Exception: + log.info(f"split data error,{line}", exc_info=True) + return ad_channel + + def gather_data(last_time, gameid, channelid, ad_channel): begin = (datetime.datetime.strptime(last_time, "%Y-%m-%d %H:%M:%S") - datetime.timedelta( minutes=TimeDelay)).strftime("%Y-%m-%d %H:%M:%S") @@ -103,8 +116,9 @@ def run(gameid, channelid, ad_channel): def main(): gameid = 2001 channel = 6001 - ad_channel = "" - run(gameid, channel, ad_channel) + ad_channel = get_ad_chanllel(gameid) + for item in ad_channel: + run(gameid, channel, item) if __name__ == "__main__":