forked from ashtonwebster/meesh_autotest
-
Notifications
You must be signed in to change notification settings - Fork 0
/
meesh_autotest.py
48 lines (37 loc) · 1.22 KB
/
meesh_autotest.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
from lxml import html
import requests
import os
import errno
import sys
if (len(sys.argv) != 3):
print "usage: python meesh_autotest.py <lower_bound_id> <upper_bound_id>"
exit()
url = "https://cmsc420.cs.umd.edud/meeshquest/part2/input/"
try:
os.makedirs("input")
except OSError:
if not (os.path.isdir("input")):
raise
try:
os.makedirs("output")
except OSError:
if not (os.path.isdir("output")):
raise
for x in range(int(sys.argv[1]), int(sys.argv[2])):
print "getting id " + str(x)
page = requests.get('https://cmsc420.cs.umd.edu/meeshquest/part2/input/' + str(x), verify=False)
tree = html.fromstring(page.text)
filename = (tree.xpath('/html/body/div[2]/div/h4[1]/text()')[0].split("Uploaded As: ")[1])
input_filename = "input/" + str(x) + "_" + filename
#print filename
input = tree.xpath('//div[@class="container theme-showcase"]/div[@class="jumbotron"]/pre/text()')[0]
#print input[0]
f = open(input_filename, 'w')
f.write(input)
f.close()
output_filename = "output/" + str(x) + "_" + filename
output = tree.xpath('/html/body/div[2]/div/textarea/text()')[0]
#print output
f = open(output_filename, 'w')
f.write(output)
f.close()