# HG changeset patch
# User Durham Goode <durham@fb.com>
# Date 1428344480 25200
# Node ID cf846d47bb7e8bf0f354590833fb76803f528774
# Parent  3625d006e81b5d860609533f90b3167bc9a9dd21
setup: allow including inhibit in the build

Adds the INCLUDE_INHIBIT environment variable to allow including the inhibit
extension in the build.

diff --git a/setup.py b/setup.py
--- a/setup.py
+++ b/setup.py
@@ -1,6 +1,7 @@
 # Copied from histedit setup.py
 # Credit to Augie Fackler <durin42@gmail.com>
 
+import os
 from distutils.core import setup
 from os.path import dirname, join
 
@@ -14,6 +15,13 @@
           if "'" in line:
             return line.split("'")[1]
 
+py_modules = [
+    'hgext.evolve',
+]
+
+if os.environ.get('INCLUDE_INHIBIT'):
+    py_modules.append('hgext.inhibit')
+
 setup(
     name='hg-evolve',
     version=get_version('hgext/evolve.py'),
@@ -25,5 +33,5 @@
     long_description=open('README').read(),
     keywords='hg mercurial',
     license='GPLv2+',
-    py_modules=['hgext.evolve'],
+    py_modules=py_modules
 )