diff options
Diffstat (limited to 'obsapi/repoflags.py')
| -rw-r--r-- | obsapi/repoflags.py | 96 |
1 files changed, 26 insertions, 70 deletions
diff --git a/obsapi/repoflags.py b/obsapi/repoflags.py index 3689711..c7badc7 100644 --- a/obsapi/repoflags.py +++ b/obsapi/repoflags.py @@ -3,7 +3,7 @@ from lxml import etree from collections import namedtuple -RepoFlag = namedtuple('RepoFlag', 'status repository arch') +RepoFlag = namedtuple('RepoFlag', 'flag status repository arch') valid_flag_types = ('build', 'publish', 'useforbuild', 'debuginfo') if sys.version_info >= (3, 0): @@ -12,9 +12,8 @@ if sys.version_info >= (3, 0): class RepoFlags(): - def __init__(self, flag_type=None, xml=None): + def __init__(self, xml): - self.flag_type = flag_type self.root = None if isinstance(xml, (str, unicode, bytes)): @@ -25,10 +24,19 @@ class RepoFlags(): elif isinstance(xml, etree._Element): self.root = xml - if self.flag_type is None and self.root is None: - self.flag_type = 'build' + def __str__(self): + lines = [] - self.__verify() + for flag_type in self.flag_types: + lines.append('{}:'.format(flag_type)) + for flag in self.get_flag(flag_type): + line = '' + for attribute in [flag.status, flag.repository, flag.arch]: + if attribute is not None: + line = ' '.join([line, attribute]) + lines.append(line) + + return '\n'.join(lines) def import_xml(self, xml): try: @@ -36,72 +44,20 @@ class RepoFlags(): except Exception as e: raise e - def __verify(self): - """ Make sure flag_type and xml data match. - """ - - # if self.flag_type set, ensure it matches the tag of the root xml - # element. If not matching, try to find the matching element within - # the xml tree and set root to that. If not found, set root to a new - # empty tree based on the flag_type - if self.flag_type is not None: - if self.flag_type not in valid_flag_types: - raise ValueError('Invalid repo flag type %s' % self.flag_type) - if self.root is None: - self.root = etree.Element(self.flag_type) - return - if self.flag_type == self.root.tag: - return - - element = self.root.find(self.flag_type) - if element is not None: - self.root = element - return - - # element matching flag_type not found in xml, create new element - self.root = etree.Element(self.flag_type) - return - else: - # If flag_type is not set, set it to the root element tag - flag_type = self.root.tag - if flag_type in valid_flag_types: - return - - # if root element is not a valid repo flag type, search for the - # first element that is a valid flag type and use it. - for flag_type in valid_flag_types: - element = self.root.find(flag_type) - if element is not None: - self.flag_type = flag_type - self.root = element - return - - self.flag_type = 'build' - self.root = etree.Element(self.flag_type) - @property def flags(self): for element in self.root: - if element.tag not in ['enable', 'disable']: - continue - status = element.tag - repo = element.get('repository', None) - arch = element.get('arch', None) - yield(RepoFlag(status=status, repository=repo, arch=arch)) - - def set(self, status, repo=None, arch=None): - if status not in ['enable', 'disable']: - raise ValueError('Invalid Flag status %s' % status) + for sub in element: + if sub.tag in ['enable', 'disable']: + flag = element.tag + status = sub.tag + repo = sub.get('repository', None) + arch = sub.get('arch', None) + yield(RepoFlag(flag=flag, status=status, repository=repo, arch=arch)) - element = etree.Element(status) - if repo: - element.set('repository', repo) - if arch: - element.set('arch', arch) - self.root.append(element) + def get_flag(self, flag_type): + return [f for f in self.flags if f.flag == flag_type] - def enable(self, repo=None, arch=None): - self.set('enable', repo, arch) - - def disable(self, repo=None, arch=None): - self.set('disable', repo, arch) + @property + def flag_types(self): + return set([f.flag for f in self.flags]) |