diff options
Diffstat (limited to 'lib/python2.7/site-packages/setoolsgui/setools/categoryquery.py')
-rw-r--r-- | lib/python2.7/site-packages/setoolsgui/setools/categoryquery.py | 55 |
1 files changed, 55 insertions, 0 deletions
diff --git a/lib/python2.7/site-packages/setoolsgui/setools/categoryquery.py b/lib/python2.7/site-packages/setoolsgui/setools/categoryquery.py new file mode 100644 index 0000000..d4d7c4c --- /dev/null +++ b/lib/python2.7/site-packages/setoolsgui/setools/categoryquery.py @@ -0,0 +1,55 @@ +# Copyright 2015, Tresys Technology, LLC +# +# This file is part of SETools. +# +# SETools is free software: you can redistribute it and/or modify +# it under the terms of the GNU Lesser General Public License as +# published by the Free Software Foundation, either version 2.1 of +# the License, or (at your option) any later version. +# +# SETools 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 Lesser General Public License for more details. +# +# You should have received a copy of the GNU Lesser General Public +# License along with SETools. If not, see +# <http://www.gnu.org/licenses/>. +# +import logging + +from . import compquery +from . import mixins + + +class CategoryQuery(mixins.MatchAlias, compquery.ComponentQuery): + + """ + Query MLS Categories + + Parameter: + policy The policy to query. + + Keyword Parameters/Class attributes: + name The name of the category to match. + name_regex If true, regular expression matching will + be used for matching the name. + alias The alias name to match. + alias_regex If true, regular expression matching + will be used on the alias names. + """ + + def results(self): + """Generator which yields all matching categories.""" + self.log.info("Generating results from {0.policy}".format(self)) + self.log.debug("Name: {0.name!r}, regex: {0.name_regex}".format(self)) + self.log.debug("Alias: {0.alias}, regex: {0.alias_regex}".format(self)) + + for cat in self.policy.categories(): + if not self._match_name(cat): + continue + + if not self._match_alias(cat): + continue + + yield cat |