def oj_status_edit(oj_id):
'''
define operations about editing oj status
:param oj_id: oj_id
:return: page
'''
oj = OJList.query.get_or_404(oj_id)
form = ModifyOJStatus()
if form.validate_on_submit():
oj.name = form.oj_name.data
oj.description = form.description.data
oj.url = form.url.data
oj.vjudge = form.vjudge.data
oj.status = form.status.data
db.session.add(oj)
db.session.commit()
current_user.log_operation('Edit oj %s Status, oj_id is %s' %
(oj.name, str(oj.id)))
flash('Update oj status successful!')
return redirect(url_for('admin.oj_list'))
form.oj_name.data = oj.name
form.description.data = oj.description
form.url.data = oj.url
form.vjudge.data = oj.vjudge
form.status.data = oj.status
return render_template('admin/oj-status_edit.html', form=form, oj=oj)
def rejudge_contest_problem(contest_id):
'''
define operation about rejudge contest problem
:param contest_id: contest_id
:return: redirect
'''
contest = Contest.query.get_or_404(contest_id)
# get problem_id from GET request
problem_id = request.args.get('problem_id', -1, type=int)
# if we get a problem id
if problem_id != -1:
# try to get the problem
problem = Problem.query.get(problem_id)
# the problem id is valid
if problem is not None:
# rejudge it
submissions = contest.submissions.filter_by(
problem_id=problem.id).all()
for item in submissions:
item.status = 0
db.session.add(item)
db.session.commit()
current_user.log_operation(
'Rejudge problem %s from contest %s, problem id is %s, contest_id is %s'
% (problem.title, contest.contest_name, str(
problem.id), str(contest.id)))
flash(u'rejudge请求提交成功')
else:
flash(u'题目列表中不存在该题目！')
else:
flash(u'请指定题目ID')
return redirect(url_for('admin.contest_detail', contest_id=contest_id))
def tag_insert():
'''
define add tag operation
:return: page
'''
tag = Tag()
form = ModifyTag()
if form.validate_on_submit():
tag.tag_name = form.tag_name.data
db.session.add(tag)
db.session.commit()
current_user.log_operation('Add tag %s' % tag.tag_name)
flash('Add tag sucessful!')
return redirect(url_for('admin.tag_list'))
form.tag_name.data = ''
return render_template('admin/tag_edit.html', form=form)
def tag_edit(tag_id):
'''
define edit tag operation
:param tag_id: tag id
:return: page
'''
tag = Tag.query.get_or_404(tag_id)
form = ModifyTag()
if form.validate_on_submit():
tag.tag_name = form.tag_name.data
db.session.add(tag)
db.session.commit()
current_user.log_operation('Edit tag %s, tag_id is %s' %
(tag.tag_name, str(tag.id)))
flash('Tag update sucessful!')
return redirect(url_for('admin.tag_list'))
form.tag_name.data = tag.tag_name
return render_template('admin/tag_edit.html', form=form)
def blog_insert():
'''
define operation about insert blog
:return: page
'''
blog = Blog()
form = ModifyBlog()
if form.validate_on_submit():
blog.title = form.title.data
blog.content = form.content.data
blog.author_username = current_user.username
blog.public = form.public.data
db.session.add(blog)
db.session.commit()
current_user.log_operation('Add blog %s, blog_id is %s' %
(blog.title, str(blog.id)))
flash('Add blog successful!')
return redirect(url_for('admin.blog_list'))
return render_template('admin/blog_add.html', form=form)
def oj_status_insert():
'''
define operations about adding oj status
:return: page
'''
oj = OJList()
form = ModifyOJStatus()
if form.validate_on_submit():
oj.name = form.oj_name.data
oj.description = form.description.data
oj.url = form.url.data
oj.vjudge = form.vjudge.data
oj.status = form.status.data
db.session.add(oj)
db.session.commit()
current_user.log_operation('Add oj %s, oj_id is %s' %
(oj.name, str(oj.id)))
flash('Add oj status successful!')
return redirect(url_for('admin.oj_list'))
return render_template('admin/oj-status_add.html', form=form)
def oj_status_delete():
'''
define operations about deleting oj status
:return: page
'''
# get problem_id from GET request
oj_id = request.args.get('oj_id', -1, type=int)
if oj_id != -1:
oj = OJList.query.get(oj_id)
if oj is not None:
db.session.delete(oj)
db.session.commit()
current_user.log_operation('Delete oj %s, oj_id is %s' %
(oj.name, str(oj.id)))
flash('Delete oj successful!')
return redirect(url_for('admin.oj_list'))
else:
flash('No such oj in oj_list!')
else:
flash('No such oj in oj_list!')
return redirect(url_for('admin.oj_list'))
def user_edit(user_id):
'''
deal with user edit operation
:param user_id: user_id
:return: page
'''
user = User.query.get_or_404(user_id)
form = ModifyUser(user)
if form.validate_on_submit():
if user.role.permission >= current_user.role.permission and current_user.username != user.username and current_user.role.permission != 0xff:
current_user.log_operation(
'Try to edit a high level permission user %s, user_id is %s' %
(user.username, str(user.id)))
flash(u"您无法编辑一个更高权限的用户!")
return redirect(url_for('admin.user_list'))
# if change to a high level role, the operation can not be exec
elif Role.query.get_or_404(
form.role_id.data
).permission >= current_user.role.permission and current_user.role.permission != 0xff:
current_user.log_operation(
'Try to grant user %s a high level permission, user_id is %s' %
(user.username, str(user.id)))
flash(u"您不能给用户授予高于您本身的权限!")
return redirect(url_for('admin.user_edit', user_id=user_id))
user.email = form.email.data
user.confirmed = form.confirmed.data
user.nickname = form.nickname.data
user.gender = form.gender.data
user.major = form.major.data
user.degree = form.degree.data
user.country = form.country.data
user.address = form.address.data
user.school = form.school.data
user.student_num = form.student_num.data
user.phone_num = form.phone_num.data
user.about_me = form.about_me.data
user.role_id = form.role_id.data
db.session.add(user)
db.session.commit()
current_user.log_operation('Edit user %s, user_id is %s' %
(user.username, str(user.id)))
flash('Update successful!')
return redirect(url_for('admin.user_list'))
form.email.data = user.email
form.username.data = user.username
form.nickname.data = user.nickname
form.confirmed.data = user.confirmed
form.role_id.data = user.role_id
form.gender.data = user.gender
form.major.data = user.major
form.degree.data = user.degree
form.address.data = user.address
form.country.data = user.country
form.school.data = user.school
form.student_num.data = user.student_num
form.phone_num.data = user.phone_num
form.about_me.data = user.about_me
return render_template('admin/user_edit.html', form=form, user=user)
def delete_contest_problem(contest_id):
'''
define contest delete problem operation
:param contest_id: contest_id
:return: page
'''
contest = Contest.query.get_or_404(contest_id)
# get problem_id from GET request
problem_id = request.args.get('problem_id', -1, type=int)
# if we get a problem id
if problem_id != -1:
# try to get the problem
problem = Problem.query.get(problem_id)
# the problem id is valid
if problem is not None:
# get the contest_problem data from the sql
contest_problem = contest.problems.filter_by(
problem_id=problem.id).first()
# problem is in the contest
if contest_problem:
# delete it
db.session.delete(contest_problem)
db.session.commit()
current_user.log_operation(
'Delete problem %s from contest %s, problem id is %s, contest_id is %s'
% (problem.title, contest.contest_name, str(
problem.id), str(contest.id)))
flash(u'删除成功')
else:
flash(u'比赛题目列表中没有该题目！')
else:
flash(u'题目列表中不存在该题目！')
else:
flash(u'请指定题目ID')
return redirect(url_for('admin.add_contest_problem',
contest_id=contest_id))
def add_contest_problem(contest_id):
'''
define contest add problem operation
:param contest_id: contest id
:return: page
'''
contest = Contest.query.get_or_404(contest_id)
form = AddContestProblem()
if form.validate_on_submit():
problem = Problem.query.get(form.problem_id.data)
if problem is not None:
if ContestProblem.query.filter_by(
problem_id=problem.id,
contest_id=contest_id).first() != None:
flash(u'题目已添加至比赛中，请勿重复添加!')
return redirect(
url_for('admin.add_contest_problem',
contest_id=contest_id))
contest_problem = ContestProblem(contest=contest, problem=problem)
contest_problem.problem_index = contest.problems.count() + 1000
if form.problem_alias != '' and form.problem_alias is not None:
contest_problem.problem_alias = form.problem_alias.data
else:
contest_problem.problem_alias = problem.title
db.session.add(contest_problem)
db.session.commit()
current_user.log_operation(
'Add problem %s to contest %s, problem id is %s, contest_id is %s'
% (problem.title, contest.contest_name, str(
problem.id), str(contest.id)))
flash(u'添加题目成功!')
else:
flash(u'题目不存在!')
return render_template('admin/contest_add_problem.html',
form=form,
contest=contest)
def blog_edit(blog_id):
'''
define operation about edit blog
:return: page
'''
blog = Blog.query.get_or_404(blog_id)
form = ModifyBlog()
if form.validate_on_submit():
blog.title = form.title.data
blog.content = form.content.data
blog.author_username = current_user.username
blog.public = form.public.data
blog.last_update = datetime.utcnow()
db.session.add(blog)
db.session.commit()
current_user.log_operation('Edit blog %s, blog_id is %s' %
(blog.title, str(blog.id)))
flash('Edit blog successful!')
return redirect(url_for('admin.blog_list'))
form.title.data = blog.title
form.content.data = blog.content
form.public.data = blog.public
return render_template('admin/blog_edit.html', form=form)
def problem_insert():
'''
deal with the insert problem operation
:return: page
'''
problem = Problem()
form = ModifyProblem()
if form.validate_on_submit():
if Problem.query.filter_by(oj_id=form.oj_id.data).order_by(
Problem.remote_id.desc()).first() is None:
problem.remote_id = 1
else:
problem.remote_id = Problem.query.filter_by(
oj_id=form.oj_id.data).order_by(
Problem.remote_id.desc()).first().id + 1
problem.oj_id = form.oj_id.data
problem.title = form.title.data
problem.time_limit = form.time_limit.data
problem.memory_limit = form.memory_limit.data
problem.special_judge = form.special_judge.data
problem.submission_num = form.submission_num.data
problem.accept_num = form.accept_num.data
problem.description = form.description.data
problem.input = form.input.data
problem.output = form.output.data
problem.sample_input = form.sample_input.data
problem.sample_output = form.sample_output.data
problem.source_name = form.source_name.data
problem.hint = form.hint.data
problem.author = form.author.data
problem.visible = form.visible.data
problem.last_update = datetime.utcnow()
db.session.add(problem)
db.session.commit()
# add tags to the problem, delete old tags
t = TagProblem.query.filter_by(problem_id=problem.id).all()
for tag in t:
db.session.delete(t)
db.session.commit()
for i in form.tags.data:
t = TagProblem(tag_id=i, problem_id=problem.id)
db.session.add(t)
# db.session.commit()
db.session.commit()
current_user.log_operation('Insert problem "%s", problem_id is %s' %
(problem.title, str(problem.id)))
return redirect(url_for('admin.problem_list'))
form.oj_id.data = problem.oj_id
form.title.data = problem.title
form.time_limit.data = problem.time_limit
form.memory_limit.data = problem.memory_limit
form.special_judge.data = problem.special_judge
form.submission_num.data = problem.submission_num
form.accept_num.data = problem.accept_num
form.description.data = problem.description
form.input.data = problem.input
form.output.data = problem.output
form.sample_input.data = problem.sample_input
form.sample_output.data = problem.sample_output
form.source_name.data = problem.source_name
form.hint.data = problem.hint
form.author.data = problem.hint
form.visible.data = problem.visible
form.tags.data = problem.tags
return render_template('admin/problem_insert.html',
form=form,
problem=problem)
def contest_edit(contest_id):
'''
define edit contest operation
:param contest_id: contest_id
:return: page
'''
contest = Contest.query.get_or_404(contest_id)
form = ModifyContest()
if form.validate_on_submit():
contest.contest_name = form.contest_name.data
contest.start_time = form.start_time.data - timedelta(hours=8)
contest.end_time = form.end_time.data - timedelta(hours=8)
# end_time is smaller than start_time, we use the 5 hours later time as the end time
if contest.end_time <= contest.start_time:
contest.end_time = contest.start_time + timedelta(hours=5)
# Five types contest
if form.type.data == 1:
# contest we need to registe, no password, no verify
contest.style = 1
contest.verify = False
contest.password = ''
elif form.type.data == 2 or form.type.data == 4:
# contest we need to registe or pre register of onsite contest, no password, but has verify by the manager
contest.style = form.type.data
contest.verify = True
contest.password = ''
elif form.type.data == 3:
# contest we need to registe, has the password, no verify
contest.style = 3
contest.verify = False
contest.password = form.password.data
if form.password.data == '':
flash(u'密码保护的比赛密码不能为空!')
return redirect(
url_for('admin.contest_insert', contest_id=contest.id))
elif form.type.data == 5:
# contest onsite, no password, no verify
contest.style = 5
contest.verify = False
contest.password = ''
contest.description = form.description.data
contest.announce = form.announce.data
contest.visible = form.visible.data
contest.manager_username = form.manager.data
contest.rank_frozen = form.rank_frozen.data
db.session.add(contest)
db.session.commit()
current_user.log_operation('Edit contest %s, contest_id is %s' %
(contest.contest_name, str(contest.id)))
flash(u'编辑比赛成功!')
# Todo: check if the contest.id is good for use
return redirect(
url_for('admin.add_contest_problem', contest_id=contest.id))
form.contest_name.data = contest.contest_name
form.start_time.data = contest.start_time + timedelta(hours=8)
form.end_time.data = contest.end_time + timedelta(hours=8)
form.type.data = contest.style
form.password.data = contest.password
form.description.data = contest.description
form.announce.data = contest.announce
form.visible.data = contest.visible
form.manager.data = contest.manager_username
form.rank_frozen.data = contest.rank_frozen
return render_template('admin/contest_edit.html',
form=form,
contest=contest,
current_time=datetime.utcnow())