diff --git a/pipenv/__init__.py b/pipenv/__init__.py index 1506235e..57648e8a 100644 --- a/pipenv/__init__.py +++ b/pipenv/__init__.py @@ -15,6 +15,7 @@ v_path = os.path.abspath(os.path.sep.join([os.path.dirname(os.path.realpath(__fi sys.path.insert(0, v_path) from .cli import cli +from . import resolver if __name__ == '__main__': cli() diff --git a/pipenv/resolver.py b/pipenv/resolver.py index 5577ad8c..6d49dced 100644 --- a/pipenv/resolver.py +++ b/pipenv/resolver.py @@ -13,13 +13,7 @@ for _dir in ('vendor', 'patched', '..'): def which(*args, **kwargs): return sys.executable - -def resolve(packages, pre, sources, verbose, clear): - import pipenv.utils - return pipenv.utils.resolve_deps(packages, which, project=project, pre=pre, sources=sources, clear=clear, verbose=verbose) - - -if __name__ == '__main__': +def main(): is_verbose = '--verbose' in ' '.join(sys.argv) do_pre = '--pre' in ' '.join(sys.argv) do_clear = '--clear' in ' '.join(sys.argv) @@ -53,6 +47,10 @@ if __name__ == '__main__': project = pipenv.core.project + def resolve(packages, pre, sources, verbose, clear): + import pipenv.utils + return pipenv.utils.resolve_deps(packages, which, project=project, pre=pre, sources=sources, clear=clear, verbose=verbose) + results = resolve(packages, pre=do_pre, sources=project.sources, verbose=is_verbose, clear=do_clear) @@ -62,3 +60,7 @@ if __name__ == '__main__': print(json.dumps(results)) else: print(json.dumps([])) + + +if __name__ == '__main__': + main() diff --git a/setup.py b/setup.py index 0df4cc01..cd4e1ebe 100644 --- a/setup.py +++ b/setup.py @@ -117,7 +117,8 @@ setup( entry_points={ 'console_scripts': [ 'pipenv=pipenv:cli', - 'pewtwo=pipenv.patched.pew.pew:pew' + 'pewtwo=pipenv.patched.pew.pew:pew', + 'pipenv-resolver=pipenv.resolver:main' ], }, install_requires=required,