diff --git a/setup.py b/setup.py index e0d323bae6e3e887dcdc4b8b8142784fb33638fa..3bb3f3d7cbdd3b032aa2d8c89e7b085056c80b1f 100644 --- a/setup.py +++ b/setup.py @@ -8,14 +8,17 @@ import py_compile class APIGenBuild(build_py): def run(self): build_versions = ['3.0', '3.1', '3.2'] + default_version = '3.2' for version in build_versions: target_dir = os.path.join(self.build_lib, 'netdb_client', f"api{version.replace('.', '')}") self.mkpath(target_dir) environ = os.environ.copy() environ['NETDB_VERSION'] = version - gen = subprocess.run(['net-api-generator', 'python', f'--output-dir={target_dir}'], check=True, env=environ) + subprocess.run(['net-api-generator', 'python', f'--output-dir={target_dir}'], check=True, env=environ) target_dir = os.path.join(self.build_lib, 'netdb_client') - gen = subprocess.run(['net-api-generator', 'python', f'--output-dir={target_dir}'], check=True, env=environ) + environ = os.environ.copy() + environ['NETDB_VERSION'] = default_version + subprocess.run(['net-api-generator', 'python', f'--output-dir={target_dir}'], check=True, env=environ) build_py.run(self)