import django
from contest.models import *

django.setup()


def add_exist_problem_to_contest(problems, contest_id):
    try:
        contest = Contest.objects.get(pk=contest_id)
    except Contest.DoesNotExist:
        print "Contest Doesn't Exist!"
        return
    i = 1
    for problem in problems:
        print "Add the problem:"
        print problem.title
        print "The sort Index is" + str(i) + " You Can modify it latter as you like~"
        ContestProblem.objects.create(contest=contest, sort_index=str(i),
                              title=problem.title, description=problem.description,
                              input_description=problem.input_description,
                              output_description=problem.output_description,
                              samples=problem.samples,
                              test_case_id=problem.test_case_id,
                              hint=problem.hint,
                              created_by=problem.created_by,
                              time_limit=problem.time_limit,
                              memory_limit=problem.memory_limit)
        i += 1
    return