auto_urge2.py 2 KB
import requests
import time
import datetime
import pymysql
import random
import traceback
from log_settings import *
import logging
from auto_request import login, time_convs, Urge
from auto_request import host, port, passwd, user, db
import threading


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 from user_extra where user_id =" + str(i[0]) + " and is_shadow =0 ")
            data = cursor.fetchall()
            user_id = list(data)
            if user_id:
                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


if __name__ == "__main__":

    try:
        time_list = [1, 2, 3, 4, 5, 6, 7]
        for i in time_list:
            numtime1, numtime2 = time_convs(i + 1, i - 1)
            logging.info("get numtime1:%s" % numtime1)
            user_id = get_data(numtime1, numtime2)
            urge_num = 1
            if urge_num == 1:
                for i in range(urge_num):
                    try:
                        for j in user_id and user_id:
                            id = int(j[0])
                            # time.sleep(random.randint(10, 50))
                            cookies = login()
                            if cookies is not None:
                                Urge(cookies, id)
                    except:
                        pass
    except:
        logging.error("catch exception,main:%s" % traceback.format_exc())