def update(sid, language): global pg_session # 首先抓取提交的必要信息 print("Update a submission: sid=%s, lang=%s" % (sid, language)) submission = pg_session.query( pg_models.Submission).filter_by(id=sid).first() # 获得submission模型 code = pg_session.query(pg_models.SubmissionCode).filter_by( submission_id=sid).first().code['code'] # 获得code内容 problem = pg_session.query(pg_models.Problem).filter_by( id=submission.problem_id).first() # 获得problem模型 oj, lang_id = oj_env_message[language] # 获得oj名称和发送的语言代号 pid = problem.origin_pid # 获得题目编号 ret, update_msg = updater_functions[oj](submission.id, pid, lang_id, code) # 返回值为(status, update).status为空表示不存在轮询数据。update不为空表示需要对提交做一些初始化(也可能是直接finish) # 取得回执信息。拥有回执信息的提交,需要放入轮询队列。 # 回执的信息根据oj自己决定。但是至少需要包括如下数据: # oj:oj的名称 # 下面也会自动补全一些数据。 if ret is not None: if 'oj' not in ret: ret['oj'] = oj Submission.mark(submission.id, ret) if update_msg is not None: AlchemyUtils.update_submission(submission, update_msg) pg_session.commit()
def handler(sid): status = Submission.get_status(sid) # 获得该提交的数据。 oj = status['oj'] if status is not None and 'oj' in status else None if oj is not None: print("Report Submission: sid=%s, oj=%s" % (sid, oj)) is_ok, new_status, update = reporter_functions[oj](sid, status) # 回执的ret为(isOk, status, update)。包括轮询完成标志和新的status,以及做出的修改。 # update包括如下内容: # status, score, time, memory, finished if is_ok and status is not None: Submission.unmark(sid) else: Submission.push(sid) Submission.mark(sid, new_status) if update is not None: # 该项不为None时,需要对sdustoj的提交做出修改。 submission = pg_session.query( pg_models.Submission).filter_by(id=sid).first() AlchemyUtils.update_submission(submission, update) pg_session.commit() else: Submission.unmark(sid)
def update(**kwargs): sid = int(kwargs['sid']) # 获得提交信息 language_id = language[kwargs['language']] submission = pg_session.query(pg_models.Submission).filter_by(id=sid).first() code = pg_session.query(pg_models.SubmissionCode).filter_by(submission_id=sid).first().code['code'] # 根据分拆的题目生成相应的HUSTOJ提交 problem_test = pg_session.query(pg_models.ProblemTestData).filter_by( problem_id=submission.problem_id, deleted=False ).all() ok = False submission_create = [] submission_mark = [] tried = 0 while ok is False: # 查询题目,生成提交 ok = True submission_create.clear() submission_mark.clear() tried += 1 for pt in problem_test: title = get_problem_title(pt.problem_id, pt.test_data_id) problem = mysql_session.query(mysql_models.Problem).filter_by(title=title).first() if problem is None: # 若未找到题目,可能是题目尚未更新,更新题目后重新生成 update_problem(pid=pt.problem_id) ok = False break solution = mysql_models.Solution( problem_id=problem.problem_id, user_id=user['user_id'], language=language_id, ip=str(submission.ip), code_length=submission.length, judger=judger, in_date=submission.submit_time ) submission_create.append(solution) submission_mark.append((pt.test_data_id, solution)) if (not ok) and tried > try_max: # 若尝试一定次数后仍未找到题目,有可能此提交信息有问题,放弃生成 break mysql_session.add_all(submission_create) mysql_session.commit() code_create = [] code_user_create = [] sub_mark = [] for tid, sub in submission_mark: print_log('Submission Updated: %s' % (sub.solution_id,)) # 向数据库中写入代码 code_create.append(mysql_models.SourceCode( solution_id=sub.solution_id, source=code )) code_user_create.append(mysql_models.SourceCodeUser( solution_id=sub.solution_id, source=code )) # 标记提交 sub_mark.append((tid, sub.solution_id)) mysql_session.add_all(code_create) mysql_session.add_all(code_user_create) mysql_session.commit() Submission.mark(submission.id, sub_mark)
def update(**kwargs): global mysql_session, pg_session # 可以看出来该函数的作用是:根据获得的提交ID,将此提交同步到hustoj上进行评测。 # 同时,还将这次提交的状态加入本地的解析队列,使问题等待被解析回填到sdustoj上。 sid = int(kwargs['sid']) # 获得本次提交的ID # 获得提交信息 language_name = kwargs['language'] language_id = language[language_name] # 提交所使用的语言 submission = pg_session.query(pg_models.Submission).filter_by(id=sid).first() # 查询sdustoj数据库中的该submission code = pg_session.query(pg_models.SubmissionCode).filter_by(submission_id=sid).first().code['code'] # 查询该提交的代码 # by hk in 2017.05.18====== # 查询编程限制。 limits = pg_session.query(pg_models.Limit).filter_by(problem_id=submission.problem_id).all() max_length = None for limit in limits: environment = pg_session.query(pg_models.Environment).filter_by(id=limit.environment_id).first() if environment.judge_id == language_name: # 确认这是正在使用的编程环境 max_length = limit.length_limit break code_length = len(code) if max_length is not None and code_length > max_length: # 可以知道长度超出限制 finished = True status = 'LLE' table = pg_models.Submission.__table__ # 测试数据状态的表本身? pg_session.execute( table.update().where(table.c.id == sid), {'status': status, 'finished': finished} ) # 看起来这句话的意思是更新该条提交在sdustoj数据库中的信息…… submission.judge_id = judger_id # 更新评测机的ID为本机ID。 pg_session.commit() # 提交更改 return # 处理禁用单词 invalid_words = pg_session.query(pg_models.InvalidWord).filter_by(problem_id=submission.problem_id).all() for invalid_word in invalid_words: if invalid_word.word in code: # 代码包含该禁用单词 finished = True status = 'IW' table = pg_models.Submission.__table__ # 测试数据状态的表本身? pg_session.execute( table.update().where(table.c.id == sid), {'status': status, 'finished': finished} ) # 看起来这句话的意思是更新该条提交在sdustoj数据库中的信息…… submission.judge_id = judger_id # 更新评测机的ID为本机ID。 pg_session.commit() # 提交更改 return # ====================== # 根据分拆的题目生成相应的HUSTOJ提交 problem_test = pg_session.query(pg_models.ProblemTestData).filter_by( problem_id=submission.problem_id, deleted=False ).all() # 查询全部sdustoj中的该提交的题目的测试数据。 ok = False submission_create = [] submission_mark = [] tried = 0 # 定义为已经尝试的提交次数。 while ok is False: # 查询题目,生成提交 ok = True submission_create.clear() # 按照规则,每一次重跑这个while循环都是一次从头开始尝试提交的过程,因此全部清空。 submission_mark.clear() tried += 1 for pt in problem_test: # 循环遍历所有的测试数据。 title = get_problem_title(pt.problem_id, pt.test_data_id) # 按照规范获取该测试数据对应的题目的title try: problem = mysql_session.query(mysql_models.Problem).filter_by(title=title).first() except OperationalError: # MySQL数据库因为长连接断开。重连一次。 mysql_session = MysqlSession() problem = None # 从hustoj数据库查询每一组测试数据对应的题目 if problem is None: # 若未找到题目,可能是题目尚未更新,更新题目后重新生成 update_problem(pid=pt.problem_id) ok = False # 这代表将会跳出提交过程并从头开始尝试提交。 break # 将跳转到下面那个if solution = mysql_models.Solution( problem_id=problem.problem_id, user_id=user['user_id'], language=language_id, ip=str(submission.ip), code_length=submission.length, judger=judger, in_date=submission.submit_time ) # 构造一个hustoj的提交 submission_create.append(solution) # 这个列表记录了所有的提交。 submission_mark.append((pt.test_data_id, solution)) # 这个列表记录了一个sdustoj测试数据ID与hustoj提交信息的元组。 # 将这个提交记入列表。 if (not ok) and tried > try_max: # 如果不断跳出(即一直找不到更新的题目) # 若尝试一定次数后仍未找到题目,有可能此提交信息有问题,放弃生成 break mysql_session.add_all(submission_create) mysql_session.commit() # 执行到这里就是已经成功生成了全部的提交,将它们同步到hustoj数据库 code_create = [] code_user_create = [] sub_mark = [] # 标记的列表。内容为元组(测试数据ID,提交submission的ID). for tid, sub in submission_mark: # 提取所有的测试数据ID与提交信息内容 print_log('Submission Updated: %s' % (sub.solution_id,)) # 向数据库中写入代码 code_create.append(mysql_models.SourceCode( # 似乎这个SourceCode是提交的代码 solution_id=sub.solution_id, source=code )) code_user_create.append(mysql_models.SourceCodeUser( # 这个SourceCodeUser是……? solution_id=sub.solution_id, source=code )) # 标记提交 sub_mark.append((tid, sub.solution_id)) mysql_session.add_all(code_create) mysql_session.add_all(code_user_create) # 什么鬼……为什么hustoj有这么个表? mysql_session.commit() Submission.mark(submission.id, sub_mark) # 向提交中传入这组标记。分别是(提交ID,标记数据(测试数据ID,提交submission的ID)).