Bläddra i källkod

Merge pull request #122 from vmarkovtsev/master

Add the check for empty input in labours.py
Vadim Markovtsev 6 år sedan
förälder
incheckning
220b31c842
1 ändrade filer med 5 tillägg och 2 borttagningar
  1. 5 2
      labours.py

+ 5 - 2
labours.py

@@ -1,7 +1,5 @@
 #!/usr/bin/env python3
 import argparse
-from datetime import datetime, timedelta
-from importlib import import_module
 import io
 import json
 import os
@@ -12,6 +10,8 @@ import tempfile
 import threading
 import time
 import warnings
+from datetime import datetime, timedelta
+from importlib import import_module
 
 try:
     from clint.textui import progress
@@ -129,6 +129,9 @@ class YamlReader(Reader):
             print("\nInvalid unicode in the input: %s\nPlease filter it through "
                   "fix_yaml_unicode.py" % e)
             sys.exit(1)
+        if data is None:
+            print("\nNo data has been read - has Hercules crashed?")
+            sys.exit(1)
         self.data = data
 
     def get_run_times(self):