【HDOJ】3337 Guess the number

神一样的题目。简言之，利用手段获得测试用例的第一行，输出结果。
很显然利用wa, TLE, OLE等judge status可以获得测试用例。
因此，果断Python写一个acm提交机器人。依赖lxml库。
测试用例是The Ascii of H.

#!/usr/env python

import urllib2
import urllib
import cookielib
import logging
from urlparse import *
from time import sleep
from lxml import etree
import string

global url, n_vol
url  = "http://acm.hdu.edu.cn"
n_vol = 46
judgeStatus_xpath = "/html/body/table/tr[4]/td/div/table/tr[2]/td[3]/font"

def initLog():
    # os.chdir(parentDirPath)
    suffix = 'hdu'
    logging.basicConfig(
                level=logging.DEBUG,
                format='%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s',
                datefmt='%a, %d %b %Y %H:%M:%S',
                filename=__file__[:-3]+'_'+suffix+'.log',
                filemode='w')
    
    
def acmHdu_Login(user, password):
    login_page = urljoin(url, "userloginex.php?action=login")
    try:
        'create cookie'
        cj = cookielib.CookieJar()
        opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cj))
        opener.addheaders = [('User-agent',
                 'Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1)')]
        data = urllib.urlencode({"username":user, "userpass":password})
        urllib2.install_opener(opener)
        urllib2.urlopen(login_page, data)
        
    except Exception, e:
        print "Login unsucceed"
        print e
    
def acmHdu_Submit(pid, lang, code):
    submit_url = urljoin(url, "submit.php?action=submit")
    data = urllib.urlencode({"problemid":pid, "language":lang, "usercode":code})
    req = urllib2.Request(submit_url, data)
    response = urllib2.urlopen(req)
    content = response.read()
    
    
def acmHdu_Status(user, pid, lang):
    sleep(5)
    status_url = urljoin(url, "status.php?first=&pid=%s&user=%s&lang=%s&status=0" % (str(pid), user, str(lang+1)))
    # print status_url
    content = urllib2.urlopen(status_url).read()
    with open("C:UsershoohDesktopst.html", "w") as fout:
        fout.write(content)
        
        
def acmHdu_Status_Parse(user, pid, lang):
    sleep(5)
    status_url = urljoin(url, "status.php?first=&pid=%s&user=%s&lang=%s&status=0" % (str(pid), user, str(lang+1)))
    # print status_url
    content = urllib2.urlopen(status_url).read()
    tree = etree.HTML(content)
    infoList = map(
        lambda ele: ele.text, tree.xpath(judgeStatus_xpath)
    )
    return infoList[0] if infoList else ""
    

class constForSubmit:
    can = ' ' + string.digits + string.uppercase + string.lowercase
    pid = 3337
    lang = 2
    template = string.Template(
'''
/* 3337 */
#include <iostream>
#include <string>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <vector>
#include <deque>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <ctime>
#include <cstring>
#include <climits>
#include <cctype>
#include <cassert>
#include <functional>
#include <iterator>
#include <iomanip>
using namespace std;
//#pragma comment(linker,"/STACK:102400000,1024000")

#define sti                set<int>
#define stpii            set<pair<int, int> >
#define mpii            map<int,int>
#define vi                vector<int>
#define pii                pair<int,int>
#define vpii            vector<pair<int,int> >
#define rep(i, a, n)     for (int i=a;i<n;++i)
#define per(i, a, n)     for (int i=n-1;i>=a;--i)
#define clr                clear
#define pb                 push_back
#define mp                 make_pair
#define fir                first
#define sec                second
#define all(x)             (x).begin(),(x).end()
#define SZ(x)             ((int)(x).size())
#define lson            l, mid, rt<<1
#define rson            mid+1, r, rt<<1|1

char pattern[105];
int len;

void init() {
    len = 0;
    pattern[len++] = ' ';
    rep(i, 0, 10)
        pattern[len++] = '0'+i;
    rep(i, 0, 26)
        pattern[len++] = 'A'+i;
    rep(i, 0, 26)
        pattern[len++] = 'a'+i;
    puts(pattern);
}

void WA() {
    puts("Wa");
}

void TLE() {
    while (true);
}

void OLE() {
    while (true)
        puts("123");
}

int main() {
    ios::sync_with_stdio(false);
    #ifndef ONLINE_JUDGE
        freopen("data.in", "r", stdin);
        freopen("data.out", "w", stdout);
    #endif
    
    init();
    
    char line[20];
    
    gets(line);
    int len = strlen(line);
    int index = $index;
    int th = $th;
    
    if (index >= len)
        OLE();
    
    if (pattern[th] == line[index]) {
        WA();
    } else {
        while    (true)
            ;
    }
    
    #ifndef ONLINE_JUDGE
        printf("time = %d.\n", (int)clock());
    #endif
    
    return 0;
}


''')

class CFS(constForSubmit):
    pass
    
    
def acmHdu_Submit3337(user, pid, lang, template):
    maxl = 20
    varDict = dict()
    cn = len(CFS.can)
    ret = ""
    for index in xrange(maxl):
        flag = False
        for th in xrange(cn):
            varDict["index"] = index
            varDict["th"] = th
            code = template.safe_substitute(varDict)
            acmHdu_Submit(pid, lang, code)
            while True:
                status = acmHdu_Status_Parse(user, pid, lang).strip()
                if status=="Running" or status=="Queuing" or status=="Compiling":
                    continue
                print status
                if status == "Wrong Answer":
                    ### find the correct char
                    ret += CFS.can[th]
                    flag = True
                    break
                elif status == "Output Limit Exceeded":
                    ### find the last char
                    logging.debug("Result is %s" % (ret))
                    return ret
                elif status == "Time Limit Exceeded":
                    break
                else:
                    continue
            if flag:
                break
        logging.debug("%d: ret = %s" % (index, ret))
    return ret
    
    
if __name__ == "__main__":
    # user = "Bombe16"
    # password = "496528674"
    # initLog();
    # logging.debug("acm submit begin...")
    # acmHdu_Login(user, password)
    # acmHdu_Submit(CFS.pid, CFS.lang, CFS.code)
    # acmHdu_Status(user, CFS.pid, CFS.lang)
    # logging.debug("acm submit end...")
    
    user = "Bombe16"
    password = "496528674"
    initLog();
    logging.debug("acm submit begin...")
    acmHdu_Login(user, password)
    line = acmHdu_Submit3337(user, CFS.pid, CFS.lang, CFS.template)
    with open("F:Qt_prjhdojdata.in", "w") as fout:
        fout.write(line)
    logging.debug("acm submit end...")