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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
import os
import codecs
import re
import requests
import html
import html2text
def rewrite(src,dest):
with open(src,'rb') as fin:
with open(dest,'wb') as fout:
fout.write(fin.read())
def copy():
srcpath = 'F:/files/ACM'
destpath = 'F:/files/ACM_DEST'
files = os.listdir(srcpath)
for name in [x for x in files if os.path.isfile(srcpath + '/' + x)]:
perfix,suffix = os.path.splitext(name)
src = srcpath + '/' + name
dest = destpath + '/' + perfix + '.cpp'
rewrite(src,dest)
print(src,dest,'OK')
def query_by_name(name):
url = 'http://acm.nyist.edu.cn/JudgeOnline/search_result.php?pid=&title=' + name +'&content=&source=&score_least=&score_great=&totalaccept_least=&totalaccept_great=&submit='
r = requests.get(url)
rs = re.findall(r'/JudgeOnline/problem.php\?pid=(\d+)">(.*)</a>',r.content.decode('utf-8'))
return rs
def convert():
srcpath = 'F:/files/ACM_DEST'
files = os.listdir(srcpath)
for name in files:
rs = re.findall(r'(^\d+)(.*)',name)
if len(rs) > 0:
if type(rs[0]) == tuple:
t = rs[0]
print(t[0]+'_'+t[1])
os.rename(srcpath + '/' + name,srcpath + '/' + t[0]+'_'+t[1])
print('AUTO_RENAME:',name,'-->',t[0]+'_'+t[1])
else:
perfix = os.path.splitext(name)[0]
print('query:',name)
result = query_by_name(perfix)
print(result)
pid = input('input:')
os.rename(srcpath+'/'+name,srcpath+'/'+pid+'_'+name)
print('RENAME:',name,'-->',pid+'_'+name)
def get_tag(name):
return '-' * 10 + name + '-' * 10
def to_text(src):
return html2text.html2text(src)
def get_problem(pid):
r = requests.get('http://acm.nyist.edu.cn/JudgeOnline/problem.php?pid='+pid)
rs = re.findall(r'<DD>([\w\W]*?)</DD>',r.content.decode('utf-8'))
details = [html.unescape(x).strip() for x in rs]
text_list = []
text_list.append(get_tag('description'))
text_list.append(to_text(details[0]))
text_list.append(get_tag('input'))
text_list.append(to_text(details[1]))
text_list.append(get_tag('output'))
text_list.append(to_text(details[2]))
text_list.append(get_tag('sample_input'))
text_list.append(details[3].replace(r'<PRE id="sample_input">','').replace(r'</PRE>',''))
text_list.append(get_tag('sample_putput'))
text_list.append(details[4].replace(r'<PRE id="sample_output">','').replace(r'</PRE>',''))
return '\n'.join(text_list)
def complete():
srcpath = 'H:/acm'
files = [x for x in os.listdir(srcpath) if not x.startswith('00')]
for name in files:
pid = name.split('_')[0]
print('WRITE:',pid,name)
code = ''
with open(srcpath + '/' + name,'r+',encoding='utf-8') as f:
code = f.read()
f.seek(0)
f.write('/*\n')
f.write(get_problem(pid))
f.write('\n*/\n')
f.write('/////////////////////////////\n')
f.write(code)
def find_java_file():
srcpath = 'H:/acm'
files = os.listdir(srcpath)
for name in files:
with open(srcpath + '/' + name,'rb') as f:
code = f.read().decode('utf-8')
if code.find('class') > 0:
print(name)
def main():
find_java_file()
if __name__ == '__main__':
main()
|