Lib.robot.htmldata.jartemplate.py Maven / Gradle / Ivy

Go to download
#  Copyright 2008-2015 Nokia Solutions and Networks
#
#  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
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.

import os
from posixpath import normpath, join
from contextlib import contextmanager
from java.io import BufferedReader, InputStreamReader

# Works only when running from jar
from org.robotframework.RobotRunner import getResourceAsStream


class HtmlTemplate(object):
    _base_dir = '/Lib/robot/htmldata/'

    def __init__(self, filename):
        self._path = normpath(join(self._base_dir, filename.replace(os.sep, '/')))

    def __iter__(self):
        with self._reader as reader:
            line = reader.readLine()
            while line is not None:
                yield line.rstrip()
                line = reader.readLine()

    @property
    @contextmanager
    def _reader(self):
        stream = getResourceAsStream(self._path)
        reader = BufferedReader(InputStreamReader(stream, 'UTF-8'))
        try:
            yield reader
        finally:
            reader.close()