def run(self): if not self.state.document.settings.file_insertion_enabled: raise self.warning('"%s" directive disabled.' % self.name) source = self.state_machine.input_lines.source( self.lineno - self.state_machine.input_offset - 1) source_dir = os.path.dirname(os.path.abspath(source)) file_path = rst.directives.path(self.arguments[0]) absolute_path = os.path.normpath(os.path.join(source_dir, file_path)) path = relative_path(None, absolute_path) settings = self.state.document.settings try: settings.record_dependencies.add(path) include_file = FileInput( source_path=path, encoding=self.options.get("encoding", settings.input_encoding), error_handler=settings.input_encoding_error_handler, ) except UnicodeEncodeError as error: raise self.severe(f'Problems with "{self.name}" directive path:\n' f'Cannot encode input file path "{path}" ' "(wrong locale?).") except IOError as error: raise self.severe( f'Problems with "{self.name}" directive path:\n{ErrorString(error)}.' ) startline = self.options.get("start-line", None) endline = self.options.get("end-line", None) try: if startline or endline is not None: lines = include_file.readlines() rawtext = "".join(lines[startline:endline]) else: rawtext = include_file.read() except UnicodeError as error: raise self.severe( f'Problem with "{self.name}" directive:\n{ErrorString(error)}') document = new_document(absolute_path, settings) MystParser().parse(rawtext, document) return document.children
def run(self): if not self.state.document.settings.file_insertion_enabled: raise self.warning('"%s" directive disabled.' % self.name) path = directives.path(self.arguments[0]) # ALL the included files are relative to the repository root. # we need to remove absolute paths if path.startswith('/'): raise self.severe('Problem with "%s" directive path:\npath ' \ 'should be relative' % self.name) encoding = self.options.get( 'encoding', self.state.document.settings.input_encoding) tab_width = self.options.get('tab-width', self.state.document.settings.tab_width) try: self.state.document.settings.record_dependencies.add(path) include_file = FileInput( source=BlohgFile(path), encoding=encoding, error_handler=(self.state.document.settings.\ input_encoding_error_handler)) except IOError as error: raise self.severe('Problems with "%s" directive path:\n%s.' % (self.name, ErrorString(error))) startline = self.options.get('start-line', None) endline = self.options.get('end-line', None) try: if startline or (endline is not None): lines = include_file.readlines() rawtext = ''.join(lines[startline:endline]) else: rawtext = include_file.read() except UnicodeError as error: raise self.severe('Problem with "%s" directive:\n%s' % (self.name, ErrorString(error))) # start-after/end-before: no restrictions on newlines in match-text, # and no restrictions on matching inside lines vs. line boundaries after_text = self.options.get('start-after', None) if after_text: # skip content in rawtext before *and incl.* a matching text after_index = rawtext.find(after_text) if after_index < 0: raise self.severe('Problem with "start-after" option of "%s" ' 'directive:\nText not found.' % self.name) rawtext = rawtext[after_index + len(after_text):] before_text = self.options.get('end-before', None) if before_text: # skip content in rawtext after *and incl.* a matching text before_index = rawtext.find(before_text) if before_index < 0: raise self.severe('Problem with "end-before" option of "%s" ' 'directive:\nText not found.' % self.name) rawtext = rawtext[:before_index] if 'literal' in self.options: # Convert tabs to spaces, if `tab_width` is positive. if tab_width >= 0: text = rawtext.expandtabs(tab_width) else: text = rawtext literal_block = nodes.literal_block(rawtext, text, source=path) literal_block.line = 1 return [literal_block] else: include_lines = statemachine.string2lines(rawtext, tab_width, convert_whitespace=1) self.state_machine.insert_input(include_lines, path) return []