class TestStyleState(unittest.TestCase):
def setUp(self):
self.style_state = StyleState(style_map)
def test_init(self):
self.assertIsInstance(self.style_state, StyleState)
def test_apply_start_inline_style(self):
self.assertEqual(self.style_state.styles, [])
self.style_state.apply(Command('start_inline_style', 0, 'ITALIC'))
self.assertEqual(self.style_state.styles, ['ITALIC'])
def test_apply_stop_inline_style(self):
self.style_state.apply(Command('start_inline_style', 0, 'ITALIC'))
self.style_state.apply(Command('stop_inline_style', 0, 'ITALIC'))
self.assertEqual(self.style_state.styles, [])
def test_is_unstyled_default(self):
self.assertEqual(self.style_state.is_unstyled(), True)
def test_is_unstyled_styled(self):
self.style_state.apply(Command('start_inline_style', 0, 'ITALIC'))
self.assertEqual(self.style_state.is_unstyled(), False)
def test_get_style_value_empty(self):
self.assertEqual(self.style_state.get_style_value(), '')
def test_get_style_value_single(self):
self.style_state.apply(Command('start_inline_style', 0, 'ITALIC'))
self.assertEqual(self.style_state.get_style_value(), '')
def test_get_style_value_multiple(self):
self.style_state.apply(Command('start_inline_style', 0, 'HIGHLIGHT'))
self.assertEqual(self.style_state.get_style_value(), 'text-decoration: underline;')
def test_create_node_unstyled(self):
self.assertEqual(DOM.get_tag_name(self.style_state.create_nodes('Test text')[0]), 'textnode')
self.assertEqual(DOM.get_text_content(self.style_state.create_nodes('Test text')[0]), 'Test text')
def test_create_node_unicode(self):
self.assertEqual(DOM.get_text_content(self.style_state.create_nodes('🍺')[0]), '🍺')
def test_create_node_styled(self):
self.style_state.apply(Command('start_inline_style', 0, 'ITALIC'))
self.assertEqual(DOM.get_tag_name(self.style_state.create_nodes('Test text')[0]), 'em')
self.assertEqual(self.style_state.create_nodes('Test text')[0].get('style'), None)
self.assertEqual(DOM.get_text_content(self.style_state.create_nodes('Test text')[0]), 'Test text')
self.style_state.apply(Command('stop_inline_style', 9, 'ITALIC'))
def test_create_node_styled_multiple(self):
self.style_state.apply(Command('start_inline_style', 0, 'BOLD'))
self.style_state.apply(Command('start_inline_style', 0, 'ITALIC'))
self.assertEqual(self.style_state.get_style_tags(), ['em', 'strong'])
self.assertEqual(DOM.get_tag_name(self.style_state.create_nodes('wow')[0]), 'em')
self.assertEqual(DOM.get_tag_name(DOM.get_children(self.style_state.create_nodes('wow')[0])[0]), 'strong')
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)
self.composite_decorators = config.get('composite_decorators', [])
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,
self.composite_decorators)
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_nodes = self.style_state.create_nodes(
text, block=block, entity_stack=entity_state.entity_stack)
entity_state.render_entitities(element, style_nodes)
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')
