Oh my god sys.path[0] is so good

abp/server.py

@@ -7,7 +7,7 @@ import json
 import threading
 import time
 import os
-from graph import GraphState
+from graphstate import GraphState
 
 
 class VizHandler(SimpleHTTPRequestHandler):
@@ -18,7 +18,7 @@ class VizHandler(SimpleHTTPRequestHandler):
         SimpleHTTPRequestHandler.__init__(self, *args, **kwargs)
 
     def get_state(self):
-        """ Get the current graph state """
+        """ Get the current graphstate state """
         self.send_response(200)
         self.send_header('Content-Type', 'application/json')
         self.end_headers()
@@ -77,13 +77,13 @@ def demograph():
 
 
 if __name__ == '__main__':
-    os.chdir("/home/pete/physics/abp/static")
+    where = os.path.join(sys.path[0], "../static")
+    os.chdir(where)
     server = Server()
     server.start()
 
     g = demograph()
 
-
     while True:
         server.update(g)
         time.sleep(1)

abp/viz.py

@@ -2,7 +2,7 @@ import networkx as nx
 from matplotlib import pyplot as plt
 import clifford
 import numpy as np
-from graph import GraphState
+from graphstate import GraphState
 
 VOP_COLORS = ["red", "green", "blue", "orange", "yellow", "purple", "black", "white"]
 

tests/test_graph.py

@@ -1,4 +1,4 @@
-from abp.graph import GraphState, DiffedGraphState
+from abp.graphstate import GraphState, DiffedGraphState
 from abp import clifford
 import time
 

tests/test_json.py

@@ -1,4 +1,4 @@
-from abp.graph import GraphState, DiffedGraphState
+from abp.graphstate import GraphState, DiffedGraphState
 from abp import clifford
 import time
 import json

tests/test_viz.py

@@ -1,4 +1,4 @@
-from abp.graph import GraphState
+from abp.graphstate import GraphState
 from abp import viz