Move render table to cli helper

This commit is contained in:
henryruhs
2025-04-28 10:03:25 +02:00
parent 964fab8724
commit b11cb07aea
3 changed files with 38 additions and 35 deletions
+1 -33
View File
@@ -1,9 +1,8 @@
from logging import Logger, basicConfig, getLogger
from typing import Tuple
import facefusion.choices
from facefusion.common_helper import get_first, get_last
from facefusion.typing import LogLevel, TableContents, TableHeaders
from facefusion.typing import LogLevel
def init(log_level : LogLevel) -> None:
@@ -41,37 +40,6 @@ def create_message(message : str, module_name : str) -> str:
return message
def table(headers : TableHeaders, contents : TableContents) -> None:
package_logger = get_package_logger()
table_column, table_separator = create_table_parts(headers, contents)
package_logger.info(table_separator)
package_logger.info(table_column.format(*headers))
package_logger.info(table_separator)
for content in contents:
content = [ value if value else '' for value in content ]
package_logger.info(table_column.format(*content))
package_logger.info(table_separator)
def create_table_parts(headers : TableHeaders, contents : TableContents) -> Tuple[str, str]:
column_parts = []
separator_parts = []
widths = [ len(header) for header in headers ]
for content in contents:
for index, value in enumerate(content):
widths[index] = max(widths[index], len(str(value)))
for width in widths:
column_parts.append('{:<' + str(width) + '}')
separator_parts.append('-' * width)
return '| ' + ' | '.join(column_parts) + ' |', '+-' + '-+-'.join(separator_parts) + '-+'
def enable() -> None:
get_package_logger().disabled = False