from reversion import revisions

from judge.models.choices import (
    ACE_THEMES,
    EFFECTIVE_MATH_ENGINES,
    MATH_ENGINES_CHOICES,
    TIMEZONE,
)
from judge.models.comment import Comment, CommentLock, CommentVote
from judge.models.contest import (
    Contest,
    ContestMoss,
    ContestParticipation,
    ContestProblem,
    ContestSubmission,
    ContestTag,
    Rating,
    ContestProblemClarification,
    ContestsSummary,
)
from judge.models.interface import BlogPost, MiscConfig, NavigationBar, validate_regex
from judge.models.message import PrivateMessage, PrivateMessageThread
from judge.models.problem import (
    LanguageLimit,
    LanguageTemplate,
    License,
    Problem,
    ProblemGroup,
    ProblemTranslation,
    ProblemType,
    Solution,
    TranslatedProblemQuerySet,
    ProblemPointsVote,
)
from judge.models.problem_data import (
    CHECKERS,
    ProblemData,
    ProblemTestCase,
    problem_data_storage,
    problem_directory_file,
)
from judge.models.profile import (
    Organization,
    OrganizationRequest,
    Profile,
    Friend,
    OrganizationProfile,
)
from judge.models.runtime import Judge, Language, RuntimeVersion
from judge.models.submission import (
    SUBMISSION_RESULT,
    Submission,
    SubmissionSource,
    SubmissionTestCase,
)

from judge.models.test_formatter import TestFormatterModel
from judge.models.ticket import Ticket, TicketMessage
from judge.models.volunteer import VolunteerProblemVote
from judge.models.pagevote import PageVote, PageVoteVoter
from judge.models.bookmark import BookMark, MakeBookMark
from judge.models.course import Course
from judge.models.notification import Notification, NotificationProfile
from judge.models.test_formatter import TestFormatterModel

revisions.register(Profile, exclude=["points", "last_access", "ip", "rating"])
revisions.register(Problem, follow=["language_limits"])
revisions.register(LanguageLimit)
revisions.register(LanguageTemplate)
revisions.register(Contest, follow=["contest_problems"])
revisions.register(ContestProblem)
revisions.register(Organization)
revisions.register(BlogPost)
revisions.register(Solution)
revisions.register(Judge, fields=["name", "created", "auth_key", "description"])
revisions.register(Language)
revisions.register(
    Comment, fields=["author", "time", "page", "score", "body", "hidden", "parent"]
)
revisions.register(ProblemTranslation)
revisions.register(ProblemPointsVote)
revisions.register(ContestMoss)
revisions.register(ProblemData)
revisions.register(ProblemTestCase)
revisions.register(ContestParticipation)
revisions.register(Rating)
revisions.register(PageVoteVoter)
revisions.register(VolunteerProblemVote)
revisions.register(MakeBookMark)
revisions.register(Course)
del revisions