Missing resources handled as a readable error

1 files changed, 6 insertions, 4 deletions

diff --git a/jfr_playoff/filemanager.py b/jfr_playoff/filemanager.py
index 990b9f2..3a6c19e 100644
--- a/jfr_playoff/filemanager.py
+++ b/jfr_playoff/filemanager.py
@@ -51,6 +51,11 @@ class PlayoffFileManager(object):
         return self.output_file
 
     def copy_file(self, filename, path):
+        source_path = unicode(os.path.join(
+            os.path.dirname(__main__.__file__), filename))
+        if not os.path.exists(source_path):
+            raise IOError('File: %s missing from runtime directory' % (
+                filename))
         output_path = os.path.join(self.output_path, path)
         output_dir = os.path.dirname(output_path)
         if len(output_dir) > 0:
@@ -61,10 +66,7 @@ class PlayoffFileManager(object):
                 os.makedirs(output_dir)
         PlayoffLogger.get('filemanager').info(
             'copying file to %s', output_path)
-        shutil.copy(
-            unicode(os.path.join(
-                os.path.dirname(__main__.__file__), filename)),
-            unicode(output_path))
+        shutil.copy(source_path, unicode(output_path))
         self.register_file(output_path)
         return output_path