Source code for l2tscaffolder.lib.code_formatter

# -*- coding: utf-8 -*-
"""Formatter for generated code."""

from yapf.yapflib import yapf_api


[docs]class CodeFormatter: """Formats code in files.""" def __init__(self, yapf_path: str): """Initializes the code formatter. Args: yapf_path (str): path to the yapf style file. """ super().__init__() self.yapf_path = yapf_path
[docs] def Format(self, code: str) -> str: """Formats the code. Args: code (str): code to format Returns: str: the formatted code """ return yapf_api.FormatCode(code, style_config=self.yapf_path)