#!/usr/bin/env python
# -*- coding: utf-8 -*-

from selenium import webdriver
import datetime
import time


def login():
  # 打開淘寶登錄頁，并進(jìn)行掃碼登錄
  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):
  is_buyed = False
  # 點擊購物車?yán)锶x按鈕
  if choose == 2:
    print("請手動勾選需要購買的商品")
  while True:
    now = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S.%f')
    print('現(xiàn)在時間：', now)
    # 對比時間，時間到的話就點擊結(jié)算
    if now > times:
      if choose == 1:
        while True:
          try:
            if browser.find_element_by_id("J_SelectAllcbx1"):
              browser.find_element_by_id("J_SelectAllcbx1").click()
              print('嘗試全選')
              break
          except:
            print("找不到全選按鈕")
      # 點擊結(jié)算按鈕
      try:
        if browser.find_element_by_id("J_Go"):
          browser.find_element_by_id("J_Go").click()
          print("結(jié)算成功")
      except:
        pass
      while True:
        try:
          if browser.find_element_by_link_text('提交訂單') and is_buyed == False:
            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.005)


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 = input("到時間自動勾選購物車請輸入“1”，否則輸入“2”：")
  buy(times, choose)