diff --git a/ryu/hooks.py b/ryu/hooks.py index d7a2a8ddf..8a1158941 100644 --- a/ryu/hooks.py +++ b/ryu/hooks.py @@ -17,7 +17,7 @@ import os import sys -from setuptools.command import easy_install +from setuptools.command.easy_install import ScriptWriter from ryu import version @@ -33,7 +33,7 @@ def save_orig(): """Save original easy_install.get_script_args. This is necessary because pbr's setup_hook is sometimes called before ours.""" - _main_module()._orig_get_script_args = easy_install.get_script_args + _main_module()._orig_get_script_args = ScriptWriter.get_args def setup_hook(config): @@ -57,7 +57,7 @@ def my_get_script_args(*args, **kwargs): return _main_module()._orig_get_script_args(*args, **kwargs) packaging.override_get_script_args = my_get_script_args - easy_install.get_script_args = my_get_script_args + ScriptWriter.get_args = my_get_script_args # another hack to allow setup from tarball. orig_get_version = packaging.get_version