예제 #1
0
    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
예제 #2
0
 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 []