tbAutoBuy

仅供学习参考

from selenium import webdriver
import datetime
import time

# ====  标识登录状态、重试次数 ====
MAX_LOGIN_RETRY_TIMES = 2

current_retry_login_times = 0
login_success = False


def __login_operates():
    driver.get("https://www.taobao.com")
    try:
        if driver.find_element_by_link_text("亲，请登录"):
            print("没登录，开始点击登录按钮...")
            driver.find_element_by_link_text("亲，请登录").click()
            print("请使用手机淘宝扫描屏幕上的二维码进行登录...")
            time.sleep(15)
    except:
        print("已登录，开始执行跳转...")
        global login_success
        global current_retry_login_times
        login_success = True
        current_retry_login_times = 0


def login():
    print("开始尝试登录...")
    __login_operates()
    global current_retry_login_times
    while current_retry_login_times < MAX_LOGIN_RETRY_TIMES:
        current_retry_login_times = current_retry_login_times + 1
        print("当前尝试登录次数：" + str(current_retry_login_times))
        __login_operates()
        if login_success:
            print("登录成功")
            break
        else:
            print("等待登录中...")

    if not login_success:
        print("规定时间内没有扫码登录淘宝成功，执行失败，退出脚本!!!")
        exit(0)

    # time.sleep(3)
    now = datetime.datetime.now()
    print('login success:', now.strftime('%Y-%m-%d %H:%M:%S'))


def __refresh_keep_alive():
    # 重新加载购物车页面，定时操作，防止长时间不操作退出登录
    driver.get("https://cart.taobao.com/cart.htm")
    print("刷新购物车界面，防止登录超时...")
    time.sleep(60)


def keep_login_and_wait():
    print("当前距离抢购时间点还有较长时间，开始定时刷新防止登录超时...")
    while True:
        current_time = datetime.datetime.now()
        if (buy_time_object - current_time).seconds > 180:
            __refresh_keep_alive()
        else:
            print("抢购时间点将近，停止自动刷新，准备进入抢购阶段...")
            break


def buy():
    # 打开购物车
    driver.get("https://cart.taobao.com/cart.htm")
    time.sleep(1)

    # 点击购物车里全选按钮
    if driver.find_element_by_id("J_SelectAll1"):
        driver.find_element_by_id("J_SelectAll1").click()
        print("已经选中购物车中全部商品 ...")

    submit_succ = False
    retry_submit_times = 0
    while True:
        now = datetime.datetime.now()
        if now >= buy_time_object:
            print("到达抢购时间，开始执行抢购...尝试次数：" + str(retry_submit_times))
            if submit_succ:
                print("订单已经提交成功，无需继续抢购...")
                break
            if retry_submit_times > 50:
                print("重试抢购次数达到上限，放弃重试...")
                break

            retry_submit_times = retry_submit_times + 1

            try:
                # 点击结算按钮
                if driver.find_element_by_id("J_Go"):
                    driver.find_element_by_id("J_Go").click()
                    print("已经点击结算按钮...")
                    click_submit_times = 0
                    while True:
                        try:
                            if click_submit_times < 10:
                                driver.find_element_by_link_text('提交订单').click()
                                print("已经点击提交订单按钮")
                                submit_succ = True
                                break
                            else:
                                print("提交订单失败...")
                        except Exception as ee:
                            # print(ee)
                            print("没发现提交订单按钮，可能页面还没加载出来，重试...")
                            click_submit_times = click_submit_times + 1
                            time.sleep(0.1)
            except Exception as e:
                print(e)
                print("不好，挂了，提交订单失败了...")

        time.sleep(0.5)


if __name__ == '__main__':
    # ==== 设定抢购时间 （修改此处，指定抢购时间点）====
    cur_time = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
    BUY_TIME = input(f"请输入抢购时间，格式如 {cur_time} :\n")
    buy_time_object = datetime.datetime.strptime(BUY_TIME, '%Y-%m-%d %H:%M:%S')
    print(f"已设置抢购时间为：{buy_time_object}")
    now_time = datetime.datetime.now()
    if now_time > buy_time_object:
        print("当前已过抢购时间，请确认抢购时间是否填错...")
        exit(0)

    print("正在打开chrome浏览器...")
    # 让浏览器不要显示当前受自动化测试工具控制的提醒
    option = webdriver.ChromeOptions()
    option.add_argument('disable-infobars')
    #windows
    driver = webdriver.Chrome(executable_path='D:\Programs\Python\Python36\chromedriver.exe', chrome_options=option)
    #linux
   # driver = webdriver.Chrome(executable_path='assets/chromedriver', chrome_options=option)
    driver.maximize_window()
    print("chrome浏览器已经打开...")

    login()
    keep_login_and_wait()
    buy()