# HG changeset patch
# User timeless <timeless@mozdev.org>
# Date 1452573929 0
# Node ID a7e3b72cf7562cd6167c537b8cbd93adb06a6527
# Parent  17b85d739b6270453914b9bac2354bb5af43c96f
setup: show how to set the module policy for imports

This is not technically needed, since mercurial.__version__
does not exist as a native module, but, without this style wrappings,
if something else had a native flavor, the module loader would get
upset.

In principle, the `env` object is trying to set HGMODULEPOLICY for
children, so, conceptually we should set it for this in-process
child.

diff -r 17b85d739b62 -r a7e3b72cf756 setup.py
--- a/setup.py	Wed Mar 09 15:47:01 2016 +0000
+++ b/setup.py	Tue Jan 12 04:45:29 2016 +0000
@@ -212,10 +212,17 @@
         f.write('version = "%s"\n' % version)
 
 try:
+    oldpolicy = os.environ.get('HGMODULEPOLICY', None)
+    os.environ['HGMODULEPOLICY'] = 'py'
     from mercurial import __version__
     version = __version__.version
 except ImportError:
     version = 'unknown'
+finally:
+    if oldpolicy is None:
+        del os.environ['HGMODULEPOLICY']
+    else:
+        os.environ['HGMODULEPOLICY'] = oldpolicy
 
 class hgbuild(build):
     # Insert hgbuildmo first so that files in mercurial/locale/ are found