Merge pull request #2310 from elopez/dev-colorless

slither: utils: respect colorization state when printing tables
pull/2311/head
alpharush 9 months ago committed by GitHub
commit 029d5db8b8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  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