[python] HDU自动登录提交代码程序

调了一个下午，被python的正则绊住了；在C#上运作好好的式子在python老是报错，原来python的断言式必须是固定长度的，像类似(?<=[^>].*?)的零宽度正回顾后发断言是不允许出现的，这点python做的没有C# 方便，弄得我只好分了好几个步骤来做.

 __author__ = 'wuminye'
 import time
 import urllib
 import urllib2
 import cookielib
 import re

 std = ['Queuing', 'Compiling', 'Running']

 hostaddr = 'http://acm.hdu.edu.cn'
 loginaddr = '/userloginex.php?action=login'
 submaddr = '/submit.php?action=submit'
 statusaddr = '/status.php'

 cj = cookielib.LWPCookieJar()
 cookie_support = urllib2.HTTPCookieProcessor(cj)
 #httpHandler = urllib2.HTTPHandler(debuglevel=1)
 opener = urllib2.build_opener(cookie_support, urllib2.HTTPHandler)
 urllib2.install_opener(opener)
 header = {'User-Agent': 'IE10'}

 def login(username, password):
     """

     :rtype : object
     """
     postData = {'username': username,
                 'userpass': password,
                 'submit': 'Sign In'}
     postData = urllib.urlencode(postData)
     request = urllib2.Request(hostaddr + loginaddr, postData, header)
     response = urllib2.urlopen(request, timeout=10)
     if response.getcode() != 200 and response.getcode() != 302:
         print 'login web error!'
         return False
     ans = response.read()
     if ans.find("Sign Out") == -1:
         return False
     return True

 def querystatus():
     postData = {'user': 'wuminye',
                 'lang': 0,
                 'first': '',
                 'pid': '',
                 'status': 0}
     postData = urllib.urlencode(postData)
     request = urllib2.Request(hostaddr + statusaddr + '?' + postData, headers=header)
     response = urllib2.urlopen(request, timeout=10)
     if response.getcode() != 200 and response.getcode() != 302:
         print 'query web error!'
         return False
     ans = response.read()
     m = re.search('(?<=<td height=22px>).*?</td>(<td.*?</td>){8}(?=</tr>)', ans)
     ans = '<td>' + m.group(0)
     mylist = []
     for i in range(0, 7):
         m = re.search(r"(?<=<td).+?(?=</td>)", ans)
         t = m.group()[1:]
         n = t.find('>')
         if n != -1:
             t = t[n + 1:]
         t = re.search('[^<]+', t)
         mylist.append(t.group())
         ans = ans[m.end() + 5:]
     for i in range(0, 4):
         ans = ans[ans.find('>') + 1:]
     ans = re.search(r"[^<]+", ans).group()
     mylist.append(ans)
     return mylist

 def subbmit(pid, lan, code):
     postData = {'problemid': pid,
                 'language': lan,
                 'usercode': code,
                 'submit': 'Submit'}
     postData = urllib.urlencode(postData)
     request = urllib2.Request(hostaddr + submaddr, postData, header)
     response = urllib2.urlopen(request, timeout=10)
     if response.getcode() != 200 and response.getcode() != 302:
         print 'subbmit web error!'
         return False
     print 'Subbmit Done!'
     ans = querystatus()
     while ans[2] in std:
         time.sleep(1)
         print 'Status:', ans[2]
         ans = querystatus()
     print ans