I attached a cleaned up version that does avoid the duplicated code.

Attachment: basearchonly.conf
Description: Binary data

#!/usr/bin/python

# This program 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 2 of the License, or
# (at your option) any later version.
#
# This program 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 Library General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
#
# Copyright 2007 by Adel Gadllah

import re, os
from yum.plugins import TYPE_CORE

requires_api_version = '2.3'
plugin_type = TYPE_CORE


def exclude_hook(conduit):


	""" Only install i386 packages when told to do so """
	if os.uname()[-1] == 'x86_64':
		basearch(conduit, "x86","i?86$")
	
	""" Only install ppc64 packages when told to do so """	
	if os.uname()[-1] == 'ppc64':
		basearch(conduit, "ppc","ppc64$")

	""" Only install sparc64 packages when told to do so """	
	if os.uname()[-1] == 'sparc64':
		basearch(conduit, "sparc","sparc64$")


def basearch(conduit, barch, regex):
	
	exclude = []
	blacklist = []
	skip = 0
	conf , cmd = conduit.getCmdLine()
	packageList = conduit.getPackages()
	
	if cmd[0] != "install":
		return

	""" get blacklist from config file """	
	
	conflist = conduit.confString(barch, 'blacklist')
	if conflist:
		tmp = conflist.split(",")
		for confitem in tmp:
			blacklist.append(confitem.strip())
	
	""" decide which packages we want to exclude """	

	for userpkg in cmd:
		for skippkg in blacklist:
			if userpkg == skippkg:
				skip = 1
		if not re.search(regex, userpkg) and not skip:
			exclude.append(userpkg)
		skip = 0

	""" exclude the packages """	
	
	for pkg in packageList:
		if not re.search(regex, pkg.arch):
			continue
		for expkg in exclude:
			if expkg == pkg.name:
				conduit.delPackage(pkg)
				conduit.info(3, "--> excluded %s.%s" % (pkg.name, pkg.arch))	

_______________________________________________
Yum-devel mailing list
[email protected]
https://lists.dulug.duke.edu/mailman/listinfo/yum-devel

Reply via email to