需求:
- 把自己之前在contest里面的代码提取出来。
- 实现批量提交contest和problem里面的题目。
- -
过程:
总共大概花了4个小时,一晚上,一个类一个文件的方法写起来真的爽,一晚上没停住手。
自己首先写的是下载器,首先明确需求。
- 可以模拟登陆。
- 可以post请求。
- 可以下载网页。
- 为了省事,把提交题目的功能也整合里面了。
代码如下,实现起来没啥困难,毕竟已经是轻车熟路了。
# coding:utf8
import urllib2
import cookielib
import urllib
class Down(object):
def __init__(self):
#初始化常用网址
self.login_url = 'http://acm.sdut.edu.cn/onlinejudge2/index.php/Home/Login/login'
self.home_url = 'http://acm.sdut.edu.cn/onlinejudge2/index.php/Home'
self.submit_url = 'http://acm.sdut.edu.cn/onlinejudge2/index.php/Home/Solution/submitsolution/pid/1110.html'
#安装cookie处理
cj = cookielib.CookieJar()
cookie_support = urllib2.HTTPCookieProcessor(cj)
opener = urllib2.build_opener(cookie_support)
urllib2.install_opener(opener)
#初始化cookie
home_temp = urllib2.urlopen(self.home_url)
self.headers = {
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/54.0.2840.71 Safari/537.36',
'Referer': 'http://acm.sdut.edu.cn/onlinejudge2/index.php/Home/Login/login',
'Host': 'acm.sdut.edu.cn'}
self.login()
def post(self, url, post_data = None):
#下载网页和POST请求
if post_data is None:
request = urllib2.Request(url, headers = self.headers)
response = urllib2.urlopen(request)
return response.read()
else:
post_data = urllib.urlencode(post_data)
request = urllib2.Request(url, post_data, self.headers)
response = urllib2.urlopen(request)
return response