# Selenium 常用于網頁的自動化測試，在此我們借助它自動化淘寶下單的過程
!pip install selenium
!which python

from selenium import webdriver
import datetime
import time

now = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S.%f')
print(now)

def login():
    # 打開淘寶首頁，通過掃碼登錄
    browser.get("https://www.taobao.com")
    if browser.find_element_by_link_text("請登錄"):
        browser.find_element_by_link_text("請登錄").click()
        print(f"請盡快掃碼登錄")
    time.sleep(10)

def picking():
    # 打開購物車列表頁面
    browser.get("https://cart.taobao.com/cart.htm")
    time.sleep(3)
    # method = 0全選購物車
    method = 0
    if method == 0:
        while True:
            try:
                if browser.find_element_by_id("J_SelectAll1"):
                    browser.find_element_by_id("J_SelectAll1").click()
                    break
            except:
                print(f"找不到購買按鈕")
    #method = 1 手動勾選
    else:
        print(f"請手動勾選需要購買的商品")
        time.sleep(5)

#等待搶購時間，定時秒殺，這里我們定義一個buy函數
def buy(times):
    print(times)
    while True:
        now = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S.%f')
        # 對比時間，時間到的話就點擊結算
        if now > times:
            # 點擊結算按鈕
            while True:
                try:
                    if browser.find_element_by_link_text("結 算"):
                        browser.find_element_by_link_text("結 算").click()
                        print(f"結算成功，準備提交訂單")
                        break
                except:
                    pass
            # 點擊提交訂單按鈕
            while True:
                try:
                    if browser.find_element_by_link_text('提交訂單'):
                        browser.find_element_by_link_text('提交訂單').click()
                        print(f"搶購成功，請盡快付款")
                except:
                    print(f"再次嘗試提交訂單")
            time.sleep(0.01)

browser = webdriver.Chrome()
login()
picking()

order_time = input("請輸入搶購時間，格式如(2021-06-08 19:30:00.000000):")

buy(order_time)