jartemplate.py

Go to the documentation of this file.

#  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
#
#      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():
    
    _base_dir = '/Lib/robot/htmldata/'
 
    def __init__(self, filename):
        self._path_path = normpath(join(self._base_dir_base_dir, filename.replace(os.sep, '/')))
 
    def __iter__(self):
        with self._reader_reader_reader as reader:
            line = reader.readLine()
            while line is not None:
                yield line.rstrip()
                line = reader.readLine()
 
    @property
    @contextmanager
    _reader = property
    
    def _reader(self):
        stream = getResourceAsStream(self._path_path)
        reader = BufferedReader(InputStreamReader(stream, 'UTF-8'))
        try:
            yield reader
        finally:
            reader.close()