From 311e90473db8c4c5507700ad7ccaf7ea3a8ea6dc Mon Sep 17 00:00:00 2001 From: Matt Wright Date: Tue, 7 Aug 2012 16:38:04 -0400 Subject: [PATCH 1/2] add pip module --- pip | 182 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 182 insertions(+) create mode 100755 pip diff --git a/pip b/pip new file mode 100755 index 00000000000..9c277295fc0 --- /dev/null +++ b/pip @@ -0,0 +1,182 @@ +#!/usr/bin/python + +PIP = None +VIRTUALENV = None +ENV = None + + +def _get_full_name(name, version=None): + return name if version is None else name + '==' + version + + +def _find_pip(): + paths = ['/usr/local/bin', '/usr/bin'] + + if ENV: + paths = [os.path.join(ENV, 'bin')] + paths + + for p in paths: + pe = p + '/pip' + if os.path.exists(pe): + return pe + + module.fail_json(msg='pip is not installed') + + +def _find_virtualenv(): + paths = ['/usr/local/bin', '/usr/bin'] + + for p in paths: + ve = p + '/virtualenv' + if os.path.exists(ve): + return ve + + module.fail_json(msg='virtualenv is not installed') + + +def _ensure_virtualenv(): + if os.path.exists(os.path.join(ENV, 'bin', 'activate')): + return 0, '', '' + else: + return _run('%s %s' % (VIRTUALENV, ENV)) + + +def _is_package_installed(name, version=None): + rc, status_stdout, status_stderr = _run('%s freeze' % PIP) + return _get_full_name(name, version).lower() in status_stdout.lower() + + +def _run(cmd): + # returns (rc, stdout, stderr) from shell command + process = subprocess.Popen(cmd, stdout=subprocess.PIPE, + stderr=subprocess.PIPE, shell=True) + stdout, stderr = process.communicate() + return (process.returncode, stdout, stderr) + + +def main(): + arg_spec = dict( + state=dict(default='present', choices=['absent', 'present', 'latest']), + name=dict(default=None, required=False), + version=dict(default=None, required=False), + requirements=dict(default=None, required=False), + virtualenv=dict(default=None, required=False) + ) + + module = AnsibleModule(argument_spec=arg_spec) + + global PIP + global VIRTUALENV + global ENV + + rc = 0 + err = '' + out = '' + + ENV = module.params['virtualenv'] + PIP = _find_pip() + + if ENV: + VIRTUALENV = _find_virtualenv() + + rc_venv, out_venv, err_venv = _ensure_virtualenv() + + rc += rc_venv + out += out_venv + err += err_venv + + state = module.params['state'] + name = module.params['name'] + version = module.params['version'] + requirements = module.params['requirements'] + command_map = dict(present='install', absent='uninstall', latest='install') + + if state == 'latest' and version is not None: + module.fail_json(msg='If `state` is set to `latest` the `version` ' + 'paramater must not be specified.') + + if state == 'latest' and requirements is not None: + module.fail_json(msg='If `state` is set to `latest` the `requirements` ' + 'paramater must not be specified.') + + if name is not None and '==' in name: + module.fail_json(msg='It looks like you specified the version number ' + 'in the library name. Use the `version` parameter ' + 'to specify versio instead') + + if version is not None and name is None: + module.fail_json(msg='The `version` parameter must be used with the ' + '`name` parameter and not with the `requirements` ' + 'paramter') + + if name is None and requirements is None: + module.fail_json(msg='You must specify a python library name via ' + 'the `name` parameter or a requirements file via ' + 'the `requirements` paramter') + + if name and requirements: + module.fail_json(msg='Both `name` and `requirements` were specified. ' + 'Specify only the python library name via the ' + '`name` parameter or a requirements file via the ' + '`requirements` parameter') + + cmd = None + installed = None + + if requirements: + cmd = '%s %s -r %s --use-mirrors' % (PIP, command_map[state], requirements) + rc_pip, out_pip, err_pip = _run(cmd) + + rc += rc_pip + out += out_pip + err += err_pip + + def did_install(out): + return 'Successfully installed' in out + + changed = ((did_install(out) and state == 'present') or + (not did_install(out) and state == 'absent')) + + if name and state == 'latest': + cmd = '%s %s %s --upgrade' % (PIP, command_map[state], name) + + rc_pip, out_pip, err_pip = _run(cmd) + + rc += rc_pip + out += out_pip + err += err_pip + + changed = 'Successfully installed' in out_pip + + elif name: + installed = _is_package_installed(name, version) + + changed = ((installed and state == 'absent') or + (not installed and state == 'present')) + + if changed: + full_name = _get_full_name(name, version) if state == 'present' else name + + cmd = '%s %s %s' % (PIP, command_map[state], full_name) + + if state == 'absent': + cmd = cmd + ' -y' + else: + cmd = cmd + ' --use-mirrors' + + rc_pip, out_pip, err_pip = _run(cmd) + + rc += rc_pip + out += out_pip + err += err_pip + + if rc != 0: + module.fail_json(msg=err, cmd=cmd) + + module.exit_json(changed=changed, cmd=cmd, name=name, version=version, + state=state, requirements=requirements, virtualenv=ENV) + +# this is magic, see lib/ansible/module_common.py +#<> + +main() From a5ec3233d56b1325f6e48078d275ea5de0ce31d4 Mon Sep 17 00:00:00 2001 From: Matt Wright Date: Wed, 8 Aug 2012 10:35:07 -0400 Subject: [PATCH 2/2] Updates per ansible/ansible#796 --- pip | 40 ++++++++++++++++++++++++++++------------ 1 file changed, 28 insertions(+), 12 deletions(-) diff --git a/pip b/pip index 9c277295fc0..62cfa7b5926 100755 --- a/pip +++ b/pip @@ -1,4 +1,23 @@ #!/usr/bin/python +# -*- coding: utf-8 -*- + +# (c) 2012, Matt Wright +# +# This file is part of Ansible +# +# Ansible is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# Ansible is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with Ansible. If not, see . +# PIP = None VIRTUALENV = None @@ -46,6 +65,10 @@ def _is_package_installed(name, version=None): return _get_full_name(name, version).lower() in status_stdout.lower() +def _did_install(out): + return 'Successfully installed' in out + + def _run(cmd): # returns (rc, stdout, stderr) from shell command process = subprocess.Popen(cmd, stdout=subprocess.PIPE, @@ -65,10 +88,6 @@ def main(): module = AnsibleModule(argument_spec=arg_spec) - global PIP - global VIRTUALENV - global ENV - rc = 0 err = '' out = '' @@ -93,16 +112,16 @@ def main(): if state == 'latest' and version is not None: module.fail_json(msg='If `state` is set to `latest` the `version` ' - 'paramater must not be specified.') + 'parameter must not be specified.') if state == 'latest' and requirements is not None: module.fail_json(msg='If `state` is set to `latest` the `requirements` ' - 'paramater must not be specified.') + 'parameter must not be specified.') if name is not None and '==' in name: module.fail_json(msg='It looks like you specified the version number ' 'in the library name. Use the `version` parameter ' - 'to specify versio instead') + 'to specify version instead') if version is not None and name is None: module.fail_json(msg='The `version` parameter must be used with the ' @@ -131,11 +150,8 @@ def main(): out += out_pip err += err_pip - def did_install(out): - return 'Successfully installed' in out - - changed = ((did_install(out) and state == 'present') or - (not did_install(out) and state == 'absent')) + changed = ((_did_install(out) and state == 'present') or + (not _did_install(out) and state == 'absent')) if name and state == 'latest': cmd = '%s %s %s --upgrade' % (PIP, command_map[state], name)