symbol gen script

2 files changed, 6 insertions, 3 deletions

diff --git a/build.py b/build.py
index d096792..91a66fb 100644
--- a/build.py
+++ b/build.py
@@ -1,17 +1,20 @@
 from fbs.cmdline import command
 from os.path import dirname
 import subprocess
-
+import sys
 import fbs.cmdline
 import fbs
 
+sys.path.append('src/main/python/')
+
 @command
 def compileResources():
     subprocess.call("pyrcc5 -o src/main/python/resources/resources.py src/main/ui/resources.rcc", shell=True)
 
 @command
 def symbolGen():
-    exec(open('src/main/python/utils/custom.py').read())
+    from utils import custom
+    custom.main()
     
 @command
 def build():
diff --git a/src/main/python/utils/custom.py b/src/main/python/utils/custom.py
index fedecdb..2c8ffc6 100644
--- a/src/main/python/utils/custom.py
+++ b/src/main/python/utils/custom.py
@@ -279,7 +279,7 @@ class OutputBox(QTextEdit):
         self.setFontWeight(10)
         self.setHtml(text)
         
-if __name__ == '__main__':      # 1. Instantiate ApplicationContext
+def main():     # 1. Instantiate ApplicationContext
     #if app is launched directly
     from .app import app
     import sys