ansible/library/apt
2012-03-26 12:49:13 -07:00

152 lines
4.1 KiB
Python
Executable file

#!/usr/bin/python -tt
# (c) 2012, Flowroute LLC
# Written by Matthew Williams <matthew@flowroute.com>
# Based on yum module written by Seth Vidal <skvidal at fedoraproject.org>
#
# This module 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.
#
# This software 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 this software. If not, see <http://www.gnu.org/licenses/>.
#
import os
import sys
import apt
import shlex
import subprocess
try:
import json
except ImportError:
import simplejson as json
def run_apt(command):
try:
cmd = subprocess.Popen(command, shell=True,
stdout=subprocess.PIPE, stderr=subprocess.PIPE)
out, err = cmd.communicate()
except (OSError, IOError), e:
rc = 1
err = str(e)
out = ''
except:
rc = 1
err = traceback.format_exc()
out = ''
if out is None:
out = ''
if err is None:
err = ''
else:
rc = cmd.returncode
return rc, out, err
def ensure(state, pkgspec):
cache = apt.Cache()
cache.update()
cache.open(None)
try:
pkg = cache[pkgspec]
except:
msg = "No package matching '%s' is available" % pkgsepc
return {'changed': False, 'failed': True, 'msg': msg}
if state == 'installed':
#check if package is installed
if pkg.is_installed:
return {'changed': False, 'failed': False}
cmd = "apt-get -q -y install '%s'" % pkgspec
rc, out, err = run_apt(cmd)
#pkg.mark_install()
#cache.commit(apt.progress.base.OpProgress())
elif state == 'removed':
#check if package is installed
if not pkg.is_installed:
return {'changed': False, 'failed': False}
cmd = "apt-get -q -y remove '%s'" % pkgspec
rc, out, err = run_apt(cmd)
#pkg.mark_delete()
#cache.commit(apt.progress.base.OpProgress())
else:
return {'failed': True, 'msg': "Unknown state: %s" % state }
return {'changed': True, 'failed': False}
def update(args):
#generic update routine
pass
def remove_only(pkgspec):
# remove this pkg and only this pkg - fail if it will require more to remove
pass
def main():
# state=installed pkg=pkgspec
# state=removed pkg=pkgspec
if len(sys.argv) == 1:
msg = "the apt module requires arguments (-a)"
return 1, msg
argfile = sys.argv[1]
if not os.path.exists(argfile):
msg = "Argument file not found"
return 1, msg
args = open(argfile, 'r').read()
items = shlex.split(args)
if not len(items):
msg = "the apt module requires arguments (-a)"
return 1, msg
# if nothing else changes - it fails
results = { 'changed':False,
'failed':True,
'results':'',
'errors':'',
'msg':"; ".join(items) }
params = {}
for x in items:
try:
(k, v) = x.split("=", 1)
except ValueError:
msg = "invalid arguments: %s" % args
return 1, msg
params[k] = v
if 'state' in params:
if 'pkg' not in params:
results['msg'] = "No pkg specified"
else:
state = params['state']
pkgspec = params['pkg']
devnull = open(os.devnull, 'w')
results = ensure(state, pkgspec)
print json.dumps(results)
return 0, None
if __name__ == "__main__":
rc, msg = main()
if rc != 0: # something went wrong emit the msg
print json.dumps({
"failed" : bool(rc),
"msg" : msg
})
sys.exit(rc)