X-Git-Url: https://osm.etsi.org/gitweb/?p=osm%2FPLA.git;a=blobdiff_plain;f=osm_pla%2Fplacement%2Fmznplacement.py;fp=osm_pla%2Fplacement%2Fmznplacement.py;h=3dcf0bc659a01167ffb4a8df6e919fefd035ba13;hp=56a448d170553310a3c5b729c824c49da1d3dec9;hb=de4f7820a050a7ed18aa93a298c5a59e379e752b;hpb=a6806e969b6cab9a9db1b44305db51d59f7759d2

diff --git a/osm_pla/placement/mznplacement.py b/osm_pla/placement/mznplacement.py
index 56a448d..3dcf0bc 100755
--- a/osm_pla/placement/mznplacement.py
+++ b/osm_pla/placement/mznplacement.py
@@ -18,7 +18,7 @@ import itertools
 
 import pymzn
 from jinja2 import Environment
-from jinja2.loaders import FileSystemLoader
+from jinja2.loaders import FileSystemLoader, PackageLoader, ChoiceLoader
 
 
 class MznPlacementConductor(object):
@@ -88,7 +88,8 @@ class MznModelGenerator(object):
     NsPlacementData objects. Uses jinja2 as templating language for the model
     '''
     default_j2_template = "osm_pla_dynamic_template.j2"
-    template_search_path = ['osm_pla/placement', '../placement', '/pla/osm_pla/placement']
+    template_search_path = ['osm_pla/placement', '../placement', '/pla/osm_pla/placement',
+                            './', '/usr/lib/python3/dist-packages/osm_pla/placement']
 
     def __init__(self, log):
         '''
@@ -111,7 +112,9 @@ class MznModelGenerator(object):
 
     def _load_jinja_template(self, template_name=default_j2_template):
         """loads the jinja template used for model generation"""
-        env = Environment(loader=FileSystemLoader(MznModelGenerator.template_search_path))
+        loader1 = FileSystemLoader(MznModelGenerator.template_search_path)
+        loader2 = PackageLoader('osm_pla', '.')
+        env = Environment(loader=ChoiceLoader([loader1, loader2]))
         return env.get_template(template_name)