From d89d07550fc52dd158ae00b5beb230bfc6db9016 Mon Sep 17 00:00:00 2001 From: Seth Vidal Date: Wed, 24 Oct 2012 00:21:24 -0400 Subject: [PATCH] first cut at yum module with enable/disable repo --- library/yum | 111 ++++++++++++++++++++++++++++++++++++++++------------ 1 file changed, 87 insertions(+), 24 deletions(-) diff --git a/library/yum b/library/yum index 013f20e34a..5f336d3201 100755 --- a/library/yum +++ b/library/yum @@ -49,10 +49,27 @@ options: required: false choices: [ "present", "latest", "absent" ] default: "present" + enablerepo: + description: + - repoid of repositories to enable for the install/update operation + - these repos will not persist beyond the transaction + - multiple repos separated with a , + required: false + default: null + aliases: [] + disablerepo: + description: + - repoid of repositories to disable for the install/update operation + - these repos will not persist beyond the transaction + - multiple repos separated with a , + required: false + default: null + aliases: [] + examples: - - code: "yum: name=httpd state=latest" - - code: "yum: name=httpd state=removed" - - code: "yum: name=httpd state=installed" + - code: yum name=httpd state=latest + - code: yum name=httpd state=removed + - code: yum name=httpd enablerepo=testing state=installed notes: [] # informational: requirements for nodes requirements: [ yum, rpm ] @@ -91,13 +108,18 @@ def po_to_nevra(po): else: return '%s-%s-%s.%s' % (po.name, po.version, po.release, po.arch) -def is_installed(module, repoq, pkgspec, conf_file, qf=def_qf): +def is_installed(module, repoq, pkgspec, conf_file, qf=def_qf, en_repos=[], dis_repos=[]): if not repoq: pkgs = [] try: my = yum_base(conf_file) + for rid in en_repos: + my.repos.enableRepo(rid) + for rid in dis_repos: + my.repos.disableRepo(rid) + e,m,u = my.rpmdb.matchPackageNames([pkgspec]) pkgs = e + m if not pkgs: @@ -121,13 +143,18 @@ def is_installed(module, repoq, pkgspec, conf_file, qf=def_qf): return [] -def is_available(module, repoq, pkgspec, conf_file, qf=def_qf): +def is_available(module, repoq, pkgspec, conf_file, qf=def_qf, en_repos=[], dis_repos=[]): if not repoq: pkgs = [] try: my = yum_base(conf_file) + for rid in en_repos: + my.repos.enableRepo(rid) + for rid in dis_repos: + my.repos.disableRepo(rid) + e,m,u = my.pkgSack.matchPackageNames([pkgspec]) pkgs = e + m if not pkgs: @@ -148,7 +175,7 @@ def is_available(module, repoq, pkgspec, conf_file, qf=def_qf): return [] -def is_update(module, repoq, pkgspec, conf_file, qf=def_qf): +def is_update(module, repoq, pkgspec, conf_file, qf=def_qf, en_repos=[], dis_repos=[]): if not repoq: @@ -158,6 +185,11 @@ def is_update(module, repoq, pkgspec, conf_file, qf=def_qf): try: my = yum_base(conf_file) + for rid in en_repos: + my.repos.enableRepo(rid) + for rid in dis_repos: + my.repos.disableRepo(rid) + pkgs = my.returnPackagesByDep(pkgspec) + my.returnInstalledPackagesByDep(pkgspec) if not pkgs: e,m,u = my.pkgSack.matchPackageNames([pkgspec]) @@ -184,13 +216,18 @@ def is_update(module, repoq, pkgspec, conf_file, qf=def_qf): return [] -def what_provides(module, repoq, req_spec, conf_file, qf=def_qf): +def what_provides(module, repoq, req_spec, conf_file, qf=def_qf, en_repos=[], dis_repos=[]): if not repoq: pkgs = [] try: my = yum_base(conf_file) + for rid in en_repos: + my.repos.enableRepo(rid) + for rid in dis_repos: + my.repos.disableRepo(rid) + pkgs = my.returnPackagesByDep(req_spec) + my.returnInstalledPackagesByDep(req_spec) if not pkgs: e,m,u = my.pkgSack.matchPackageNames([req_spec]) @@ -307,7 +344,7 @@ def run(command): return rc, out, err -def install(module, items, repoq, yum_basecmd, conf_file): +def install(module, items, repoq, yum_basecmd, conf_file, en_repos, dis_repos): res = {} res['results'] = [] @@ -328,7 +365,7 @@ def install(module, items, repoq, yum_basecmd, conf_file): nvra = local_nvra(spec) # look for them in the rpmdb - if is_installed(module, repoq, nvra, conf_file): + if is_installed(module, repoq, nvra, conf_file, en_repos, dis_repos): # if they are there, skip it continue pkg = spec @@ -340,7 +377,7 @@ def install(module, items, repoq, yum_basecmd, conf_file): # range requires or file-requires or pkgname :( else: # look up what pkgs provide this - pkglist = what_provides(module, repoq, spec, conf_file) + pkglist = what_provides(module, repoq, spec, conf_file, en_repos, dis_repos) if not pkglist: res['msg'] += "No Package matching '%s' found available, installed or updated" % spec module.fail_json(**res) @@ -350,7 +387,7 @@ def install(module, items, repoq, yum_basecmd, conf_file): found = False for this in pkglist: - if is_installed(module, repoq, this, conf_file): + if is_installed(module, repoq, this, conf_file, en_repos, dis_repos): found = True res['results'].append('%s providing %s is already installed' % (this, spec)) break @@ -381,7 +418,7 @@ def install(module, items, repoq, yum_basecmd, conf_file): module.exit_json(**res) -def remove(module, items, repoq, yum_basecmd, conf_file): +def remove(module, items, repoq, yum_basecmd, conf_file, en_repos, dis_repos): res = {} res['results'] = [] @@ -397,14 +434,14 @@ def remove(module, items, repoq, yum_basecmd, conf_file): pkg = spec # req or pkgname remove else: - pkglist = is_installed(module, repoq, spec, conf_file) + pkglist = is_installed(module, repoq, spec, conf_file, en_repos, dis_repos) if not pkglist: res['msg'] += "No Package matching '%s' found installed" % spec module.exit_json(**res) found = False for this in pkglist: - if is_installed(module, repoq, this, conf_file): + if is_installed(module, repoq, this, conf_file, en_repos, dis_repos): found = True if not found: @@ -431,7 +468,7 @@ def remove(module, items, repoq, yum_basecmd, conf_file): module.exit_json(**res) -def latest(module, items, repoq, yum_basecmd, conf_file): +def latest(module, items, repoq, yum_basecmd, conf_file, en_repos, dis_repos): res = {} res['results'] = [] @@ -448,12 +485,12 @@ def latest(module, items, repoq, yum_basecmd, conf_file): pkg = spec # dep/pkgname - find it else: - if is_installed(module, repoq, spec, conf_file): + if is_installed(module, repoq, spec, conf_file, en_repos, dis_repos): basecmd = 'update' else: basecmd = 'install' - pkglist = what_provides(module, repoq, spec, conf_file) + pkglist = what_provides(module, repoq, spec, conf_file, en_repos, dis_repos) if not pkglist: res['msg'] += "No Package matching '%s' found available, installed or updated" % spec res['failed']=True @@ -461,11 +498,11 @@ def latest(module, items, repoq, yum_basecmd, conf_file): nothing_to_do = True for this in pkglist: - if basecmd == 'install' and is_available(module, repoq, this, conf_file): + if basecmd == 'install' and is_available(module, repoq, this, conf_file, en_repos, dis_repos): nothing_to_do = False break - if basecmd == 'update' and is_update(module, repoq, this, conf_file): + if basecmd == 'update' and is_update(module, repoq, this, conf_file, en_repos, dis_repos): nothing_to_do = False break @@ -501,13 +538,14 @@ def latest(module, items, repoq, yum_basecmd, conf_file): module.exit_json(**res) -def ensure(module, state, pkgspec, conf_file): +def ensure(module, state, pkgspec, conf_file, enablerepo, disablerepo): # take multiple args comma separated items = pkgspec.split(',') yum_basecmd = [yumbin, '-d', '1', '-y'] + if not repoquery: repoq = None else: @@ -518,12 +556,33 @@ def ensure(module, state, pkgspec, conf_file): if repoq: repoq += ['-c', conf_file] + dis_repos =[] + en_repos = [] + if disablerepo: + dis_repos = disablerepo.split(',') + if enablerepo: + en_repos = enablerepo.split(',') + + for repoid in en_repos: + r_cmd = ['--enablerepo', repoid] + yum_basecmd.extend(r_cmd) + + if repoq: + repoq.extend(r_cmd) + + for repoid in dis_repos: + r_cmd = ['--disablerepo', repoid] + yum_basecmd.extend(r_cmd) + + if repoq: + repoq.extend(r_cmd) + if state in ['installed', 'present']: - install(module, items, repoq, yum_basecmd, conf_file) + install(module, items, repoq, yum_basecmd, conf_file, en_repos, dis_repos) elif state in ['removed', 'absent']: - remove(module, items, repoq, yum_basecmd, conf_file) + remove(module, items, repoq, yum_basecmd, conf_file, en_repos, dis_repos) elif state == 'latest': - latest(module, items, repoq, yum_basecmd, conf_file) + latest(module, items, repoq, yum_basecmd, conf_file, en_repos, dis_repos) # should be caught by AnsibleModule argument_spec return dict(changed=False, failed=True, results='', errors='unexpected state') @@ -546,6 +605,8 @@ def main(): name=dict(aliases=['pkg']), # removed==absent, installed==present, these are accepted as aliases state=dict(default='installed', choices=['absent','present','installed','removed','latest']), + enablerepo=dict(), + disablerepo=dict(), list=dict(), conf_file=dict(default=None), ), @@ -564,7 +625,9 @@ def main(): else: pkg = params['name'] state = params['state'] - res = ensure(module, state, pkg, params['conf_file']) + enablerepo = params.get('enablerepo', '') + disablerepo = params.get('disablerepo', '') + res = ensure(module, state, pkg, params['conf_file'], enablerepo, disablerepo) module.fail_json(msg="we should never get here unless this all failed", **res) # this is magic, see lib/ansible/module_common.py