more PY2

Author: albu-diku

mig/shared/templates/__init__.py

@@ -25,15 +25,21 @@
 # -- END_HEADER ---
 #
 
-from collections import ChainMap
 import errno
 from jinja2 import meta as jinja2_meta, select_autoescape, Environment, \
     FileSystemLoader, FileSystemBytecodeCache
 import os
 import weakref
 
+from mig.shared.compat import PY2
 from mig.shared.defaults import MIG_BASE
 
+
+if PY2:
+    from chainmap import ChainMap
+else:
+    from collections import ChainMap
+
 TEMPLATES_DIR = os.path.abspath(os.path.dirname(__file__))
 TEMPLATES_CACHE_DIR = os.path.join(TEMPLATES_DIR, '__jinja__')