slither: utils: respect colorization state when printing tables

Closes #2229
pull/2310/head
Emilio López 9 months ago
parent e876d61fd5
commit b66b3e091f
  1. 9
      slither/utils/myprettytable.py

@ -1,7 +1,10 @@
from typing import List, Dict, Union
from prettytable import PrettyTable
from prettytable.colortable import ColorTable, Themes
from slither.utils.colors import Colors
class MyPrettyTable:
def __init__(self, field_names: List[str], pretty_align: bool = True): # TODO: True by default?
@ -19,8 +22,12 @@ class MyPrettyTable:
def add_row(self, row: List[Union[str, List[str]]]) -> None:
self._rows.append(row)
def to_pretty_table(self) -> ColorTable:
def to_pretty_table(self) -> PrettyTable:
if Colors.COLORIZATION_ENABLED:
table = ColorTable(self._field_names, theme=Themes.OCEAN)
else:
table = PrettyTable(self._field_names)
for row in self._rows:
table.add_row(row)
if len(self._options["set_alignment"]):

Loading…
Cancel
Save