3 files changed, 47 insertions, 20 deletions
diff --git a/geneticone/backend/__init__.py b/geneticone/backend/__init__.py
index 16c3a2e..5238765 100644
--- a/geneticone/backend/__init__.py
+++ b/geneticone/backend/__init__.py
@@ -20,6 +20,7 @@ import portage
 
 # this is set to "var/lib/portage/world" by default - so we add the leading /
 portage.WORLD_FILE = "/"+portage.WORLD_FILE
+portage.settings = None
 
 # portage tree vars
 porttree = gentoolkit.porttree
diff --git a/geneticone/backend/flags.py b/geneticone/backend/flags.py
index 1251bfd..2da0a94 100644
--- a/geneticone/backend/flags.py
+++ b/geneticone/backend/flags.py
@@ -468,6 +468,26 @@ TESTING_PATH_IS_DIR = os.path.isdir(TESTING_PATH)
 newTesting = {}
 arch = ""
 
+def remove_new_testing (cpv):
+	if isinstance(cpv, package.Package):
+		cpv = cpv.get_cpv()
+	
+	try:
+		del newTesting[cpv]
+	except KeyError:
+		pass
+
+def new_testing_status (cpv):
+	if isinstance(cpv, package.Package):
+		cpv = cpv.get_cpv()
+
+	if cpv in newTesting:
+		for file, line in newTesting[cpv]:
+			if line == "-1": return False
+			else: return True
+
+	return None
+
 def set_testing (pkg, enable):
 	"""Enables the package for installing when it is marked as testing (~ARCH).
 	@param pkg: the package
@@ -484,12 +504,11 @@ def set_testing (pkg, enable):
 	if not cpv in newTesting: 
 		newTesting[cpv] = []
 
-	debug("arch: "+arch)
 	for file, line in newTesting[cpv]:
 		if (enable and line != "-1") or (not enable and line == "-1"):
 			newTesting[cpv].remove((file, line))
 
-	if (enable and (pkg.get_mask_status() % 3 == 0)) or (not enable and (pkg.get_mask_status() % 3 != 0)):
+	if (enable and not pkg.is_testing(allowed=True)) or (not enable and pkg.is_testing(allowed=True)):
 		return
 
 	if not enable:
diff --git a/geneticone/backend/package.py b/geneticone/backend/package.py
index fec346d..7b917a8 100644
--- a/geneticone/backend/package.py
+++ b/geneticone/backend/package.py
@@ -30,25 +30,33 @@ class Package (gentoolkit.Package):
 		if isinstance(cpv, gentoolkit.Package):
 			cpv = cpv.get_cpv()
 		gentoolkit.Package.__init__(self, cpv)
+		self._status = portage.getmaskingstatus(self.get_cpv(), settings = gentoolkit.settings)
 	
-	def get_mask_status(self):
-		"""Gets the numeric mask status of a package. The return value can be translated as a string when taking the following list of modes: [ "  ", " ~", " -", "M ", "M~", "M-" ]
+	def is_missing_keyword(self):
+		if "missing keyword" in self._status:
+			return True
+		return False
 
-		This method adapted from equery 0.1.4 
-		Original author: Karl Trygve Kalleberg <karltk@gentoo.org>
+	def is_testing(self, allowed = False):
+		testArch = "~" + self.get_settings("ARCH")
+		if not allowed:
+			if testArch in self.get_env_var("KEYWORDS").split():
+				return True
+			return False
+		else:
+			status = flags.new_testing_status(self.get_cpv())
+			if status == None:
+				if testArch+" keyword" in self._status:
+					return True
+				return False
+			else:
+				return status
+	
+	def set_testing(self, enable = True):
+		flags.set_testing(self, enable)
 
-		@returns: mask status
-		@rtype: int"""
-		
-		pkgmask = 0
-		if self.is_masked():
-			pkgmask = pkgmask + 3
-		keywords = self.get_env_var("KEYWORDS").split()
-		if "~" + gentoolkit.settings["ARCH"] in keywords:
-			pkgmask = pkgmask + 1
-		elif "-*" in keywords or "-" + gentoolkit.settings["ARCH"] in keywords:
-			pkgmask = pkgmask + 2
-		return pkgmask
+	def remove_new_testing(self):
+		flags.remove_new_testing(self.get_cpv())
 
 	def is_masked (self):
 		"""Returns True if either masked by package.mask or by profile.
@@ -62,8 +70,7 @@ class Package (gentoolkit.Package):
 			else:
 				debug("BUG in flags.new_masking_status. It returns "+status)
 		else:
-			status = portage.getmaskingstatus(self._cpv, settings = gentoolkit.settings)
-			if "profile" in status or "package.mask" in status:
+			if "profile" in self._status or "package.mask" in self._status:
 				return True
 			return False