diff options
| -rw-r--r-- | pym/portage/env/config.py | 58 | ||||
| -rw-r--r-- | pym/portage/tests/env/config/test_PackageKeywordsFile.py | 2 | ||||
| -rw-r--r-- | pym/portage/tests/env/config/test_PackageUseFile.py | 33 |
3 files changed, 90 insertions, 3 deletions
diff --git a/pym/portage/env/config.py b/pym/portage/env/config.py index 05bb49274..eccd35e08 100644 --- a/pym/portage/env/config.py +++ b/pym/portage/env/config.py @@ -11,7 +11,7 @@ class PackageKeywords(UserDict): A base class stub for things to inherit from; some people may want a database based package.keywords or something Internally dict has pairs of the form - {'cpv':['key1','key2','key3'...] + {'cpv':['keyword1','keyword2','keyword3'...] """ data = {} @@ -35,7 +35,7 @@ class PackageKeywordsFile(PackageKeywords): def __init__( self, filename ): self.fname = filename - def load(self, recursive=False): + def load(self, recursive): """ Package.keywords files have comments that begin with #. The entries are of the form: @@ -58,3 +58,57 @@ class PackageKeywordsFile(PackageKeywords): self.data[key].append(items) else: self.data[key] = items + +class PackageUse(UserDict): + """ + A base class stub for things to inherit from; some people may want a database based package.keywords or something + + Internally dict has pairs of the form + {'cpv':['flag1','flag22','flag3'...] + """ + + data = {} + + def iteritems(self): + return self.data.iteritems() + + def __hash__(self): + return hash(self.data) + + def __contains__(self, other): + return other in self.data + + def keys(self): + return self.data.keys() + +class PackageUseFile(PackageUse): + """ + Inherits from PackageUse; implements a file-based backend. Doesn't handle recursion yet. + """ + def __init__(self, filename): + self.fname = filename + + def load(self, recursive): + """ + Package.keywords files have comments that begin with #. + The entries are of the form: + >>> atom useflag1 useflag2 useflag3.. + useflags may optionally be negative with a minus sign (-) + >>> atom useflag1 -useflag2 useflag3 + """ + + if os.path.exists( self.fname ): + f = open(self.fname, 'rb') + for line in f: + if line.startswith('#'): + continue + split = line.split() + if len(split): + # Surprisingly this works for iterables of length 1 + # fex ['sys-apps/portage','foo','bar'] becomes {'sys-apps/portage':['foo','bar']} + key, items = split[0],split[1:] + # if they specify the same cpv twice; stack the values (append) instead of overwriting. + if key in self.data: + self.data[key].append(items) + else: + self.data[key] = items diff --git a/pym/portage/tests/env/config/test_PackageKeywordsFile.py b/pym/portage/tests/env/config/test_PackageKeywordsFile.py index f39ef96d7..6c5475881 100644 --- a/pym/portage/tests/env/config/test_PackageKeywordsFile.py +++ b/pym/portage/tests/env/config/test_PackageKeywordsFile.py @@ -19,7 +19,7 @@ class PackageKeywordsFileTestCase(TestCase): self.BuildFile() f = PackageKeywordsFile(self.fname) - f.load() + f.load(recursive=False) for cpv, keyword in f.iteritems(): self.assertEqual( cpv, self.cpv ) [k for k in keyword if self.assertTrue(k in self.keywords)] diff --git a/pym/portage/tests/env/config/test_PackageUseFile.py b/pym/portage/tests/env/config/test_PackageUseFile.py new file mode 100644 index 000000000..6dd76eeca --- /dev/null +++ b/pym/portage/tests/env/config/test_PackageUseFile.py @@ -0,0 +1,33 @@ +# test_PackageUseFile.py -- Portage Unit Testing Functionality +# Copyright 2007 Gentoo Foundation +# Distributed under the terms of the GNU General Public License v2 +# $Id: test_PackageUseFile.py 6182 2007-03-06 07:35:22Z antarus $ + +from portage.tests import TestCase +from portage.env.config import PackageUseFile + +class PackageUseFileTestCase(TestCase): + + fname = 'package.use' + cpv = 'sys-apps/portage' + useflags = ['cdrom', 'far', 'boo', 'flag', 'blat'] + + def testPackageUseLoad(self): + """ + A simple test to ensure the load works properly + """ + self.BuildFile() + f = PackageUseFile(self.fname) + f.load(recursive=False) + for cpv, use in f.iteritems(): + self.assertEqual( cpv, self.cpv ) + [flag for flag in use if self.assertTrue(flag in self.useflags)] + self.NukeFile() + + def BuildFile(self): + f = open(self.fname, 'wb') + f.write("%s %s" % (self.cpv, ' '.join(self.useflags))) + + def NukeFile(self): + import os + os.unlink(self.fname) |