| Server IP : 153.92.12.186 / Your IP : 216.73.217.129 Web Server : LiteSpeed System : Linux id-dci-web1986.main-hosting.eu 5.14.0-611.26.1.el9_7.x86_64 #1 SMP PREEMPT_DYNAMIC Thu Jan 29 05:24:47 EST 2026 x86_64 User : u686484674 ( 686484674) PHP Version : 8.0.30 Disable Function : system, exec, shell_exec, passthru, mysql_list_dbs, ini_alter, dl, symlink, link, chgrp, leak, popen, apache_child_terminate, virtual, mb_send_mail MySQL : OFF | cURL : ON | WGET : ON | Perl : OFF | Python : OFF | Sudo : OFF | Pkexec : OFF Directory : /proc/self/root/opt/cloudlinux/venv/lib/python3.11/site-packages/prettytable/ |
Upload File : |
from __future__ import annotations
from .prettytable import PrettyTable
try:
from colorama import init
init()
except ImportError:
pass
RESET_CODE = "\x1b[0m"
class Theme:
def __init__(
self,
default_color: str = "",
vertical_char: str = "|",
vertical_color: str = "",
horizontal_char: str = "-",
horizontal_color: str = "",
junction_char: str = "+",
junction_color: str = "",
) -> None:
self.default_color = Theme.format_code(default_color)
self.vertical_char = vertical_char
self.vertical_color = Theme.format_code(vertical_color)
self.horizontal_char = horizontal_char
self.horizontal_color = Theme.format_code(horizontal_color)
self.junction_char = junction_char
self.junction_color = Theme.format_code(junction_color)
@staticmethod
def format_code(s: str) -> str:
"""Takes string and intelligently puts it into an ANSI escape sequence"""
if s.strip() == "":
return ""
elif s.startswith("\x1b["):
return s
else:
return f"\x1b[{s}m"
class Themes:
DEFAULT = Theme()
OCEAN = Theme(
default_color="96",
vertical_color="34",
horizontal_color="34",
junction_color="36",
)
class ColorTable(PrettyTable):
def __init__(self, field_names=None, **kwargs) -> None:
super().__init__(field_names=field_names, **kwargs)
# TODO: Validate option
self.theme = kwargs.get("theme") or Themes.DEFAULT
@property
def theme(self) -> Theme:
return self._theme
@theme.setter
def theme(self, value: Theme):
self._theme = value
self.update_theme()
def update_theme(self) -> None:
theme = self._theme
self._vertical_char = (
theme.vertical_color
+ theme.vertical_char
+ RESET_CODE
+ theme.default_color
)
self._horizontal_char = (
theme.horizontal_color
+ theme.horizontal_char
+ RESET_CODE
+ theme.default_color
)
self._junction_char = (
theme.junction_color
+ theme.junction_char
+ RESET_CODE
+ theme.default_color
)
def get_string(self, **kwargs) -> str:
return super().get_string(**kwargs) + RESET_CODE