# !/usr/bin/env python
# -*- coding: utf-8 -*-
# 2019/03/16
# 淘寶秒殺腳本，掃碼登錄版
from selenium import webdriver
import datetime
import time
 
 
def login():
  # 打開淘寶登錄頁，并進行掃碼登錄
  browser.get("https://www.taobao.com")
  time.sleep(3)
  if browser.find_element_by_link_text("親，請登錄"):
    browser.find_element_by_link_text("親，請登錄").click()
    print("請在15秒內(nèi)完成掃碼")
    time.sleep(15)
    browser.get("https://cart.taobao.com/cart.htm")
  time.sleep(3)
 
  now = datetime.datetime.now()
  print('login success:', now.strftime('%Y-%m-%d %H:%M:%S'))
 
 
def buy(times, choose):
  # 點擊購物車里全選按鈕
  if choose == 2:
    print("請手動勾選需要購買的商品")
  while True:
    now = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S.%f')
    # 對比時間，時間到的話就點擊結算
    if now > times:
      if choose == 1:
        while True:
          try:
            if browser.find_element_by_id("J_SelectAll2"):
              browser.find_element_by_id("J_SelectAll2").click()
              break
          except:
            print("找不到購買按鈕")
      # 點擊結算按鈕
      while True:
        try:
          if browser.find_element_by_link_text("結 算"):
            browser.find_element_by_link_text("結 算").click()
            print("結算成功")
            break
        except:
          pass
      while True:
        try:
          if browser.find_element_by_link_text('提交訂單'):
            browser.find_element_by_link_text('提交訂單').click()
            now1 = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S.%f')
            print("搶購成功時間：%s" % now1)
        except:
          print("再次嘗試提交訂單")
      time.sleep(0.01)
 
 
if __name__ == "__main__":
  times = input("請輸入搶購時間，格式如(2018-09-06 11:20:00.000000):")
  # 時間格式："2018-09-06 11:20:00.000000"
  browser = webdriver.Chrome()
  browser.maximize_window()
  login()
  choose = int(input("到時間自動勾選購物車請輸入“1”，否則輸入“2”："))
  buy(times, choose)