ADD:新用户添加问题

wechat_action/login_ad.py

@@ -87,7 +87,8 @@ class LogIn:
             sql_session.execute(update_res)
             sql_session.commit()
         else:
-            wechat_cookies_info = {'user_id': log_ad.user_id, 'cookies': wechat_cookies, 'scan_action': 'done'}
+            wechat_cookies_info = {'user_id': log_ad.user_id, 'cookies': wechat_cookies, 'wechat_id': wechat_id,
+                                   'scan_action': 'done'}
             wechat_insert = sql_tools.save_wechat_cookies(wechat_cookies_info=wechat_cookies_info,
                                                           table_wechat_cookies=log_ad.wechat_cookies_table)
             sql_session.execute(wechat_insert)
@@ -127,7 +128,7 @@ class LogIn:
             e_one.scroll(10000,100000);
             ''')
         time.sleep(0.1)
-        # TODO:有时间去除掉这个time.sleep
+        #TODO:去除掉time.sleep
         time.sleep(5)
         WebDriverWait(self.driver, 100).until(
             lambda driver: len([self.driver.find_elements_by_link_text('广告投放')]) == 1)

wechat_action/sql_tools.py

@@ -16,7 +16,8 @@ def save_wechat_cookies(wechat_cookies_info, table_wechat_cookies):
     insert_layout = insert_wechat_cookies.values \
         (user_id=wechat_cookies_info['user_id'],
          cookies=wechat_cookies_info['cookies'],
-         scan_action=wechat_cookies_info['scan_action']
+         scan_action=wechat_cookies_info['scan_action'],
+         wechat_id=wechat_cookies_info['wechat_id']
          )
     return insert_layout