aboutsummaryrefslogtreecommitdiffstats
path: root/src/tools/ivigenerator/generate.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/tools/ivigenerator/generate.py')
-rwxr-xr-xsrc/tools/ivigenerator/generate.py12
1 files changed, 12 insertions, 0 deletions
diff --git a/src/tools/ivigenerator/generate.py b/src/tools/ivigenerator/generate.py
index ba6c213..f1cdb90 100755
--- a/src/tools/ivigenerator/generate.py
+++ b/src/tools/ivigenerator/generate.py
@@ -48,6 +48,7 @@ from path import Path
from qface.generator import FileSystem, Generator
from qface.watch import monitor
+from qface.utils import load_filters
import generator.builtin_config as builtin_config
import generator.global_functions as global_functions
@@ -124,6 +125,17 @@ def generate(tplconfig, moduleConfig, annotations, imports, src, dst):
global_functions.register_global_functions(generator)
register_filters(generator)
+ # Add the current path to the module search path
+ # This makes it possible to import our filters.py and friends
+ # from the plugin filters
+ sys.path.append(os.path.join(os.path.dirname(__file__)))
+
+ # Add a module specific extra filter if found
+ extra_filter_path = os.path.dirname(tplconfig) + '/{0}/filters.py'.format(os.path.basename(tplconfig))
+ if os.path.exists(extra_filter_path):
+ extra_filters = load_filters(Path(extra_filter_path))
+ generator.filters = extra_filters
+
validateSystem(srcFile, system)
# Make sure the config tag is available for all our symbols