Source code for robot.parsing.parser.fileparser

#  Copyright 2008-2015 Nokia Networks
#  Copyright 2016-     Robot Framework Foundation
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  See the License for the specific language governing permissions and
#  limitations under the License.

import os.path

from robot.utils import is_pathlike, is_string

from ..lexer import Token
from ..model import (File, CommentSection, SettingSection, VariableSection,
                     TestCaseSection, KeywordSection)

from .blockparsers import Parser, TestCaseParser, KeywordParser

[docs]class FileParser(Parser): def __init__(self, source=None): Parser.__init__(self, File(source=self._get_path(source))) def _get_path(self, source): if not source: return None if is_string(source) and '\n' not in source and os.path.isfile(source): return source if is_pathlike(source) and source.is_file(): return str(source) return None
[docs] def handles(self, statement): return True
[docs] def parse(self, statement): parser_class = { Token.SETTING_HEADER: SettingSectionParser, Token.VARIABLE_HEADER: VariableSectionParser, Token.TESTCASE_HEADER: TestCaseSectionParser, Token.KEYWORD_HEADER: KeywordSectionParser, Token.COMMENT_HEADER: CommentSectionParser, Token.COMMENT: ImplicitCommentSectionParser, Token.ERROR: ImplicitCommentSectionParser, Token.EOL: ImplicitCommentSectionParser }[statement.type] parser = parser_class(statement) self.model.sections.append(parser.model) return parser
[docs]class SectionParser(Parser): model_class = None def __init__(self, header): Parser.__init__(self, self.model_class(header))
[docs] def handles(self, statement): return statement.type not in Token.HEADER_TOKENS
[docs] def parse(self, statement): self.model.body.append(statement) return None
[docs]class SettingSectionParser(SectionParser): model_class = SettingSection
[docs]class VariableSectionParser(SectionParser): model_class = VariableSection
[docs]class CommentSectionParser(SectionParser): model_class = CommentSection
[docs]class ImplicitCommentSectionParser(SectionParser):
[docs] def model_class(self, statement): return CommentSection(body=[statement])
[docs]class TestCaseSectionParser(SectionParser): model_class = TestCaseSection
[docs] def parse(self, statement): if statement.type == Token.TESTCASE_NAME: parser = TestCaseParser(statement) self.model.body.append(parser.model) return parser return SectionParser.parse(self, statement)
[docs]class KeywordSectionParser(SectionParser): model_class = KeywordSection
[docs] def parse(self, statement): if statement.type == Token.KEYWORD_NAME: parser = KeywordParser(statement) self.model.body.append(parser.model) return parser return SectionParser.parse(self, statement)