import pymysql import random import traceback import logging from vest.request.auto_request import login, time_convs, Urge from vest.request.auto_request import host, port, passwd, user, db from libs.error import logging_exception def get_data(numtime1, numtime2): try: pc = pymysql.connect(host=host, user=user, passwd=passwd, db=db, port=3306) cursor = pc.cursor() cursor.execute( "SELECT distinct(user_id) FROM topic WHERE is_online=1 and create_time > '%s' and create_time < '%s'" % ( numtime1, numtime2)) data = cursor.fetchall() user_id = list(data) topic_id_list = [] for i in user_id and user_id: cursor.execute( "select user_id,is_shadow from user_extra where user_id =" + str( i[0]) + " and is_online =1 and is_deleted =0") data = cursor.fetchall() user_id = list(data) if (user_id and user_id[0][1] == 0) or len(user_id) == 0: topic_id_list.append(i) logging.info("Database version : %s " % topic_id_list) pc.close() return topic_id_list except: logging.error("catch exception,get_data:%s" % traceback.format_exc()) return None def auto_urge2(): try: numtime1, numtime2 = time_convs(8, 0) logging.info("get numtime1:%s" % numtime1) logging.info("get numtime2:%s" % numtime2) user_id = get_data(numtime1, numtime2) for j in user_id and user_id: urge_num = random.randint(0, 1) if urge_num == 1: id = int(j[0]) cookies = login() if cookies is not None: Urge(cookies, id) except: logging_exception() logging.error("catch exception,main:%s" % traceback.format_exc())