# encoding: utf-8 
''''' 
@author: Techzero 
@email: techzero@163.com 
@time: 2014-5-18 下午5:06:29 
''' 
import cStringIO 
import getopt 
import time 
import urllib2 
import subprocess 
import sys 
 
from datetime import datetime 
 
MEDIA_PLAYER = 'C:/Program Files/Windows Media Player/wmplayer.exe' 
MEDIA_FILE = 'D:/notify.mp3' 
CHROME = 'C:/Program Files (x86)/Google/Chrome/Application/chrome.exe' 
URL = 'http://detail.ju.taobao.com/home.htm?spm=608.2214381.2.1.SY0wVT&item_id=16761325430&id=10000002801432' 
NO_X11 = False 
 
def get_current_button(): 
 '''''獲取當(dāng)前按鈕狀態(tài)''' 
 content = urllib2.urlopen(URL).read() #獲取頁(yè)面內(nèi)容 
  
 buf = cStringIO.StringIO(content.decode('gbk').encode('utf8')) #將頁(yè)面內(nèi)容轉(zhuǎn)換為輸入流 
 current_button = None 
 for line in buf: 
  line = line.strip(' \n\r') #去掉回車(chē)換行 
   
  if line.find(r'<a href="#" rel="external nofollow" class="extra notice J_BuyButtonSub">開(kāi)團(tuán)提醒</a>') != -1: 
   current_button = '開(kāi)團(tuán)提醒' 
   break 
  elif line.find(r'<div class="main-box chance ">') != -1: 
   current_button = '還有機(jī)會(huì)' 
   break 
  elif line.find(r'<span class="out floatright">賣(mài)光了...</span>') != -1: 
   current_button = '賣(mài)光了' 
   break 
  elif line.find(r'<span class="out floatright">已結(jié)束...</span>') != -1: 
   current_button = '已結(jié)束' 
   break 
  elif line.find(r'<input type="submit" class="buyaction J_BuySubmit" title="馬上搶" value="馬上搶"/>') != -1: 
   current_button = '馬上搶' 
   break 
   
 buf.close() 
 return current_button 
 
 
def notify(): 
 '''''發(fā)出通知并用Chrome打開(kāi)秒殺頁(yè)面''' 
 subprocess.Popen([MEDIA_PLAYER, MEDIA_FILE]) 
 if not NO_X11: 
  subprocess.Popen([CHROME, URL]) 
  print '打開(kāi)頁(yè)面' 
 
 
def monitor_button(interval, last): 
 '''''開(kāi)始監(jiān)視按鈕''' 
 elapse = 0 
 while elapse < last: 
  current_button = get_current_button() 
 
  now = datetime.now() 
  print '%d-%d-%d %d:%d:%d - 現(xiàn)在按鈕是 %s' % (now.year, now.month, now.day, now.hour, now.minute, now.second, current_button) 
 
  if current_button == '馬上搶' or current_button == '還有機(jī)會(huì)': 
   print '趕緊搶購(gòu)！' 
   notify() 
   break 
  elif current_button == '賣(mài)光了' or current_button == '已結(jié)束': 
   print '下次再試吧！' 
   break 
  else: 
   print '還沒(méi)開(kāi)始呢，再等等吧！' 
 
  time.sleep(interval) 
  elapse += interval 
 
 
def usage(): 
 print ''''' 
usage: monitor_mac_price.py [options] 
 
Options: 
 -i interval: 30 seconds by default. 
 -l last: 1800 seconds by default. 
 -h: Print this usage. 
 -X: Run under no X11. 
''' 
 
if __name__ == '__main__': 
 try: 
  opts, args = getopt.getopt(sys.argv[1:], 'i:l:hX') 
 except getopt.GetoptError, err: 
  print str(err) 
  sys.exit(1) 
 
 interval = 0.1 
 last = 1800 
 
 for opt, val in opts: 
  if opt == '-i': 
   interval = int(val) 
  elif opt == '-l': 
   last = int(val) 
  elif opt == '-X': 
   NO_X11 = True 
  elif opt == '-h': 
   usage() 
   sys.exit() 
 
 monitor_button(interval, last)