From 563bfe9bf572b50fb0a1a9880d5bf1690e9d0f89 Mon Sep 17 00:00:00 2001 From: Oliver Walters Date: Thu, 12 Nov 2020 16:10:00 +1100 Subject: [PATCH] Further fixes to tasks.py --- tasks.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tasks.py b/tasks.py index f4ddf5e0b6..30641c7c81 100644 --- a/tasks.py +++ b/tasks.py @@ -6,6 +6,7 @@ from shutil import copyfile import random import string import os +import sys def apps(): """ @@ -257,11 +258,11 @@ def export_records(c, filename='data.json'): if response not in ['y', 'yes']: print("Cancelled export operation") - return 0 + sys.exit(1) - cmd = f'dumpdata --exclude contenttypes --exclude auth.permission --indent 2 --output {filename}' + cmd = f'dumpdata --exclude contenttypes --exclude auth.permission --indent 2 --output {filename}' - manage(c, cmd, pty=True) + manage(c, cmd, pty=True) @task(help={'filename': 'Input filename'}) def import_records(c, filename='data.json'): @@ -275,7 +276,7 @@ def import_records(c, filename='data.json'): if not os.path.exists(filename): print(f"Error: File '{filename}' does not exist") - return -1 + sys.exit(1) print(f"Importing database records from '{filename}'")