Make viewer runnable from outer dirs.

mihneadb · May 22, 2016 · ad595aa · ad595aa
1 parent f9685ec
commit ad595aa
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/execution_trace/viewer/viewer.py b/execution_trace/viewer/viewer.py
@@ -1,13 +1,16 @@
 # Run with `python viewer.py PATH_TO_RECORD_JSON.
 
 import json
+import os
 import sys
 
 from flask import Flask, jsonify
 from flask.helpers import send_from_directory
 
 
 app = Flask(__name__)
+viewer_root = os.path.abspath(os.path.dirname(__file__))
+
 
 # `main` inits these.
 # File containing `record` output.
@@ -18,7 +21,7 @@
 
 @app.route("/")
 def hello():
-    return send_from_directory('.', 'index.html')
+    return send_from_directory(viewer_root, 'index.html')
 
 
 @app.route("/source.json")