diff options
author | Sebastian Koslowski <sebastian.koslowski@gmail.com> | 2018-09-22 14:21:23 -0700 |
---|---|---|
committer | Martin Braun <martin.braun@ettus.com> | 2019-02-17 11:18:05 -0800 |
commit | 8e999024124c0dc7cade628fdf0d662f20e706ad (patch) | |
tree | 63cda9d395f747e5956ad3d530597e2d25d6e591 /grc/core/blocks/_flags.py | |
parent | 2ee69a3e5f69f6cbd600bfd9202d9b42443569f6 (diff) |
grc: fix block flags
Diffstat (limited to 'grc/core/blocks/_flags.py')
-rw-r--r-- | grc/core/blocks/_flags.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/grc/core/blocks/_flags.py b/grc/core/blocks/_flags.py index dedbb5a485..54052b59c5 100644 --- a/grc/core/blocks/_flags.py +++ b/grc/core/blocks/_flags.py @@ -17,6 +17,8 @@ from __future__ import absolute_import +import six + class Flags(object): @@ -28,7 +30,11 @@ class Flags(object): HAS_PYTHON = 'python' HAS_CPP = 'cpp' - def __init__(self, flags): + def __init__(self, flags=None): + if flags is None: + flags = set() + if isinstance(flags, six.string_types): + flags = (f.strip() for f in flags.replace(',', '').split()) self.data = set(flags) def __getattr__(self, item): @@ -37,5 +43,8 @@ class Flags(object): def __contains__(self, item): return item in self.data + def __str__(self): + return ', '.join(self.data) + def set(self, *flags): self.data.update(flags) |