Update path so the script can be run and import atst

2018-10-16 10:17:37 -04:00 · 2018-10-16 10:17:37 -04:00 · 8bf71f10c7
commit 8bf71f10c7
parent 9d03c3e856
1 changed files with 7 additions and 0 deletions
--- a/script/remove_sample_data.py
+++ b/script/remove_sample_data.py
@ -1,3 +1,10 @@
+# Add root project dir to the python path
+import os
+import sys
+
+parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
+sys.path.append(parent_dir)
+
 import ctypes
 import sqlalchemy
 from sqlalchemy import or_, event as sqlalchemy_event