def __init__(self):
self.mysql = MySQL()
self.Judging = JudgeServer()
self.Download = ProblemServer()
self.Judging.setDaemon(True)
self.Download.setDaemon(True)
class Main:
"""
Web端通信格式：
0: 判题模块
参数：[操作数 (任务消息)]
1：Down题模块
参数：OJ缩写 Problem_id pid
2: 获取题目信息
参数：OJ缩写 Problem_id
3: 测试OJ账号
参数：OJ缩写 username password
4: 更新模块
参数：用户id
5: 测试Contest_ID
参数：OJ缩写 Contest_ID
6: 克隆比赛
参数：OJ缩写 Contest_ID
"""
def __init__(self):
self.mysql = MySQL()
self.Judging = JudgeServer()
self.Download = ProblemServer()
self.Judging.setDaemon(True)
self.Download.setDaemon(True)
def __del__(self):
self.Judging.judge_stop()
def start(self):
self.Judging.start()
self.Download.start()
connect = Connect()
while True:
message = connect.get_message()
try:
if message[0] == '0':
JudgeServer.judge_task.put(message[1:])
elif message[0] == '1':
ProblemServer.problem_task.put(message[1:])
elif message[0] == '2':
try:
p = Problem(message[1], message[2], 0)
except IndexError:
connect.receive_message("")
else:
title = p.get_title()
connect.receive_message("%s" % title)
elif message[0] == '3':
connect.receive_message("0")
elif message[0] == '4':
pass
elif message[0] == '5':
oj = message[1]
username = message[2]
password = message[3]
cid = int(message[4])
c = Contest(oj, username, password, cid, -1)
connect.receive_message("%d" % c.get_problem_number())
elif message[0] == '6':
oj = message[1]
username = message[2]
password = message[3]
cid = int(message[4])
contest_id = int(message[5])
c = Contest(oj, username, password, cid, contest_id)
c.start()
else:
raise IndexError
except IndexError:
logging.error("Error message!!! message: " + ' '.join(message))
else:
connect.receive_message()
connect.close_connect()