From 9eed7d72f090cd9c37e5d24b550e917da0347555 Mon Sep 17 00:00:00 2001
From: pengtao <pengtao@kingsome.cn>
Date: Tue, 3 Sep 2019 13:39:56 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=9E=E6=97=B6=E6=96=B0=E5=A2=9E=E4=BA=BA?=
 =?UTF-8?q?=E6=95=B0=E8=AE=A1=E7=AE=97?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 gather_new_user.py | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

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__":