From 75dd3ee4139b5e076f381af3e62e25abe3227eb7 Mon Sep 17 00:00:00 2001 From: Sebastian Lohff Date: Sun, 26 Jan 2020 11:35:28 +0100 Subject: [PATCH] clear_contest: make python3 ready, remove unused import --- clear_contest.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) 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