diff options
Diffstat (limited to 'pbr/hooks.py')
-rw-r--r-- | pbr/hooks.py | 17 |
1 files changed, 13 insertions, 4 deletions
diff --git a/pbr/hooks.py b/pbr/hooks.py index f6aab78..d2375df 100644 --- a/pbr/hooks.py +++ b/pbr/hooks.py @@ -37,12 +37,12 @@ def __inject_parsed_file(value, func): value.extend(new_reqs) -def inject_requires(dist, attr, value): - __inject_parsed_file(value, requires.parse_requirements) +def inject_requires(require): + __inject_parsed_file(require, requires.parse_requirements) -def inject_dependency_links(dist, attr, value): - __inject_parsed_file(value, requires.parse_dependency_links) +def inject_dependency_links(require): + __inject_parsed_file(require, requires.parse_dependency_links) def inject_version(dist, attr, value): @@ -55,6 +55,15 @@ def inject_version(dist, attr, value): version = dist.metadata.version if version and version.startswith("#:"): + for require in 'install_requires', 'tests_require': + if require not in dist.__dict__: + dist.__dict__[require] = [] + inject_requires(dist.__dict__[require]) + print dist.__dict__[require] + if 'dependency_links' not in dist.__dict__: + dist.__dict__[require] = [] + inject_dependency_links(dist.__dict__['dependency_links']) + # Modify version number if len(version[2:]) > 0: (version_module, version_object) = version[2:].split(":") |