class TestWrapperState(unittest.TestCase):
def setUp(self):
self.wrapper_state = WrapperState({
'header-one': {
'element': 'h1'
},
'unstyled': {
'element': 'div'
},
})
def test_init(self):
self.assertIsInstance(self.wrapper_state, WrapperState)
def test_element_for_text(self):
self.assertEqual(
DOM.get_text_content(
self.wrapper_state.element_for({
'key': '5s7g9',
'text': 'Header',
'type': 'header-one',
'depth': 0,
'inlineStyleRanges': [],
'entityRanges': []
})), '')
def test_element_for_tag(self):
self.assertEqual(
DOM.get_tag_name(
self.wrapper_state.element_for({
'key': '5s7g9',
'text': 'Header',
'type': 'header-one',
'depth': 0,
'inlineStyleRanges': [],
'entityRanges': []
})), 'h1')
def test_element_for_raises(self):
with self.assertRaises(BlockException):
self.wrapper_state.element_for({
'key': '5s7g9',
'text': 'Header',
'type': 'header-two',
'depth': 0,
'inlineStyleRanges': [],
'entityRanges': []
})
def test_to_string_empty(self):
self.assertEqual(self.wrapper_state.to_string(), '')
def test_to_string_elts(self):
self.wrapper_state.element_for({
'key': '5s7g9',
'text': 'Header',
'type': 'header-one',
'depth': 0,
'inlineStyleRanges': [],
'entityRanges': []
})
self.assertEqual(self.wrapper_state.to_string(), '<h1></h1>')
def test_str_empty(self):
self.assertEqual(str(self.wrapper_state), '<WrapperState: >')
def test_str_elts(self):
self.wrapper_state.element_for({
'key': '5s7g9',
'text': 'Header',
'type': 'header-one',
'depth': 0,
'inlineStyleRanges': [],
'entityRanges': []
})
self.assertEqual(str(self.wrapper_state), '<WrapperState: <h1></h1>>')
def test_map_element_options_full(self):
self.assertEqual(Options.map([
'ul',
{
'className': 'bullet-list'
},
]), [
'ul',
{
'className': 'bullet-list'
},
])
def test_map_element_options_half(self):
self.assertEqual(Options.map([
'ul',
]), [
'ul',
{},
])
def test_map_element_options_simplest(self):
self.assertEqual(Options.map('ul'), [
'ul',
{},
])
class HTML:
"""
Entry point of the exporter. Combines entity, wrapper and style state
to generate the right HTML nodes.
"""
def __init__(self, config=None):
if config is None:
config = {}
self.entity_decorators = config.get('entity_decorators', {})
self.block_map = config.get('block_map', BLOCK_MAP)
self.style_map = config.get('style_map', STYLE_MAP)
def render(self, content_state):
"""
Starts the export process on a given piece of content state.
"""
self.wrapper_state = WrapperState(self.block_map)
self.style_state = StyleState(self.style_map)
entity_map = content_state.get('entityMap', {})
for block in content_state.get('blocks', []):
self.render_block(block, entity_map)
self.wrapper_state.clean_up()
return self.wrapper_state.to_string()
def render_block(self, block, entity_map):
element = self.wrapper_state.element_for(block)
entity_state = EntityState(self.entity_decorators, entity_map)
for (text, commands) in self.build_command_groups(block):
for command in commands:
entity_state.apply(command)
self.style_state.apply(command)
style_node = self.style_state.create_node(text)
entity_state.render_entitities(element, style_node)
def build_command_groups(self, block):
"""
Creates block modification commands, grouped by start index,
with the text to apply them on.
"""
text = block.get('text')
commands = self.build_commands(block)
# Tried using itertools.tee but for some reason that failed. Oh well.
grouped = Command.grouped_by_index(commands)
listed = list(Command.grouped_by_index(commands))
sliced = []
i = 0
for start_index, commands in grouped:
next_group = listed[i + 1] if i + 1 < len(listed) else False
stop_index = next_group[0] if next_group else 0
sliced.append((text[start_index:stop_index], list(commands)))
i += 1
return sliced
def build_commands(self, block):
"""
Build all of the manipulation commands for a given block.
- One pair to set the text.
- Multiple pairs for styles.
- Multiple pairs for entities.
"""
text_commands = Command.start_stop('text', 0, len(block.get('text')))
style_commands = self.build_style_commands(block)
entity_commands = self.build_entity_commands(block)
return text_commands + style_commands + entity_commands
def build_style_commands(self, block):
ranges = block.get('inlineStyleRanges', [])
return Command.from_ranges(ranges, 'inline_style', 'style')
def build_entity_commands(self, block):
ranges = block.get('entityRanges', [])
return Command.from_ranges(ranges, 'entity', 'key')
class TestWrapperState(unittest.TestCase):
def setUp(self):
self.wrapper_state = WrapperState(block_map)
def test_init(self):
self.assertIsInstance(self.wrapper_state, WrapperState)
def test_element_for_text(self):
self.assertEqual(self.wrapper_state.element_for({
'key': '5s7g9',
'text': 'Header',
'type': 'header-one',
'depth': 0,
'inlineStyleRanges': [],
'entityRanges': []
}).text, None)
def test_element_for_tag(self):
self.assertEqual(self.wrapper_state.element_for({
'key': '5s7g9',
'text': 'Header',
'type': 'header-one',
'depth': 0,
'inlineStyleRanges': [],
'entityRanges': []
}).tag, 'h1')
def test_element_for_raises(self):
with self.assertRaises(BlockException):
self.wrapper_state.element_for({
'key': '5s7g9',
'text': 'Header',
'type': 'header-two',
'depth': 0,
'inlineStyleRanges': [],
'entityRanges': []
})
def test_to_string_empty(self):
self.assertEqual(self.wrapper_state.to_string(), '')
def test_to_string_elts(self):
self.wrapper_state.element_for({
'key': '5s7g9',
'text': 'Header',
'type': 'header-one',
'depth': 0,
'inlineStyleRanges': [],
'entityRanges': []
})
self.assertEqual(self.wrapper_state.to_string(), '<h1></h1>')
def test_str_empty(self):
self.assertEqual(str(self.wrapper_state), '<WrapperState: >')
def test_str_elts(self):
self.wrapper_state.element_for({
'key': '5s7g9',
'text': 'Header',
'type': 'header-one',
'depth': 0,
'inlineStyleRanges': [],
'entityRanges': []
})
self.assertEqual(str(self.wrapper_state), '<WrapperState: <h1></h1>>')