def convert_robot_data_file(original_filepath): from robot.tidy import Tidy from base64 import b64encode tidy = Tidy(format='robot') if get_robot_version_num() < (3, 2) else Tidy() converted_content = tidy.file(original_filepath) if sys.version_info < (3, 0, 0): return b64encode(converted_content.encode('utf-8')) else: return str(b64encode(bytes(converted_content, 'utf-8')), 'utf-8')
def robot_source_format(source, space_count=4): if not source.strip(): return None from robot.tidy import Tidy t = Tidy(space_count=space_count) from io import StringIO s = StringIO() if isinstance(source, bytes): source = source.decode("utf-8", "replace") s.write(source) s.seek(0) formatted = t.file(s) if not formatted: return None return formatted