def setUpClass(cls):
"""Method called to prepare the test fixture."""
super(BaseModelTestCase, cls).setUpClass()
# create user
cls.User = get_user_model()
cls.user = cls.User.objects.create(username='******',
email='*****@*****.**',
password='******')
# create contest
cls.contest = Contest(
title="Concours test",
theme="Concours de test",
description="Description test",
date_begin_upload=date(2020, 12, 1),
date_end_upload=date(2020, 12, 31),
date_begin_vote=date(2021, 1, 1),
date_end_vote=date(2021, 1, 31),
date_creation=datetime.now(),
)
cls.picture = Picture(
title='test_title',
# file_name=image_file,
description='test_description',
technical='test_technical',
camera='test_camera',
place='test_place',
taken_date=date(2021, 1, 1),
# global_score = 0,
user=cls.user,
upload_date=datetime.now(),
)
cls.picture.save()
def setUpClass(cls):
"""Method called to prepare the test fixture."""
super(BaseViewTestCase, cls).setUpClass()
# create user
cls.user = get_user_model().objects.create(username='******',
email='*****@*****.**')
cls.user.set_password('123test')
cls.user.save()
# create picture
small_gif = (
b'\x47\x49\x46\x38\x39\x61\x01\x00\x01\x00\x00\x00\x00\x21\xf9\x04'
b'\x01\x0a\x00\x01\x00\x2c\x00\x00\x00\x00\x01\x00\x01\x00\x00\x02'
b'\x02\x4c\x01\x00\x3b')
cls.picture = Picture(
title='test_title',
file_name=SimpleUploadedFile(name='small.gif',
content=small_gif,
content_type='image/gif'),
description='test_description',
technical='test_technical',
camera='test_camera',
place='test_place',
taken_date=date(2021, 1, 1),
user=cls.user,
upload_date=datetime.now(),
)
cls.picture.save()
# create contests vote
cls.contest = Contest(
title='contest_vote',
theme='contest_theme_vote',
description='contest_description_vote',
date_begin_upload=date(2020, 12, 1),
date_end_upload=date(2020, 12, 31),
deposit=False,
date_begin_vote=date(2021, 1, 1),
date_end_vote=date(2021, 1, 31),
vote_open=True,
archived=False,
date_creation=datetime.now(),
contest_image=SimpleUploadedFile(name='small.gif',
content=small_gif,
content_type='image/gif'),
)
cls.contest.save()
# add picture to contest
cls.contest_picture = ContestPicture(picture=cls.picture,
contest=cls.contest)
cls.contest_picture.save()
# user login
cls.client_login = Client(HTTP_REFERER=reverse('gallery:home'))
cls.logged_in = cls.client_login.login(username='******',
password='******')
def import_tree(path):
paths = []
for root, dirs, files in os.walk(path):
for file in files:
if file.endswith('.contest'):
paths.append(os.path.join(root, file))
contest = Contest(path=os.path.join(root, file))
contest.save()
import_to_database(contest)
return paths
def addContest(request):
if not request.user.is_superuser:
return printError(
'You have no permission to this page, please contact administration'
)
form = ContestForm()
errors = []
courseid = 0
if request.method == "POST":
courseid = request.POST['courseid']
form = ContestForm(request.POST)
if form.is_valid():
data = form.cleaned_data
courseid = data['courseid']
course = get_object_or_404(Course, courseid=courseid)
if not hasPrivilageAddContest(request.user, courseid):
return HttpResponseRedirect("/")
contest = Contest(title=data['title'],
creater=request.user,
start_time=data['start_time'],
end_time=data['end_time'],
courseid=course)
contest.save()
problemList = data['problem_list'].split(',')
for p in problemList:
temp = Problem.objects.filter(pid=p)
if len(temp) == 0:
return printError("problem ** %d ** is not exists" %
int(p))
relation = ContestProblem.objects.filter(
cid=contest, pid=Problem.objects.get(pid=int(p)))
if len(relation) == 0:
relation = ContestProblem(
cid=contest, pid=Problem.objects.get(pid=int(p)))
relation.save()
return HttpResponseRedirect("/courselist/course/%d" %
int(courseid))
elif request.method == 'GET':
try:
courseid = request.GET['courseid']
except:
return printError('You should add contest in course page')
if not hasPrivilageAddContest(request.user, courseid):
return HttpResponseRedirect("/")
course = get_object_or_404(Course, courseid=courseid)
context = {"form": form, "errors": errors, 'course': course}
return render_to_response("addcontest.html",
context,
context_instance=RequestContext(request))
def setUp(self):
Language.objects.filter(oj_name='unit_test_oj', oj_language='unit_test_language',
oj_language_name='unit_test_language_name').delete()
Problem.objects.filter(remote_oj='unit_test_oj', remote_id='unit_test_pid').delete()
Contest.objects.filter(title='unit_test_contest_title', user=self.user).delete()
self.contest = Contest(title='unit_test_contest_title', user=self.user)
self.contest.save()
Language.objects.create(oj_name='unit_test_oj', oj_language='unit_test_language',
oj_language_name='unit_test_language_name').save()
Problem.objects.create(remote_oj='unit_test_oj', remote_id='unit_test_pid').save()
def save(self, user):
if user is None or user.is_authenticated is False:
return None
try:
contest = Contest(title=self.validated_data['title'],
user=str(user))
contest.save()
contest_problem_list = [
ContestProblem(remote_id=problem['remote_id'],
remote_oj=problem['remote_oj'],
contest_id=contest.id)
for problem in self.validated_data['problems']
]
ContestProblem.objects.bulk_create(contest_problem_list)
return contest
except DatabaseError:
return None
def edit_or_create_contest_block(request, contest=None):
is_success = False
if request.method == 'POST' and 'Insert' not in request.POST:
form = ContestEditForm(request.POST)
if form.is_valid():
if contest is None:
if not os.path.exists(
os.path.dirname(form.cleaned_data["path"])):
raise NoDirectoryException("There is no such directory!")
if os.path.exists(form.cleaned_data["path"]):
raise ContestExistsException("This contest already exists")
contest = Contest()
contest.path = form.cleaned_data["path"]
command_create_contest(form.cleaned_data["path"][:-8], [])
contest.name = form.cleaned_data["name"]
contest.id_method = form.cleaned_data["id_method"]
contest.statement_name = form.cleaned_data["statement_name"]
contest.statement_location = form.cleaned_data[
"statement_location"]
contest.statement_date = form.cleaned_data["statement_date"]
contest.statement_template = form.cleaned_data[
"statement_template"]
contest.save()
export_from_database(contest)
is_success = True
else:
if contest is None:
form = ContestEditForm()
else:
form = ContestEditForm(
initial={
'path': contest.path,
'name': contest.name,
'id_method': contest.id_method,
'statement_name': contest.statement_name,
'statement_location': contest.statement_location,
'statement_date': contest.statement_date,
'statement_template': contest.statement_template,
})
return {
'form': form,
'is_success': is_success,
}
def add_contest_block(request):
is_success, is_error = False, False
if request.method == 'POST':
form = AddContestForm(request.POST)
if form.is_valid():
filename = form.cleaned_data['path']
if os.path.exists(filename):
contest = Contest(path=filename)
contest.save()
import_to_database(contest)
is_success = True
else:
is_error = True
else:
form = AddContestForm()
return {
'form': form,
'is_success': is_success,
'is_error': is_error,
}
def copy_contest_block(request):
is_success, is_error = False, False
if request.method == 'POST':
form = CopyContestForm(request.POST)
if form.is_valid():
new_path = norm(form.cleaned_data['new_contest_file'])
old_path = Contest.objects.get(
id=form.cleaned_data['contest']).path
copyfile(old_path, new_path)
contest = Contest(path=new_path)
contest.save()
import_to_database(path=new_path)
contest.save()
is_success = True
else:
form = CopyContestForm()
return {
'form': form,
'is_success': is_success,
'is_error': is_error,
}
user.email="*****@*****.**"
user.is_staff = True
user.save()
>>>>>>> Zwlin
user = User()
user.username = '******'
user.set_password("123456")
user.email="*****@*****.**"
user.is_staff = True
user.save()
# 插入一个比赛
C = Contest()
C.description = '写脚本真累'
C.participant = 200
C.source = 'zwlin'
C.title = 'Testing_Contest'
C.create_user = User.objects.get(username='******')
C.save()
with open("./item.json", 'r') as f:
problems = json.load(f)
id = 1000
for item in problems:
p = Problem()
p.problem_id = str(id)
p.title = item.get('Title', '')
p.description = item.get('ProblemDescription', '')
