Reimplementation of quickpkg.py

quickpkg.py (text/plain), 3.18 KB, created by Martin Parm on 2007-02-02 13:58:14 UTC

#!/usr/bin/python
#
# Copyright 2006 Martin Parm
# Distributed under the terms of the GNU General Public License v2
#
# $Header$
# Author: Martin Parm <[e-mail]>

__author__		= "Martin Parm"
__email__		= "[e-mail]"
__version__		= "1.2.0"
__productname__	= "quickpkg"
__description__	= ""

import pwd, grp
import sys

def usage():
	print  "QUICKPKG ver %s" % __version__
	print  "USAGE: quickpkg <list of pkgs>"
	print  "    a pkg can be of the form:"
	print  "        - single depend-type atom ..."
	print  "              if portage can emerge it, quickpkg can make a package"
	print  "              for exact definitions of depend atoms, see ebuild(5)"
	print 
	print  "EXAMPLE:"
	print  "    quickpkg apache"
	print  "        package up apache, all versions of apache installed"
	print  "    quickpkg =apache-1.3.27-r1"
	print  "        package up apache, just version 1.3.27-r1"
	sys.exit(1)

def quickpkg(pkg):
	print ("Building package for %s" % pkg).ljust(60,'.'),
	sys.stdout.flush()		
	
	s=pkg.split('/')
	db=portage.dblink(s[0], s[1], portage.root, mysettings=portage.settings)
	if not db.exists():
		print "Not found!"
		return
	if not os.path.exists('/usr/portage/packages/All'):
		os.makedirs('/usr/portage/packages/All')
	if not os.path.exists('/usr/portage/packages/%s' % s[0]):
		os.makedirs('/usr/portage/packages/%s' % s[0])
	binpkg=os.path.join('/usr/portage/packages/All', s[1]+'.tbz2')

	contents=db.getcontents()
	tarball=tarfile.open(binpkg+'.$$', "w:bz2")
	for item in sorted(contents.keys()):
		if not os.path.exists(item):
			# Ignore ALL missing packages for now (due to bug #16162)
			# FIXME: Remove this when #16162 has been solved
			continue
				
		tarinfo=tarball.gettarinfo(item, './'+item)
		try:
			tarinfo.uname=unames[tarinfo.uid]
		except KeyError:
			tarinfo.uname=str(tarinfo.uid)
			unames[tarinfo.uid]=str(tarinfo.uid)
			
		try:
			tarinfo.gname=gnames[tarinfo.gid]
		except KeyError:
			tarinfo.gname=str(tarinfo.gid)
			gnames[tarinfo.gid]=str(tarinfo.gid)
			
		if contents[item][0]=='dir':
			tarinfo.linkname=''
			tarinfo.type=tarfile.DIRTYPE
			tarball.addfile(tarinfo)
		elif contents[item][0]=='obj':
			tarinfo.linkname=''
			tarinfo.type=tarfile.REGTYPE
			tarball.addfile(tarinfo, file(item))
		else:
			tarball.addfile(tarinfo)
	tarball.close()
	xpak.tbz2(binpkg+'.$$').recompose(db.getpath())
	os.rename(binpkg+'.$$', binpkg)
	if not os.path.exists(
		os.path.join('/usr/portage/packages/%s' % s[0], os.path.basename(binpkg))
	):
		os.symlink(
			os.path.join('../All', s[1]+'.tbz2'),
			os.path.join('/usr/portage/packages/%s' % s[0], s[1]+'.tbz2')
		)
	print "DONE!"
	return
	
if __name__ == '__main__':
	if len(sys.argv)<2:
		usage()

	import portage
	import xpak
	import tarfile
	import os
	
	unames=dict([(entry[2],entry[0]) for entry in pwd.getpwall()])
	gnames=dict([(entry[2],entry[0]) for entry in grp.getgrall()])
	
	# Try to match up all packages
	pkgs=[]
	for arg in sys.argv[1:]:
		try:
			matches=portage.db[portage.root]["vartree"].dbapi.match(arg)
			if not matches:
				print "Error: Could not find anything to match '%s'" % arg
				sys.exit(2)
			pkgs+=matches
		except	KeyError, e:
			print e
			sys.exit(1)
			
	# Pack everything
	for pkg in pkgs:
		quickpkg(pkg)