Merge pull request #24302 from cpennington/prioritize-access-denied-messages

Prioritize access denied messages
This commit is contained in:
Calen Pennington
2020-06-25 10:15:57 -04:00
committed by GitHub
4 changed files with 80 additions and 65 deletions

View File

@@ -5,6 +5,7 @@ Transformers.
from abc import abstractmethod
import functools
class BlockStructureTransformer(object):
@@ -177,7 +178,8 @@ class FilteringTransformerMixin(BlockStructureTransformer):
transform_block_filters calls will be combined and used in a single
tree traversal.
"""
block_structure.filter_topological_traversal(self.transform_block_filters(usage_info, block_structure))
filters = self.transform_block_filters(usage_info, block_structure)
block_structure.filter_topological_traversal(combine_filters(block_structure, filters))
@abstractmethod
def transform_block_filters(self, usage_info, block_structure):
@@ -209,3 +211,19 @@ class FilteringTransformerMixin(BlockStructureTransformer):
transformer, that is to be transformed in place.
"""
raise NotImplementedError
def combine_filters(block_structure, filters):
return functools.reduce(
_filter_chain,
filters,
block_structure.create_universal_filter()
)
def _filter_chain(accumulated, additional):
"""
Given two functions that take a block_key and return a boolean, yield
a function that takes a block key, and 'ands' the functions together
"""
return lambda block_key: accumulated(block_key) and additional(block_key)

View File

@@ -7,7 +7,7 @@ import functools
from logging import getLogger
from .exceptions import TransformerDataIncompatible, TransformerException
from .transformer import FilteringTransformerMixin
from .transformer import FilteringTransformerMixin, combine_filters
from .transformer_registry import TransformerRegistry
logger = getLogger(__name__) # pylint: disable=C0103
@@ -131,20 +131,9 @@ class BlockStructureTransformers(object):
for transformer in self._transformers['supports_filter']:
filters.extend(transformer.transform_block_filters(self.usage_info, block_structure))
combined_filters = functools.reduce(
self._filter_chain,
filters,
block_structure.create_universal_filter()
)
combined_filters = combine_filters(block_structure, filters)
block_structure.filter_topological_traversal(combined_filters)
def _filter_chain(self, accumulated, additional):
"""
Given two functions that take a block_key and return a boolean, yield
a function that takes a block key, and 'ands' the functions together
"""
return lambda block_key: accumulated(block_key) and additional(block_key)
def _transform_without_filters(self, block_structure):
"""
Transforms the given block_structure using the transform