| import os | |
| from ._base import DirectivePlugin | |
| class Include(DirectivePlugin): | |
| def parse(self, block, m, state): | |
| source_file = state.env.get('__file__') | |
| if not source_file: | |
| return {'type': 'block_error', 'raw': 'Missing source file'} | |
| encoding = 'utf-8' | |
| options = self.parse_options(m) | |
| if options: | |
| attrs = dict(options) | |
| if 'encoding' in attrs: | |
| encoding = attrs['encoding'] | |
| else: | |
| attrs = {} | |
| relpath = self.parse_title(m) | |
| dest = os.path.join(os.path.dirname(source_file), relpath) | |
| dest = os.path.normpath(dest) | |
| if dest == source_file: | |
| return { | |
| 'type': 'block_error', | |
| 'raw': 'Could not include self: ' + relpath, | |
| } | |
| if not os.path.isfile(dest): | |
| return { | |
| 'type': 'block_error', | |
| 'raw': 'Could not find file: ' + relpath, | |
| } | |
| with open(dest, 'rb') as f: | |
| content = f.read() | |
| content = content.decode(encoding) | |
| ext = os.path.splitext(relpath)[1] | |
| if ext in {'.md', '.markdown', '.mkd'}: | |
| new_state = block.state_cls() | |
| new_state.env['__file__'] = dest | |
| new_state.process(content) | |
| block.parse(new_state) | |
| return new_state.tokens | |
| elif ext in {'.html', '.xhtml', '.htm'}: | |
| return {'type': 'block_html', 'raw': content} | |
| attrs['filepath'] = dest | |
| return { | |
| 'type': 'include', | |
| 'raw': content, | |
| 'attrs': attrs, | |
| } | |
| def __call__(self, directive, md): | |
| directive.register('include', self.parse) | |
| if md.renderer and md.renderer.NAME == 'html': | |
| md.renderer.register('include', render_html_include) | |
| def render_html_include(renderer, text, **attrs): | |
| return '<pre class="directive-include">\n' + text + '</pre>\n' | |