diff --git a/clear_contest.py b/clear_contest.py index 4d24233..102aa28 100755 --- a/clear_contest.py +++ b/clear_contest.py @@ -1,8 +1,6 @@ #!/usr/bin/env python from __future__ import print_function -import datetime - # prepare environment import sys sys.path.append("..") @@ -11,11 +9,15 @@ os.environ.setdefault("DJANGO_SETTINGS_MODULE", "cqtu.settings") import django django.setup() -confirm = raw_input("Do are you sure you want to clear all contest data? Answer with uppercase YES: ") +confirm_msg = "Do are you sure you want to clear all contest data? Answer with uppercase YES: " +try: + confirm = raw_input(confirm_msg) +except NameError: + confirm = input(confirm_msg) if confirm != "YES": - print("Aborting") - sys.exit(1) + print("Aborting") + sys.exit(1) from contest.models import QSO, ShadowCall, Reference, User