Spaces:
Running
Running
Upload 8 files
Browse files- mew_log/__init__.py +22 -0
- mew_log/ansi_utils.py +148 -0
- mew_log/attr_utils.py +260 -0
- mew_log/extended_symbols.py +127 -0
- mew_log/log_utils.py +502 -0
- mew_log/mem_utils.py +144 -0
- mew_log/profile_utils.py +194 -0
- mew_log/table_utils.py +174 -0
mew_log/__init__.py
ADDED
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from .ansi_utils import AnsiCodeHelper, ansi_color_str, ansi_format_str, ansi_link_str, ansi_rainbow_str, print_to_ansi_alt_screen_buffer
|
2 |
+
from .attr_utils import get_caller_info, get_class_attributes, get_class_name, get_method_args, get_method_info, get_partial_argspec, get_prev_caller_info, get_prev_frame, get_prev_frame_from_frame, get_signature, get_source_info, get_thread_info, get_type_name, get_var_value, get_variable_info, is_bytes_like_type, is_complex_type, is_of_type, list_class_attributes
|
3 |
+
from .extended_symbols import ExtendedSymbols, draw_arrow, draw_box, format_arrow
|
4 |
+
from .log_utils import LogSpinner, allow_curly_braces, ensure_directory_exists, ensure_file_exists, format_arg, format_args, format_duration, format_kwarg, format_path, format_traceback, get_timestamp, init_log, log_error, log_file, log_function, log_info, log_text_input, log_warning, make_formatted_msg, parse_duration, run_unit_test, stringf, trace_error
|
5 |
+
from .mem_utils import bytes_to_kb, bytes_to_mb, format_mem_size_value, format_size_bytes, get_mem_size, get_mem_size_breakdown, save_obj_graph_img, take_mem_growth_snapshot
|
6 |
+
from .profile_utils import ProfileSection, allow_curly_braces, create_or_get_profile, format_profile_registry, get_profile_registry, get_profile_reports, log_profile_registry, make_time_str, profile_function, profile_start, profile_stop
|
7 |
+
from .table_utils import format_data_as_table, format_dict_as_table, format_list_as_table, format_obj_as_table, format_property, format_square_layout, format_table, print_as_square, print_square_layout, print_table, transpose_dict
|
8 |
+
|
9 |
+
__all__ = [
|
10 |
+
'AnsiCodeHelper', 'ansi_color_str', 'ansi_format_str', 'ansi_link_str', 'ansi_rainbow_str', 'print_to_ansi_alt_screen_buffer',
|
11 |
+
'get_caller_info', 'get_class_attributes', 'get_class_name', 'get_method_args', 'get_method_info', 'get_partial_argspec', 'get_prev_caller_info', 'get_prev_frame', 'get_prev_frame_from_frame', 'get_signature',
|
12 |
+
'get_source_info', 'get_thread_info', 'get_type_name', 'get_var_value', 'get_variable_info', 'is_bytes_like_type', 'is_complex_type', 'is_of_type', 'list_class_attributes',
|
13 |
+
'ExtendedSymbols', 'draw_arrow', 'draw_box', 'format_arrow',
|
14 |
+
'LogSpinner', 'allow_curly_braces', 'ensure_directory_exists', 'ensure_file_exists', 'format_arg', 'format_args', 'format_duration', 'format_kwarg', 'format_path', 'format_traceback',
|
15 |
+
'get_timestamp', 'init_log', 'log_error', 'log_file', 'log_function', 'log_info', 'log_text_input', 'log_warning', 'make_formatted_msg', 'parse_duration',
|
16 |
+
'run_unit_test', 'stringf', 'trace_error',
|
17 |
+
'bytes_to_kb', 'bytes_to_mb', 'format_mem_size_value', 'format_size_bytes', 'get_mem_size', 'get_mem_size_breakdown', 'save_obj_graph_img', 'take_mem_growth_snapshot',
|
18 |
+
'ProfileSection', 'allow_curly_braces', 'create_or_get_profile', 'format_profile_registry', 'get_profile_registry', 'get_profile_reports', 'log_profile_registry', 'make_time_str', 'profile_function', 'profile_start',
|
19 |
+
'profile_stop',
|
20 |
+
'format_data_as_table', 'format_dict_as_table', 'format_list_as_table', 'format_obj_as_table', 'format_property', 'format_square_layout', 'format_table', 'print_as_square', 'print_square_layout', 'print_table',
|
21 |
+
'transpose_dict',
|
22 |
+
]
|
mew_log/ansi_utils.py
ADDED
@@ -0,0 +1,148 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import sys
|
2 |
+
import time
|
3 |
+
|
4 |
+
# from colorama import Fore, Back, Style
|
5 |
+
# from colorama import just_fix_windows_console, init
|
6 |
+
|
7 |
+
# init(autoreset=True)
|
8 |
+
# # use Colorama to make Termcolor work on Windows too
|
9 |
+
# just_fix_windows_console()
|
10 |
+
|
11 |
+
|
12 |
+
class AnsiCodeHelper:
|
13 |
+
ansi_codes = {
|
14 |
+
'fg_colors': {
|
15 |
+
'black': '\033[30m',
|
16 |
+
'red': '\033[31m',
|
17 |
+
'green': '\033[32m',
|
18 |
+
'yellow': '\033[33m',
|
19 |
+
'blue': '\033[34m',
|
20 |
+
'magenta': '\033[35m',
|
21 |
+
'cyan': '\033[36m',
|
22 |
+
'white': '\033[37m',
|
23 |
+
'bright_black': '\033[1;30m',
|
24 |
+
'bright_red': '\033[1;31m',
|
25 |
+
'bright_green': '\033[1;32m',
|
26 |
+
'bright_yellow': '\033[1;33m',
|
27 |
+
'bright_blue': '\033[1;34m',
|
28 |
+
'bright_magenta': '\033[1;35m',
|
29 |
+
'bright_cyan': '\033[1;36m',
|
30 |
+
'bright_white': '\033[1;37m',
|
31 |
+
'reset_color': '\033[39m',
|
32 |
+
},
|
33 |
+
'bg_colors': {
|
34 |
+
'black': '\033[40m',
|
35 |
+
'red': '\033[41m',
|
36 |
+
'green': '\033[42m',
|
37 |
+
'yellow': '\033[43m',
|
38 |
+
'blue': '\033[44m',
|
39 |
+
'magenta': '\033[45m',
|
40 |
+
'cyan': '\033[46m',
|
41 |
+
'white': '\033[47m',
|
42 |
+
'reset_background': '\033[49m',
|
43 |
+
},
|
44 |
+
'text_format': {
|
45 |
+
'bold': '\033[1m',
|
46 |
+
'underline': '\033[4m',
|
47 |
+
'italic': '\033[3m',
|
48 |
+
'inverse': '\033[7m',
|
49 |
+
'blink': '\033[5m',
|
50 |
+
'hidden': '\033[8m',
|
51 |
+
'strike_through': '\033[9m',
|
52 |
+
'frame': '\033[51m',
|
53 |
+
'encircled': '\033[52m',
|
54 |
+
'overlined': '\033[53m',
|
55 |
+
'reset_format': '\033[0m',
|
56 |
+
},
|
57 |
+
'link': {
|
58 |
+
'url_start': '\033]8;;',
|
59 |
+
'url_end': '\033]8;;\033\\',
|
60 |
+
},
|
61 |
+
'utility': {
|
62 |
+
'escape_sequence_start': '\033]',
|
63 |
+
'escape_sequence_end': '\033\\',
|
64 |
+
'enable_alternate_screen_buffer': '\033[?1049h',
|
65 |
+
'disable_alternate_screen_buffer': '\033[?1049l',
|
66 |
+
}
|
67 |
+
}
|
68 |
+
|
69 |
+
@staticmethod
|
70 |
+
def get_ansi_code(category, name):
|
71 |
+
return AnsiCodeHelper.ansi_codes.get(category, {}).get(name, '')
|
72 |
+
|
73 |
+
|
74 |
+
def ansi_link_str(url, link_text="", link_color='bright_green'):
|
75 |
+
# Get the ANSI escape sequence for the default color
|
76 |
+
# Check if link text is empty
|
77 |
+
if link_text == '':
|
78 |
+
link_text = url
|
79 |
+
|
80 |
+
# Check if stdout is a terminal
|
81 |
+
if sys.stdout.isatty():
|
82 |
+
# get ansi codes
|
83 |
+
# default_color_code = AnsiCodeHelper.get_ansi_code('fg_colors', link_color)
|
84 |
+
# format_reset_code = AnsiCodeHelper.get_ansi_code('text_format', 'reset_format')
|
85 |
+
start_sequence = AnsiCodeHelper.get_ansi_code('utility', 'escape_sequence_start')
|
86 |
+
end_sequence = AnsiCodeHelper.get_ansi_code('utility', 'escape_sequence_end')
|
87 |
+
# create url link sequence
|
88 |
+
url_start = AnsiCodeHelper.get_ansi_code('link', 'url_start')
|
89 |
+
url_end = AnsiCodeHelper.get_ansi_code('link', 'url_end')
|
90 |
+
link_color_code = AnsiCodeHelper.get_ansi_code('fg_colors', link_color)
|
91 |
+
reset_format_code = AnsiCodeHelper.get_ansi_code('text_format', 'reset_format')
|
92 |
+
|
93 |
+
formatted_link_text = f"{url_start}{url}{end_sequence}{link_color_code}{link_text}{reset_format_code}{url_end}"
|
94 |
+
|
95 |
+
# log_info(f'formatted_link_text={formatted_link_text}\n\n')
|
96 |
+
|
97 |
+
return formatted_link_text
|
98 |
+
else:
|
99 |
+
# If stdout is not a terminal, return the default formatted link text
|
100 |
+
return link_text
|
101 |
+
|
102 |
+
|
103 |
+
def ansi_color_str(s, fg='white', bg=None):
|
104 |
+
if bg is None:
|
105 |
+
color_type = 'fg_colors'
|
106 |
+
color_code = AnsiCodeHelper.get_ansi_code(color_type, fg)
|
107 |
+
reset_color_code = AnsiCodeHelper.get_ansi_code(color_type, 'reset_color')
|
108 |
+
return f"{color_code}{s}{reset_color_code}"
|
109 |
+
else:
|
110 |
+
fg_color_code = AnsiCodeHelper.get_ansi_code('fg_colors', fg)
|
111 |
+
bg_color_code = AnsiCodeHelper.get_ansi_code('bg_colors', bg)
|
112 |
+
reset_bg_color_code = AnsiCodeHelper.get_ansi_code('bg_colors', 'reset_background')
|
113 |
+
reset_fg_color_code = AnsiCodeHelper.get_ansi_code('fg_colors', 'reset_color')
|
114 |
+
return f"{bg_color_code}{fg_color_code}{s}{reset_fg_color_code}{reset_bg_color_code}"
|
115 |
+
|
116 |
+
|
117 |
+
def ansi_format_str(s, format_name):
|
118 |
+
format_code = AnsiCodeHelper.get_ansi_code('text_format', format_name)
|
119 |
+
reset_format_code = AnsiCodeHelper.get_ansi_code('text_format', 'reset_format')
|
120 |
+
return f"{format_code}{s}{reset_format_code}"
|
121 |
+
|
122 |
+
|
123 |
+
def ansi_rainbow_str(message, delay=0.1, iterations=10):
|
124 |
+
# init_thread(ansi_rainbow_anim, message, delay)
|
125 |
+
rainbow_colors = ['red', 'yellow', 'green', 'blue', 'magenta', 'cyan']
|
126 |
+
for _ in range(iterations):
|
127 |
+
for color in rainbow_colors:
|
128 |
+
color_code = AnsiCodeHelper.get_ansi_code('fg_colors', color)
|
129 |
+
reset_color_code = AnsiCodeHelper.get_ansi_code('fg_colors', 'reset_color')
|
130 |
+
rainbow_str = ansi_color_str(message, color)
|
131 |
+
log_info(rainbow_str, end='\r', flush=True)
|
132 |
+
time.sleep(delay)
|
133 |
+
log_info('\n')
|
134 |
+
|
135 |
+
|
136 |
+
def print_to_ansi_alt_screen_buffer(info):
|
137 |
+
# Enable alternate screen buffer
|
138 |
+
sys.stdout.write(
|
139 |
+
f"{AnsiCodeHelper.get_ansi_code('utility', 'escape_sequence_start')}{AnsiCodeHelper.get_ansi_code('utility', 'enable_alternate_screen_buffer')}")
|
140 |
+
|
141 |
+
# Display additional information on the alternate screen buffer
|
142 |
+
log_info(info)
|
143 |
+
|
144 |
+
# Disable alternate screen buffer
|
145 |
+
sys.stdout.write(
|
146 |
+
f"{AnsiCodeHelper.get_ansi_code('utility', 'escape_sequence_start')}{AnsiCodeHelper.get_ansi_code('utility', 'disable_alternate_screen_buffer')}")
|
147 |
+
|
148 |
+
|
mew_log/attr_utils.py
ADDED
@@ -0,0 +1,260 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from ..mew_log.ansi_utils import ansi_link_str, ansi_color_str
|
2 |
+
|
3 |
+
|
4 |
+
def get_type_name(attr):
|
5 |
+
return type(attr).__name__
|
6 |
+
|
7 |
+
def is_complex_type(obj):
|
8 |
+
return is_of_type(variable, (list, set, dict)) or hasattr(variable, '__dict__')
|
9 |
+
|
10 |
+
|
11 |
+
def is_bytes_like_type(obj, bytes_like_types=(memoryview, bytes, bytearray), raise_err=False):
|
12 |
+
try:
|
13 |
+
return is_of_type(obj, bytes_like_types)
|
14 |
+
except TypeError as e:
|
15 |
+
if raise_err:
|
16 |
+
raise TypeError(f"Provided object does not match the provided types: {bytes_like_types}")
|
17 |
+
return False
|
18 |
+
|
19 |
+
def is_of_type(obj, types, raise_err=False):
|
20 |
+
if isinstance(obj, (types)):
|
21 |
+
return True
|
22 |
+
elif raise_err:
|
23 |
+
raise TypeError(f"Provided object does not match the provided types: {tuple(types)}")
|
24 |
+
return False
|
25 |
+
|
26 |
+
|
27 |
+
def get_partial_argspec(method):
|
28 |
+
if not callable(method):
|
29 |
+
return None # Not a callable object
|
30 |
+
try:
|
31 |
+
full_argspec = inspect.getfullargspec(method)
|
32 |
+
return full_argspec
|
33 |
+
except TypeError:
|
34 |
+
# Fallback to using inspect.signature
|
35 |
+
signature = get_signature(method)
|
36 |
+
if signature:
|
37 |
+
parameters = signature.parameters
|
38 |
+
args = [param for param in parameters if parameters[param].default == parameters[param].empty]
|
39 |
+
varargs = signature.varargs
|
40 |
+
varkw = signature.varkw
|
41 |
+
defaults = [parameters[param].default for param in args]
|
42 |
+
return inspect.FullArgSpec(args, varargs, varkw, defaults)
|
43 |
+
|
44 |
+
def get_signature(method):
|
45 |
+
try:
|
46 |
+
signature = inspect.signature(method)
|
47 |
+
return signature
|
48 |
+
except (TypeError, ValueError):
|
49 |
+
return None
|
50 |
+
|
51 |
+
def get_method_args(method):
|
52 |
+
|
53 |
+
full_arg_spec = get_partial_argspec(method)
|
54 |
+
if full_arg_spec:
|
55 |
+
args = [arg for arg in full_arg_spec.args if
|
56 |
+
getattr(method, arg, None) is not None and getattr(method, arg, "") != ""]
|
57 |
+
kwargs = {key: getattr(method, key, None) for key in full_arg_spec.kwonlyargs}
|
58 |
+
kwargs_defaults = {key: value for key, value in
|
59 |
+
zip(full_arg_spec.kwonlyargs, full_arg_spec.kwonlydefaults or ())}
|
60 |
+
args.extend(f"{key}={value}" for key, value in kwargs.items() if value is not None and value != "")
|
61 |
+
return args
|
62 |
+
return None
|
63 |
+
|
64 |
+
|
65 |
+
def get_source_info(attr):
|
66 |
+
try:
|
67 |
+
source_lines, line_number = inspect.getsourcelines(attr)
|
68 |
+
source_file_path = inspect.getsourcefile(attr)
|
69 |
+
source_file = os.path.relpath(source_file_path)
|
70 |
+
return f"/{source_file}::{line_number}"
|
71 |
+
except Exception as e:
|
72 |
+
return "Source info not available!"
|
73 |
+
|
74 |
+
|
75 |
+
def get_method_info(method):
|
76 |
+
args = get_method_args(method)
|
77 |
+
args_str = ", ".join(args) if args else ''
|
78 |
+
signature = get_signature(method)
|
79 |
+
return_str = f' -> {signature.return_annotation}' if signature and signature.return_annotation is not inspect.Signature.empty else ''
|
80 |
+
|
81 |
+
try:
|
82 |
+
source_info = get_source_info(method)
|
83 |
+
except Exception as e:
|
84 |
+
raise Exception("Source info not available!", e)
|
85 |
+
|
86 |
+
# Construct the file:// URL with line number for the method if available
|
87 |
+
method_file_url = f"file://{inspect.getsourcefile(method)}#L{inspect.getsourcelines(method)[1]}"
|
88 |
+
method_link = ansi_link_str(method_file_url, "Source")
|
89 |
+
|
90 |
+
# Include the link in the method signature string
|
91 |
+
method_signature = f"{signature}{return_str}: {method_link}\n-->{source_info}"
|
92 |
+
|
93 |
+
return method_signature
|
94 |
+
def get_var_value(variable):
|
95 |
+
return f'{str(variable)}' if not is_of_type(variable, (list, set, dict)) or hasattr(variable, '__dict__') else '...'
|
96 |
+
|
97 |
+
def get_variable_info(variable):
|
98 |
+
return f"<{ get_type_name(variable) }>: { get_var_value(variable) }"
|
99 |
+
|
100 |
+
def list_class_attributes(cls, verbose=True, use_color=False):
|
101 |
+
def format_str(s, fg=None):
|
102 |
+
return ansi_color_str(s, fg=fg) if use_color else s
|
103 |
+
|
104 |
+
# Determine whether cls is a class or an instance of a class
|
105 |
+
if inspect.isclass(cls):
|
106 |
+
class_name = cls.__name__
|
107 |
+
else:
|
108 |
+
class_name = cls.__class__.__name__
|
109 |
+
|
110 |
+
variables = [
|
111 |
+
f'{attribute}{get_variable_info(getattr(cls, attribute))}' if verbose else f'{attribute}<{get_type_name(getattr(cls, attribute))}>'
|
112 |
+
for attribute in dir(cls) if not attribute.startswith('__') and not callable(getattr(cls, attribute))]
|
113 |
+
methods = [f'{attribute}{get_method_info(getattr(cls, attribute))}' if verbose else f'{attribute}<method>' for
|
114 |
+
attribute in dir(cls) if not attribute.startswith('__') and callable(getattr(cls, attribute))]
|
115 |
+
|
116 |
+
variables_str = '\n'.join([f' - {format_str(var, fg="green")}' for var in variables])
|
117 |
+
methods_str = '\n'.join([f' - {format_str(method, fg="blue")}' for method in methods])
|
118 |
+
|
119 |
+
cls_name = format_str(class_name, fg="cyan") if use_color else class_name
|
120 |
+
return f'===list_class_attributes of: {cls_name}:\n===<variables>===\n{variables_str}\n===<methods>===\n{methods_str}'
|
121 |
+
|
122 |
+
def get_class_attributes(cls, verbose=True, use_color=True):
|
123 |
+
def format_str(s, fg=None):
|
124 |
+
return ansi_color_str(s, fg=fg) if use_color else s
|
125 |
+
|
126 |
+
attributes_dict = {'variables': {}, 'methods': {}}
|
127 |
+
|
128 |
+
for attribute_v in vars(cls):
|
129 |
+
if not attribute_v.startswith('__') and 'stat' not in attribute_v:
|
130 |
+
attr = getattr(cls, attribute_v)
|
131 |
+
if not callable(attr):
|
132 |
+
attr_info = get_variable_info(attr) if verbose else ''
|
133 |
+
formatted_key = format_str(attribute_v, fg="green")
|
134 |
+
formatted_value = format_str(attr_info, fg="cyan") if verbose else ''
|
135 |
+
attributes_dict['variables'][attribute_v] = f'\n ~ {formatted_key}{formatted_value}'
|
136 |
+
|
137 |
+
for attribute in dir(cls):
|
138 |
+
if not attribute.startswith('__'):
|
139 |
+
attr = getattr(cls, attribute)
|
140 |
+
if callable(attr):
|
141 |
+
method_info = get_method_info(attr) if verbose else ''
|
142 |
+
formatted_key = format_str(attribute, fg="blue")
|
143 |
+
formatted_value = format_str(method_info, fg="cyan") if verbose else ''
|
144 |
+
attributes_dict['methods'][attribute] = f'\n ~ {formatted_key}{formatted_value}'
|
145 |
+
|
146 |
+
return attributes_dict
|
147 |
+
|
148 |
+
|
149 |
+
import threading
|
150 |
+
|
151 |
+
|
152 |
+
def get_thread_info(message='', use_color=True):
|
153 |
+
current_thread = threading.current_thread()
|
154 |
+
current_thread_name = current_thread.name
|
155 |
+
current_thread_id = current_thread.ident
|
156 |
+
current_thread_alive = current_thread.is_alive()
|
157 |
+
|
158 |
+
# Construct the colored thread info
|
159 |
+
thread_info = f'thread:{current_thread_name}::{current_thread_id}::{current_thread_alive}'
|
160 |
+
if use_color:
|
161 |
+
thread_info = ansi_color_str(thread_info,fg='yellow', bg='bright_yellow')
|
162 |
+
|
163 |
+
formatted_message = f'{thread_info}:{message}'
|
164 |
+
|
165 |
+
return formatted_message
|
166 |
+
|
167 |
+
|
168 |
+
import inspect
|
169 |
+
import os
|
170 |
+
|
171 |
+
# Get the current frame
|
172 |
+
def get_prev_frame(steps=1):
|
173 |
+
curr_frame = inspect.currentframe()
|
174 |
+
|
175 |
+
# Traverse back the specified number of steps in the call stack
|
176 |
+
for _ in range(steps):
|
177 |
+
if curr_frame is not None:
|
178 |
+
curr_frame = curr_frame.f_back
|
179 |
+
|
180 |
+
if curr_frame is None:
|
181 |
+
return None
|
182 |
+
|
183 |
+
return curr_frame
|
184 |
+
|
185 |
+
|
186 |
+
def get_prev_frame_from_frame(frame, steps=1):
|
187 |
+
curr_frame = frame
|
188 |
+
|
189 |
+
# Traverse back the specified number of steps in the call stack
|
190 |
+
for _ in range(steps):
|
191 |
+
if frame is not None:
|
192 |
+
curr_frame = curr_frame.f_back
|
193 |
+
|
194 |
+
if curr_frame is None:
|
195 |
+
return None
|
196 |
+
|
197 |
+
return curr_frame
|
198 |
+
|
199 |
+
def get_prev_caller_info(message='', use_color=True, steps=99):
|
200 |
+
# Get the current frame
|
201 |
+
curr_frame = inspect.currentframe()
|
202 |
+
caller_frame = curr_frame.f_back
|
203 |
+
|
204 |
+
while not caller_frame.f_back is None:
|
205 |
+
caller_frame = caller_frame.f_back
|
206 |
+
steps -=1
|
207 |
+
if steps <= 0:
|
208 |
+
break
|
209 |
+
|
210 |
+
previous_frame = caller_frame
|
211 |
+
|
212 |
+
# Retrieve the information about the previous frame
|
213 |
+
frame_info = inspect.getframeinfo(previous_frame)
|
214 |
+
# Get the file name where the function was called
|
215 |
+
filename_with_path = frame_info.filename
|
216 |
+
# Extract only the file name
|
217 |
+
filename = os.path.basename(filename_with_path)
|
218 |
+
# Get the line number in the file where the function was called
|
219 |
+
linenumber = frame_info.lineno
|
220 |
+
# Get the function name
|
221 |
+
function = frame_info.function
|
222 |
+
# Format the string to include the passed message
|
223 |
+
caller_link = ansi_link_str(f"file:///{filename_with_path}", f"{filename}::{linenumber}::{function}") if use_color else f"{filename}::{linenumber}::{function}"
|
224 |
+
info_str = f"{caller_link}: {message}"
|
225 |
+
# Clean up to prevent reference cycles
|
226 |
+
del curr_frame
|
227 |
+
del caller_frame
|
228 |
+
del previous_frame
|
229 |
+
return info_str
|
230 |
+
|
231 |
+
|
232 |
+
def get_caller_info(message='', use_color=True):
|
233 |
+
# Get the current frame
|
234 |
+
curr_frame = inspect.currentframe()
|
235 |
+
# Get the caller's frame
|
236 |
+
caller_frame = curr_frame.f_back
|
237 |
+
# Retrieve the information about the caller's frame
|
238 |
+
frame_info = inspect.getframeinfo(caller_frame)
|
239 |
+
# Get the file name where the function was called
|
240 |
+
filename_with_path = frame_info.filename
|
241 |
+
# Extract only the file name
|
242 |
+
filename = os.path.basename(filename_with_path)
|
243 |
+
# Get the line number in the file where the function was called
|
244 |
+
linenumber = frame_info.lineno
|
245 |
+
# get the function name
|
246 |
+
function = frame_info.function
|
247 |
+
# Format the string to include the passed message
|
248 |
+
#caller_link = f"file:///{filename_with_path}", f"{filename}::{linenumber}::{function}"
|
249 |
+
caller_link = ansi_link_str(f"file:///{filename_with_path}", f"{filename}::{linenumber}::{function}") if use_color else f"{filename}::{linenumber}::{function}"
|
250 |
+
#caller_link = f"{filename}::{linenumber}::{function}"
|
251 |
+
|
252 |
+
info_str = f"{caller_link}: {message}" # file://
|
253 |
+
# Clean up to prevent reference cycles
|
254 |
+
del curr_frame
|
255 |
+
del caller_frame
|
256 |
+
return info_str
|
257 |
+
|
258 |
+
|
259 |
+
def get_class_name(obj):
|
260 |
+
return obj.__class__.__name__
|
mew_log/extended_symbols.py
ADDED
@@ -0,0 +1,127 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import enum
|
2 |
+
|
3 |
+
class ExtendedSymbols:
|
4 |
+
class Symbol(enum.Enum):
|
5 |
+
# Correctly differentiating between single and double line symbols
|
6 |
+
TOP_RIGHT_L = '\u2510' # ┐ Single line top right corner
|
7 |
+
BOTTOM_RIGHT_L = '\u2518' # ┘ Single line bottom right corner
|
8 |
+
TOP_LEFT_L = '\u250C' # ┌ Single line top left corner
|
9 |
+
BOTTOM_LEFT_L = '\u2514' # └ Single line bottom left corner
|
10 |
+
RIGHT_T = '\u2524' # ┤ Single line right T
|
11 |
+
LEFT_T = '\u251C' # ├ Single line left T
|
12 |
+
TOP_T = '\u252C' # ┬ Single line top T
|
13 |
+
BOTTOM_T = '\u2534' # ┴ Single line bottom T
|
14 |
+
CROSS = '\u253C' # ┼ Single line cross
|
15 |
+
SINGLE_HORIZONTAL = '\u2500' # ─ Single horizontal line
|
16 |
+
SINGLE_VERTICAL = '\u2502' # │ Single vertical line
|
17 |
+
|
18 |
+
# Double line versions
|
19 |
+
DOUBLE_TOP_RIGHT_L = '\u2557' # ╗ Double line top right corner
|
20 |
+
DOUBLE_BOTTOM_RIGHT_L = '\u255D' # ╝ Double line bottom right corner
|
21 |
+
DOUBLE_TOP_LEFT_L = '\u2554' # ╔ Double line top left corner
|
22 |
+
DOUBLE_BOTTOM_LEFT_L = '\u255A' # ╚ Double line bottom left corner
|
23 |
+
DOUBLE_RIGHT_T = '\u2563' # ╣ Double line right T
|
24 |
+
DOUBLE_LEFT_T = '\u2560' # ╠ Double line left T
|
25 |
+
DOUBLE_TOP_T = '\u2566' # ╦ Double line top T
|
26 |
+
DOUBLE_BOTTOM_T = '\u2569' # ╩ Double line bottom T
|
27 |
+
DOUBLE_CROSS = '\u256C' # ╬ Double line cross
|
28 |
+
DOUBLE_HORIZONTAL = '\u2550' # ═ Double horizontal line
|
29 |
+
DOUBLE_VERTICAL = '\u2551' # ║ Double vertical line
|
30 |
+
|
31 |
+
SOLID_BLOCK = '\u2588' # █ Solid block
|
32 |
+
TOP_HALF_BLOCK = '\u2580' # ▀ Top half block
|
33 |
+
BOTTOM_HALF_BLOCK = '\u2584' # ▄ Bottom half block
|
34 |
+
SHADED_BLOCK = '\u2592' # ▒ Shaded block
|
35 |
+
|
36 |
+
def __getattr__(self, name):
|
37 |
+
try:
|
38 |
+
return self.Symbol[name].value
|
39 |
+
except KeyError:
|
40 |
+
raise AttributeError(f"No such symbol: {name}")
|
41 |
+
|
42 |
+
# Make ExtendedSymbols an instance so it can be used directly
|
43 |
+
ExtendedSymbols = ExtendedSymbols()
|
44 |
+
|
45 |
+
|
46 |
+
|
47 |
+
|
48 |
+
def draw_box(msg=None, width=None, height=None, title=None, line_type='single'):
|
49 |
+
# Use extended symbols correctly with Unicode
|
50 |
+
line_char = ExtendedSymbols.SINGLE_HORIZONTAL if line_type == 'single' else ExtendedSymbols.DOUBLE_HORIZONTAL
|
51 |
+
corner_tl = ExtendedSymbols.TOP_LEFT_L if line_type == 'single' else ExtendedSymbols.DOUBLE_TOP_LEFT_L
|
52 |
+
corner_tr = ExtendedSymbols.TOP_RIGHT_L if line_type == 'single' else ExtendedSymbols.DOUBLE_TOP_RIGHT_L
|
53 |
+
corner_bl = ExtendedSymbols.BOTTOM_LEFT_L if line_type == 'single' else ExtendedSymbols.DOUBLE_BOTTOM_LEFT_L
|
54 |
+
corner_br = ExtendedSymbols.BOTTOM_RIGHT_L if line_type == 'single' else ExtendedSymbols.DOUBLE_BOTTOM_RIGHT_L
|
55 |
+
vertical_line = ExtendedSymbols.SINGLE_VERTICAL if line_type == 'single' else ExtendedSymbols.DOUBLE_VERTICAL
|
56 |
+
|
57 |
+
# Determine width based on message length if not specified
|
58 |
+
if msg and (width is None or height is None):
|
59 |
+
width = max(len(msg), width if width else 0) + 2 # Add padding
|
60 |
+
height = 3 # Minimum box height with top, middle, and bottom
|
61 |
+
|
62 |
+
# Use default dimensions if none provided
|
63 |
+
if width is None:
|
64 |
+
width = 10 # Default width
|
65 |
+
if height is None:
|
66 |
+
height = 3 # Default minimum height
|
67 |
+
|
68 |
+
# Draw top border
|
69 |
+
top_border = f"{corner_tl}{line_char * width}{corner_tr}"
|
70 |
+
if title:
|
71 |
+
title_str = f" {title} "
|
72 |
+
title_length = len(title_str)
|
73 |
+
pre_title_length = (width - title_length) // 2
|
74 |
+
post_title_length = width - title_length - pre_title_length
|
75 |
+
top_border = f"{corner_tl}{line_char * pre_title_length}{title_str}{line_char * post_title_length}{corner_tr}"
|
76 |
+
|
77 |
+
# Prepare message line or blank lines
|
78 |
+
if msg:
|
79 |
+
message_line = f"{vertical_line} {msg} {vertical_line}"
|
80 |
+
if len(msg) < width - 2:
|
81 |
+
msg = msg + ' ' * (width - 2 - len(msg))
|
82 |
+
message_line = f"{vertical_line}{msg}{vertical_line}"
|
83 |
+
else:
|
84 |
+
message_line = f"{vertical_line}{' ' * (width)}{vertical_line}"
|
85 |
+
|
86 |
+
# Draw middle lines
|
87 |
+
middle_lines = (f"{vertical_line}{' ' * width}{vertical_line}\n") * (height - 2)
|
88 |
+
middle_lines = middle_lines[:-1] # Remove the last newline for proper formatting
|
89 |
+
|
90 |
+
# Draw bottom border
|
91 |
+
bottom_border = f"{corner_bl}{line_char * width}{corner_br}"
|
92 |
+
|
93 |
+
# Combine all parts
|
94 |
+
box = f"{top_border}\n{message_line}\n{middle_lines}\n{bottom_border}"
|
95 |
+
print(box)
|
96 |
+
|
97 |
+
def format_arrow(root_symbol=None, line_type='single', length=10):
|
98 |
+
"""
|
99 |
+
Draw an arrow with a specified root symbol, line type, and length.
|
100 |
+
"""
|
101 |
+
symbol_map = {
|
102 |
+
'L': {
|
103 |
+
'single': ExtendedSymbols.BOTTOM_LEFT_L,
|
104 |
+
'double': ExtendedSymbols.DOUBLE_BOTTOM_LEFT_L,
|
105 |
+
},
|
106 |
+
'T': {
|
107 |
+
'single': ExtendedSymbols.LEFT_T,
|
108 |
+
'double': ExtendedSymbols.DOUBLE_LEFT_T
|
109 |
+
},
|
110 |
+
'+': {
|
111 |
+
'single': ExtendedSymbols.CROSS,
|
112 |
+
'double': ExtendedSymbols.DOUBLE_CROSS
|
113 |
+
}
|
114 |
+
}
|
115 |
+
|
116 |
+
# Default to a single horizontal line if undefined symbol or line type
|
117 |
+
chosen_symbol = symbol_map.get(root_symbol, {}).get(line_type, '')
|
118 |
+
line_char = ExtendedSymbols.DOUBLE_HORIZONTAL if line_type == 'double' else ExtendedSymbols.SINGLE_HORIZONTAL
|
119 |
+
|
120 |
+
# Construct the arrow string
|
121 |
+
return chosen_symbol + line_char * length + ">"
|
122 |
+
|
123 |
+
|
124 |
+
def draw_arrow(root_symbol=None, line_type='single', length=10):
|
125 |
+
format_arrow_str = format_arrow(root_symbol, line_type, length)
|
126 |
+
print(format_arrow_str)
|
127 |
+
|
mew_log/log_utils.py
ADDED
@@ -0,0 +1,502 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Standard Library Imports
|
2 |
+
import datetime
|
3 |
+
import threading
|
4 |
+
import time
|
5 |
+
import traceback
|
6 |
+
from pathlib import Path
|
7 |
+
|
8 |
+
import enum
|
9 |
+
|
10 |
+
# Third-Party Imports
|
11 |
+
from loguru import logger
|
12 |
+
|
13 |
+
logger.add("./data/log/file_{time}.log", rotation="500 MB") # Automatically rotate too big log files
|
14 |
+
|
15 |
+
|
16 |
+
# Local Imports
|
17 |
+
from ..mew_log.ansi_utils import ansi_color_str, ansi_link_str
|
18 |
+
from ..mew_log.attr_utils import get_prev_caller_info, get_caller_info, get_thread_info, get_prev_frame, get_prev_frame_from_frame, is_of_type
|
19 |
+
from ..mew_log.table_utils import format_property, format_table
|
20 |
+
from ..mew_log.extended_symbols import ExtendedSymbols, draw_box, draw_arrow, format_arrow
|
21 |
+
#from ..mew_log.mew_context import MewContext, resolve_context
|
22 |
+
|
23 |
+
# Constants
|
24 |
+
lastLog = None # Stores the most recent log entry
|
25 |
+
debug_log = [] # Stores all log entries
|
26 |
+
startTime=time.time()
|
27 |
+
#settings vars
|
28 |
+
enable_log = True
|
29 |
+
enable_log_to_file = False
|
30 |
+
log_filename = "log"
|
31 |
+
log_file_path = "data/log/log.txt"
|
32 |
+
enable_fancy = True
|
33 |
+
enable_use_color = True
|
34 |
+
|
35 |
+
type_color='yellow'
|
36 |
+
value_color='bright_yellow'
|
37 |
+
arrow_color = 'bright_white'
|
38 |
+
|
39 |
+
def allow_curly_braces(original_string):
|
40 |
+
if "{" in original_string or "}" in original_string:
|
41 |
+
escaped_string = original_string.replace("{", "{{").replace("}", "}}")
|
42 |
+
#print("Escaped String:", escaped_string) # Debug output
|
43 |
+
return escaped_string
|
44 |
+
return original_string
|
45 |
+
|
46 |
+
def format_arg(arg, use_color, fancy):
|
47 |
+
def is_complex_type(obj):
|
48 |
+
return is_of_type(obj, (list, set, dict)) or hasattr(obj, '__dict__')
|
49 |
+
|
50 |
+
type_str = f"<{type(arg).__name__}>"
|
51 |
+
value_str = repr(arg) if not isinstance(arg, str) and not fancy else format_table(arg, use_color=use_color)
|
52 |
+
newline_if_needed = '\n' if is_complex_type(arg) else ""
|
53 |
+
formatted_arg = f"{type_str}:{newline_if_needed}{value_str}"
|
54 |
+
formatted_arg = allow_curly_braces(formatted_arg)
|
55 |
+
return ansi_color_str(formatted_arg, fg=value_color) if use_color else formatted_arg
|
56 |
+
|
57 |
+
|
58 |
+
def format_kwarg(kw, arg, use_color, fancy):
|
59 |
+
name_str = ansi_color_str(kw, fg=kw_color) if use_color else kw
|
60 |
+
arg_str = format_arg(arg, use_color, fancy)
|
61 |
+
formatted_arg = f"{name_str}: {arg_str}"
|
62 |
+
return ansi_color_str(formatted_arg, fg=value_color) if use_color else formatted_arg
|
63 |
+
|
64 |
+
def format_args(use_color=enable_use_color, fancy=enable_fancy, *args, **kwargs):
|
65 |
+
formatted_args = [ format_arg(arg, use_color, fancy) for arg in args]
|
66 |
+
formatted_kwargs = {k: format_kwarg(k, v, use_color, fancy) for k, v in kwargs.items()}
|
67 |
+
return formatted_args, formatted_kwargs
|
68 |
+
|
69 |
+
def make_formatted_msg(msg, *args, **kwargs):
|
70 |
+
#fetch format vars from kwargs
|
71 |
+
fancy = kwargs.pop('fancy', enable_fancy)
|
72 |
+
use_color = kwargs.pop('use_color', enable_use_color)
|
73 |
+
context_depth = kwargs.pop('context_depth', None)
|
74 |
+
context = kwargs.pop('context', None)
|
75 |
+
# print(f"args={args}")
|
76 |
+
# print(f"kwargs={kwargs}")
|
77 |
+
|
78 |
+
formatted_args, formatted_kwargs = format_args(use_color, fancy, *args, **kwargs)
|
79 |
+
|
80 |
+
msgf = stringf(msg, *formatted_args, **formatted_kwargs)
|
81 |
+
|
82 |
+
#context = MewContext.resolve_context(context, depth, steps=5)
|
83 |
+
if fancy:
|
84 |
+
formatted_msg = f"\n ~ | {get_timestamp()} | {get_thread_info(get_prev_caller_info(msgf, use_color=use_color, steps=2))}"
|
85 |
+
else:
|
86 |
+
formatted_msg = f"\n ~ | {get_timestamp()} | {context}:{msgf}"
|
87 |
+
|
88 |
+
return formatted_msg
|
89 |
+
|
90 |
+
def stringf(s: str, *args, **kwargs):
|
91 |
+
# s = allow_curly_braces(s) # Uncomment or modify as needed
|
92 |
+
if not args and not kwargs:
|
93 |
+
#print("Both args and kwargs are empty.")
|
94 |
+
return s
|
95 |
+
else:
|
96 |
+
return s.format(*args, **kwargs)
|
97 |
+
# if not args:
|
98 |
+
# print("Args is empty.")
|
99 |
+
# if not kwargs:
|
100 |
+
# print("Kwargs is empty.")
|
101 |
+
|
102 |
+
# print(s)
|
103 |
+
# print(f"args: {args}")
|
104 |
+
# print(f"kwargs: {kwargs}")
|
105 |
+
# return s.format(*args, **kwargs)
|
106 |
+
|
107 |
+
# No arguments or keyword arguments
|
108 |
+
# stringf("Hello, World!")
|
109 |
+
|
110 |
+
# # With positional arguments
|
111 |
+
# stringf("Hello, {}!", "World")
|
112 |
+
|
113 |
+
# # With keyword arguments
|
114 |
+
# stringf("Hello, {name}!", name="Alice")
|
115 |
+
|
116 |
+
# # With both args and kwargs empty (only prints "Hello, World!")
|
117 |
+
# stringf("Hello, World!")
|
118 |
+
|
119 |
+
def ensure_directory_exists(directory):
|
120 |
+
path = Path(directory)
|
121 |
+
path.mkdir(parents=True, exist_ok=True)
|
122 |
+
|
123 |
+
def ensure_file_exists(file_path):
|
124 |
+
path = Path(file_path)
|
125 |
+
path.touch(exist_ok=True)
|
126 |
+
|
127 |
+
#from ..mew_log.mew_log_helper import MewLogHelper
|
128 |
+
|
129 |
+
# Logging Configuration
|
130 |
+
|
131 |
+
# Configure Loguru's logger to use a custom format
|
132 |
+
import sys
|
133 |
+
import io
|
134 |
+
|
135 |
+
# Set UTF-8 encoding for stdout and stderr
|
136 |
+
#sys.stdout = io.TextIOWrapper(sys.stdout.detach(), encoding='utf-8', line_buffering=True)
|
137 |
+
#sys.stderr = io.TextIOWrapper(sys.stderr.detach(), encoding='utf-8', line_buffering=True)
|
138 |
+
|
139 |
+
#logger.add(sys.stdout, format="<green>{time:YYYY-MM-DD HH:mm:ss}</green> | {level} | {message}", backtrace=True)
|
140 |
+
|
141 |
+
logger.configure(handlers=[
|
142 |
+
{
|
143 |
+
"sink": sys.stdout,
|
144 |
+
"format": "<green>{time:YYYY-MM-DD HH:mm:ss}</green> | {level} | {message}", # | <cyan>{file}:{line}:{function}</cyan>
|
145 |
+
"level": "INFO"
|
146 |
+
},
|
147 |
+
{
|
148 |
+
"sink": sys.stdout,
|
149 |
+
"format": "<green>{time:YYYY-MM-DD HH:mm:ss}</green> | {level} | {message}", # | <yellow>{file}:{line}:{function}</yellow>
|
150 |
+
"level": "WARNING"
|
151 |
+
},
|
152 |
+
{
|
153 |
+
"sink": sys.stderr,
|
154 |
+
"format": "<green>{time:YYYY-MM-DD HH:mm:ss}</green> | {level} | {message}", #| <red>{file}:{line}:{function}</red>
|
155 |
+
"level": "ERROR"
|
156 |
+
}
|
157 |
+
])
|
158 |
+
|
159 |
+
|
160 |
+
# Logging Configuration
|
161 |
+
def init_log(log_config):
|
162 |
+
global enable_log_to_file, enable_log, log_filename, log_file_path
|
163 |
+
enable_log_to_file = log_config['log_to_file']
|
164 |
+
clear_file_first = log_config['clear_file_first']
|
165 |
+
enable_log = log_config['enable_log']
|
166 |
+
log_filename = log_config['log_filename']
|
167 |
+
log_dir = 'data/log/'
|
168 |
+
ensure_directory_exists(log_dir)
|
169 |
+
log_file_path = f"data/log/{log_filename}.txt"
|
170 |
+
ensure_file_exists(log_file_path)
|
171 |
+
with open(log_file_path, 'w') as fp:
|
172 |
+
pass
|
173 |
+
|
174 |
+
|
175 |
+
print(f'init_log: \n ~ enable_log: {enable_log} \n ~ enable_log_to_file: {enable_log_to_file} \n ~ log_file_path: {log_file_path}')
|
176 |
+
|
177 |
+
|
178 |
+
def _update_log(formatted_msg_str):
|
179 |
+
global lastLog, debug_log, enable_log_to_file
|
180 |
+
|
181 |
+
lastLog = formatted_msg_str
|
182 |
+
|
183 |
+
debug_log.append(lastLog)
|
184 |
+
|
185 |
+
if enable_log_to_file:
|
186 |
+
log_file(lastLog)
|
187 |
+
|
188 |
+
|
189 |
+
def log_file(msg, file_path = None):
|
190 |
+
global log_file_path
|
191 |
+
if not enable_log:
|
192 |
+
return
|
193 |
+
if file_path is None:
|
194 |
+
file_path = log_file_path
|
195 |
+
|
196 |
+
curTime = time.time()
|
197 |
+
elapsedTime = curTime - startTime
|
198 |
+
with open(log_file_path, 'w') as fp:
|
199 |
+
fp.write(f"\n ~ | time={elapsedTime:.2f}s ~ {msg}")
|
200 |
+
#logger.add(log_file_path, format="time={elapsedTime:.2f}s ~ {msg}")
|
201 |
+
|
202 |
+
|
203 |
+
|
204 |
+
def log_info(msg, *args,**kwargs):
|
205 |
+
if not enable_log:
|
206 |
+
return
|
207 |
+
|
208 |
+
formatted_msg_str = make_formatted_msg(msg, *args,**kwargs)
|
209 |
+
|
210 |
+
_update_log(formatted_msg_str)
|
211 |
+
|
212 |
+
formatted_log_str = lastLog
|
213 |
+
print(formatted_log_str)
|
214 |
+
#logger.info(formatted_log_str)
|
215 |
+
|
216 |
+
def log_warning(msg, *args,**kwargs):
|
217 |
+
if not enable_log:
|
218 |
+
return
|
219 |
+
|
220 |
+
formatted_msg_str = make_formatted_msg(msg, *args,**kwargs)
|
221 |
+
|
222 |
+
_update_log(formatted_msg_str)
|
223 |
+
|
224 |
+
formatted_log_str = lastLog
|
225 |
+
|
226 |
+
#print(formatted_log_str)
|
227 |
+
|
228 |
+
logger.warning(formatted_log_str)
|
229 |
+
|
230 |
+
def log_error(msg, e, *args,**kwargs):
|
231 |
+
if not enable_log:
|
232 |
+
return
|
233 |
+
|
234 |
+
formatted_msg_str = f"\n===^_^===\n{make_formatted_msg(msg, *args, **kwargs)}"
|
235 |
+
formatted_msg_str += f"{trace_error(e)}\n===>_<==="
|
236 |
+
|
237 |
+
_update_log(formatted_msg_str)
|
238 |
+
|
239 |
+
formatted_log_str = lastLog
|
240 |
+
|
241 |
+
#print(formatted_log_str)
|
242 |
+
|
243 |
+
logger.error(formatted_log_str)
|
244 |
+
|
245 |
+
|
246 |
+
import functools
|
247 |
+
import traceback
|
248 |
+
|
249 |
+
def log_function(func):
|
250 |
+
@functools.wraps(func)
|
251 |
+
def wrapper(*args, **kwargs):
|
252 |
+
def get_func_args():
|
253 |
+
arg_names = func.__code__.co_varnames[:func.__code__.co_argcount]
|
254 |
+
func_args = dict(zip(arg_names, args))
|
255 |
+
func_args.update(kwargs)
|
256 |
+
|
257 |
+
try:
|
258 |
+
arg_names = func.__code__.co_varnames[:func.__code__.co_argcount]
|
259 |
+
func_args = dict(zip(arg_names, args))
|
260 |
+
func_args.update(kwargs)
|
261 |
+
func_args_str = format_table(func_args, headers=None, tablefmt='simple')
|
262 |
+
#print(get_prev_caller_info(f"({func_args_str} )"))
|
263 |
+
result = func(*args, **kwargs)
|
264 |
+
print(get_thread_info(get_prev_caller_info(f"({func_args} ) -> result: {format_arg(result, use_color=True, fancy=True)}")))
|
265 |
+
return result
|
266 |
+
except Exception as e:
|
267 |
+
arg_names = func.__code__.co_varnames[:func.__code__.co_argcount]
|
268 |
+
func_args = dict(zip(arg_names, args))
|
269 |
+
func_args.update(kwargs)
|
270 |
+
|
271 |
+
logger.error(f"Exception: {str(e)}")
|
272 |
+
logger.error(f"Format Traceback: {traceback.format_exc()}")
|
273 |
+
raise # Re-raise the exception after logging
|
274 |
+
return wrapper
|
275 |
+
|
276 |
+
# @log_function
|
277 |
+
# def dive(a, b):
|
278 |
+
# return a / b
|
279 |
+
|
280 |
+
# dive(2,3)
|
281 |
+
|
282 |
+
# dive(12,3)
|
283 |
+
|
284 |
+
# dive(14,3)
|
285 |
+
|
286 |
+
# dive(21,3)
|
287 |
+
|
288 |
+
# dive(21,0)
|
289 |
+
|
290 |
+
error_color = 'bright_red'
|
291 |
+
|
292 |
+
def format_traceback(tb_entry, depth=0, use_color=enable_use_color):
|
293 |
+
arrow_str = format_arrow('T', 'double', 3 + 2 * depth)
|
294 |
+
|
295 |
+
filename, lineno, funcname, line = tb_entry
|
296 |
+
file_path = f"file:///{filename}"
|
297 |
+
file_link = f"{filename}::{lineno}::{funcname}"
|
298 |
+
trace_link = ansi_link_str(file_path, file_link, link_color="bright_magenta") if use_color else file_link
|
299 |
+
|
300 |
+
arrow_str2 = format_arrow('L', 'double', 3 + 2 * (depth+1))
|
301 |
+
#(" " * (3 + 2 * depth))+
|
302 |
+
formatted_tb_str = f'\n ~ | {arrow_str}trace({depth}): {trace_link}\n ~ | {arrow_str2} Line: "{line}"'
|
303 |
+
return ansi_color_str(formatted_tb_str, fg=error_color) if use_color else formatted_tb_str
|
304 |
+
|
305 |
+
def trace_error(e, use_color=True, fancy=True):
|
306 |
+
exception_str = f"\n ~ | {ExtendedSymbols.DOUBLE_VERTICAL}Exception: {repr(e)}"
|
307 |
+
out_str = ansi_color_str(exception_str, fg=error_color) if use_color else exception_str
|
308 |
+
if isinstance(e, BaseException):
|
309 |
+
arrow_str = format_arrow('T', 'double', 3)
|
310 |
+
traceback_obj = e.__traceback__
|
311 |
+
if traceback_obj:
|
312 |
+
file_path = f"file:///{traceback_obj.tb_frame.f_code.co_filename}"
|
313 |
+
file_link = f"{traceback_obj.tb_frame.f_code.co_filename}::{traceback_obj.tb_lineno}"
|
314 |
+
trace_link = ansi_link_str(file_path, file_link, link_color="bright_magenta") if use_color else file_link
|
315 |
+
formatted_tb_str = f"\n ~ | {arrow_str}trace(0): {trace_link}"
|
316 |
+
out_str += ansi_color_str(formatted_tb_str, fg=error_color) if use_color else formatted_tb_str
|
317 |
+
|
318 |
+
tb = traceback.extract_tb(traceback_obj)
|
319 |
+
for i, tb_entry in enumerate(tb):
|
320 |
+
out_str += format_traceback(tb_entry, depth=i+1, use_color=use_color)
|
321 |
+
|
322 |
+
return ansi_color_str(out_str, fg=error_color) if use_color else out_str
|
323 |
+
|
324 |
+
|
325 |
+
# log_info(' ~ | test log_info: {}', 1)
|
326 |
+
# log_warning(' ~ | test log_warning: {}', 2)
|
327 |
+
|
328 |
+
|
329 |
+
def log_text_input(key: str):
|
330 |
+
log_info(f'\n ~ | Text Input[{key}]: ')
|
331 |
+
input_text = input()
|
332 |
+
log_info(f'\n ~ | Recv input: "{input_text}"')
|
333 |
+
return input_text
|
334 |
+
|
335 |
+
|
336 |
+
def format_duration(seconds):
|
337 |
+
hours, remainder = divmod(seconds, 3600)
|
338 |
+
minutes, seconds = divmod(remainder, 60)
|
339 |
+
return f"{int(hours)}h {int(minutes)}m {int(seconds)}s"
|
340 |
+
|
341 |
+
def parse_duration(duration_str):
|
342 |
+
parts = duration_str.split()
|
343 |
+
hours = int(parts[0][:-1]) if 'h' in parts[0] else 0
|
344 |
+
minutes = int(parts[1][:-1]) if 'm' in parts[1] else 0
|
345 |
+
seconds = int(parts[2][:-1]) if 's' in parts[2] else 0
|
346 |
+
return hours * 3600 + minutes * 60 + seconds
|
347 |
+
|
348 |
+
|
349 |
+
def get_timestamp():
|
350 |
+
return datetime.datetime.now().strftime("%Y-%b-%d %H:%M:%S")
|
351 |
+
|
352 |
+
|
353 |
+
def format_path(directory, depth=0, max_depth=3, use_color=True):
|
354 |
+
"""Formats the directory structure as a string up to a specified depth, with optional color.
|
355 |
+
|
356 |
+
Args:
|
357 |
+
directory (str or Path): The directory path to format.
|
358 |
+
depth (int): The current depth in the directory structure.
|
359 |
+
max_depth (int): The maximum depth to format.
|
360 |
+
use_color (bool): Whether to apply color to the output.
|
361 |
+
|
362 |
+
Returns:
|
363 |
+
str: The formatted directory structure.
|
364 |
+
"""
|
365 |
+
if max_depth is not None and depth > max_depth:
|
366 |
+
return ""
|
367 |
+
|
368 |
+
directory_path = Path(directory)
|
369 |
+
if not directory_path.is_dir():
|
370 |
+
return "The specified path is not a directory.\n"
|
371 |
+
|
372 |
+
# Build the formatted string, applying color if requested
|
373 |
+
line_prefix = "│ " * depth + "├── "
|
374 |
+
directory_name = directory_path.name
|
375 |
+
if use_color:
|
376 |
+
line_prefix = ansi_color_str(line_prefix, fg='cyan') # Assuming color_str function exists
|
377 |
+
directory_name = ansi_color_str(directory_name, fg='green') # Adjust colors as needed
|
378 |
+
|
379 |
+
formatted_str = line_prefix + directory_name + "\n"
|
380 |
+
|
381 |
+
# Sort directory contents for consistent order
|
382 |
+
sorted_items = sorted(directory_path.iterdir(), key=lambda x: (x.is_file(), x.name))
|
383 |
+
|
384 |
+
for item in sorted_items:
|
385 |
+
if item.is_dir():
|
386 |
+
# Recursive call for directories, increasing the depth
|
387 |
+
formatted_str += format_path(item, depth + 1, max_depth, use_color)
|
388 |
+
else:
|
389 |
+
# Include file name with indentation, applying color if requested
|
390 |
+
file_line = "│ " * (depth + 1) + "├── " + item.name
|
391 |
+
if use_color:
|
392 |
+
file_line = ansi_color_str(file_line, fg='white') # Example color, adjust as needed
|
393 |
+
formatted_str += file_line + "\n"
|
394 |
+
|
395 |
+
return formatted_str
|
396 |
+
|
397 |
+
|
398 |
+
# Spinner Class
|
399 |
+
class LogSpinner:
|
400 |
+
def __init__(self, message="", rainbow=True, anim_speed=0.1):
|
401 |
+
self.message = message
|
402 |
+
self.speed = anim_speed
|
403 |
+
self.colors = ['\033[31m', '\033[33m', '\033[32m', '\033[34m', '\033[35m', '\033[36m']
|
404 |
+
self.spinner = ['⠇', '⠋', '⠙', '⠸', '⠼', '⠴', '⠦', '⠧']
|
405 |
+
|
406 |
+
#self.spinner = ['|', '/', '-', '\\']
|
407 |
+
self.rainbow = rainbow
|
408 |
+
|
409 |
+
def __enter__(self):
|
410 |
+
self.stop_spinner = False
|
411 |
+
self.startTime = time.time()
|
412 |
+
# spinner_text = f'^_^ | {self.message} - time: {0.0:.2f}s | Begin!'
|
413 |
+
# print(spinner_text)
|
414 |
+
self.spinner_thread = threading.Thread(target=self.spin)
|
415 |
+
self.spinner_thread.start()
|
416 |
+
return self
|
417 |
+
|
418 |
+
def __exit__(self, exc_type, exc_value, traceback):
|
419 |
+
self.stop_spinner = True
|
420 |
+
self.spinner_thread.join()
|
421 |
+
|
422 |
+
curTime = time.time()
|
423 |
+
spinner_text = f'>_< | {self.message} - time: {format_duration(curTime - self.startTime)} | Done!'
|
424 |
+
print(spinner_text)
|
425 |
+
|
426 |
+
def spin(self):
|
427 |
+
while not self.stop_spinner:
|
428 |
+
#self.update_spinner()
|
429 |
+
if self.rainbow: self.update_color_spinner()
|
430 |
+
else: self.update_spinner()
|
431 |
+
|
432 |
+
def update_spinner(self):
|
433 |
+
for char in self.spinner:
|
434 |
+
curTime = time.time()
|
435 |
+
spinner_text = f'{self.message} - time: {format_duration(curTime - self.startTime)} |'
|
436 |
+
print(f'\r ^_^ | {char} | {spinner_text}', end='\r', flush=True)
|
437 |
+
time.sleep(self.speed)
|
438 |
+
|
439 |
+
def update_color_spinner(self):
|
440 |
+
for color in self.colors:
|
441 |
+
for char in self.spinner:
|
442 |
+
curTime = time.time()
|
443 |
+
spinner_text = f'{self.message} - time: {format_duration(curTime - self.startTime)} |'
|
444 |
+
print(f'^_^ | {color}{char}\033[0m | {spinner_text}', end='\r', flush=True)
|
445 |
+
time.sleep(self.speed)
|
446 |
+
|
447 |
+
|
448 |
+
def run_unit_test():
|
449 |
+
# Example usage:
|
450 |
+
class Person:
|
451 |
+
def __init__(self, name, age, city):
|
452 |
+
self.name = name
|
453 |
+
self.age = age
|
454 |
+
self.city = city
|
455 |
+
|
456 |
+
# Create instances of Person
|
457 |
+
person1 = Person('John', 30, 'New York')
|
458 |
+
person2 = Person('Alice', 25, 'Los Angeles')
|
459 |
+
person3 = Person('Bob', 30, 'Hong Kong')
|
460 |
+
person4 = Person('Charlie', 35, 'Shanghai')
|
461 |
+
person5 = Person('David', 40, 'Beijing')
|
462 |
+
|
463 |
+
# Define other data structures
|
464 |
+
data_dict = {'Name': 'John', 'Age': 30, 'City': 'New York'}
|
465 |
+
data_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
|
466 |
+
data_set = {'apple', 'banana', 'orange'}
|
467 |
+
data_dict_of_lists = {'Name': ['John', 'Alice'], 'Age': [30, 25], 'City': ['New York', 'Los Angeles']}
|
468 |
+
data_list_of_dicts = [person1.__dict__, person2.__dict__]
|
469 |
+
data_dict_of_dicts = {'Person1': person1.__dict__, 'Person2': person2.__dict__}
|
470 |
+
list_of_objs = [person3, person4, person5]
|
471 |
+
data_list_of_lists = [['John', 30, 'New York'], ['Alice', 25, 'Los Angeles']]
|
472 |
+
dict_of_objs = {'Alice': person3, 'Bob': person4, 'Charlie': person5}
|
473 |
+
dict_of_list_of_objs = {'Group1': [person3, person4], 'Group2': [person5]}
|
474 |
+
|
475 |
+
# Combine all data into a complex structure
|
476 |
+
complex_data = {
|
477 |
+
'data_dict': data_dict,
|
478 |
+
'data_list': data_list,
|
479 |
+
'data_set': data_set,
|
480 |
+
'data_dict_of_lists': data_dict_of_lists,
|
481 |
+
'data_list_of_dicts': data_list_of_dicts,
|
482 |
+
'data_dict_of_dicts': data_dict_of_dicts,
|
483 |
+
'list_of_objs': list_of_objs,
|
484 |
+
'data_list_of_lists': data_list_of_lists,
|
485 |
+
'dict_of_objs': dict_of_objs,
|
486 |
+
'dict_of_list_of_objs': dict_of_list_of_objs
|
487 |
+
}
|
488 |
+
|
489 |
+
# Log each unique data structure
|
490 |
+
log_info("Data Dictionary: {}", data_dict)
|
491 |
+
log_info("Data List: {}", data_list)
|
492 |
+
log_info("Data Set: {}", data_set)
|
493 |
+
log_info("Data Dictionary of Lists: {}", data_dict_of_lists)
|
494 |
+
log_info("Data List of Dicts: {}", data_list_of_dicts)
|
495 |
+
log_info("Data Dictionary of Dicts: {}", data_dict_of_dicts)
|
496 |
+
log_info("List of Objects: {}", list_of_objs)
|
497 |
+
log_info("Data List of Lists: {}", data_list_of_lists)
|
498 |
+
log_info("Dictionary of Objects: {}", dict_of_objs)
|
499 |
+
log_info("Dictionary of List of Objects: {}", dict_of_list_of_objs)
|
500 |
+
|
501 |
+
|
502 |
+
#run_unit_test()
|
mew_log/mem_utils.py
ADDED
@@ -0,0 +1,144 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
|
2 |
+
#from ..mew_core import prettify
|
3 |
+
from ..mew_log.ansi_utils import ansi_color_str
|
4 |
+
from ..mew_log.attr_utils import get_caller_info, get_var_value
|
5 |
+
from ..mew_log.table_utils import format_table
|
6 |
+
|
7 |
+
import sys
|
8 |
+
import types
|
9 |
+
import objgraph
|
10 |
+
|
11 |
+
one_mb = 2**20
|
12 |
+
|
13 |
+
def bytes_to_mb(size_in_bytes, inv_bytes_in_mb=(1.0 / (1024 * 1024))):
|
14 |
+
"""
|
15 |
+
Convert size from bytes to megabytes.
|
16 |
+
"""
|
17 |
+
return size_in_bytes * inv_bytes_in_mb
|
18 |
+
|
19 |
+
|
20 |
+
def bytes_to_kb(size_in_bytes, inv_bytes_in_kb=(1.0 / (1024))):
|
21 |
+
"""
|
22 |
+
Convert size from bytes to kilobytes.
|
23 |
+
"""
|
24 |
+
return size_in_bytes * inv_bytes_in_kb
|
25 |
+
|
26 |
+
def format_mem_size_value(size_bytes, use_color=True):
|
27 |
+
size_b = size_bytes
|
28 |
+
size_mb = bytes_to_mb(size_b)
|
29 |
+
size_kb = bytes_to_kb(size_b)
|
30 |
+
|
31 |
+
size_mb_str = f"{size_mb:.4f} MB"
|
32 |
+
size_kb_str = f"{size_kb:.4f} KB"
|
33 |
+
size_b_str = f"{size_b} B"
|
34 |
+
|
35 |
+
# Determine which size to display based on the criteria
|
36 |
+
if size_mb > 0.1: # More than 0.1 MB
|
37 |
+
display_str = size_mb_str
|
38 |
+
elif size_kb > 0.1: # More than 0.1 KB but less than 0.1 MB
|
39 |
+
display_str = size_kb_str
|
40 |
+
else: # Less than 0.1 KB
|
41 |
+
display_str = size_b_str
|
42 |
+
|
43 |
+
# Apply color if use_color is True
|
44 |
+
if use_color:
|
45 |
+
display_str = ansi_color_str(display_str, fg='cyan')
|
46 |
+
|
47 |
+
return f"Mem Size: {display_str}" # Display the selected size
|
48 |
+
|
49 |
+
|
50 |
+
def format_size_bytes(obj, use_color=True):
|
51 |
+
size_b, size_mb = get_mem_size(obj)
|
52 |
+
size_kb = bytes_to_kb(size_b)
|
53 |
+
size_mb_str = f"{size_mb:.4f} MB"
|
54 |
+
size_kb_str = f"{size_kb:.4f} KB"
|
55 |
+
size_b_str = f"{size_b} B"
|
56 |
+
|
57 |
+
# Determine which size to display based on the criteria
|
58 |
+
if size_mb > 0.1: # More than 0.1 MB
|
59 |
+
display_str = size_mb_str
|
60 |
+
elif size_kb > 0.1: # More than 0.1 KB but less than 0.1 MB
|
61 |
+
display_str = size_kb_str
|
62 |
+
else: # Less than 0.1 KB
|
63 |
+
display_str = size_b_str
|
64 |
+
|
65 |
+
# Apply color if use_color is True
|
66 |
+
if use_color:
|
67 |
+
display_str = ansi_color_str(display_str, fg='cyan')
|
68 |
+
|
69 |
+
return f"Mem Size: {display_str}" # Display the selected size
|
70 |
+
|
71 |
+
|
72 |
+
|
73 |
+
def get_mem_size(obj) -> tuple:
|
74 |
+
"""
|
75 |
+
Get the size of an object as a tuple ( bytes , megabytes )
|
76 |
+
"""
|
77 |
+
def _get_mem_size(obj):
|
78 |
+
size = sys.getsizeof(obj)
|
79 |
+
|
80 |
+
if isinstance(obj, types.GeneratorType):
|
81 |
+
# Generators don't have __sizeof__, so calculate size recursively
|
82 |
+
size += sum(_get_mem_size(item) for item in obj)
|
83 |
+
elif isinstance(obj, dict):
|
84 |
+
size += sum(_get_mem_size(key) + _get_mem_size(value) for key, value in obj.items())
|
85 |
+
elif hasattr(obj, '__dict__'):
|
86 |
+
# For objects with __dict__, include the size of their attributes
|
87 |
+
size += _get_mem_size(obj.__dict__)
|
88 |
+
|
89 |
+
return size
|
90 |
+
|
91 |
+
size_b = _get_mem_size(obj)
|
92 |
+
size_mb = bytes_to_mb(size_b)
|
93 |
+
|
94 |
+
return size_b, size_mb
|
95 |
+
|
96 |
+
def get_mem_size_breakdown(obj, do_recursive=False) -> dict:
|
97 |
+
"""
|
98 |
+
Get a breakdown of the sizes of the properties of an object.
|
99 |
+
"""
|
100 |
+
|
101 |
+
size_b, size_mb = get_mem_size(obj) #'name': obj, 'value': get_variable_value(obj)
|
102 |
+
breakdown = { 'name': obj,'type': type(obj).__name__, 'size (B, MB)': f"{size_b} B | {size_mb:.3f} MB", 'value': get_var_value(obj), }
|
103 |
+
|
104 |
+
if do_recursive:
|
105 |
+
if isinstance(obj, types.GeneratorType):
|
106 |
+
for i, item in enumerate(obj):
|
107 |
+
breakdown['generator_'+type(item).__name__][i] = get_mem_size_breakdown(item)
|
108 |
+
elif isinstance(obj, dict):
|
109 |
+
for key, value in obj.items():
|
110 |
+
if key != 'stat':
|
111 |
+
breakdown[f'[{key}]'] = get_mem_size_breakdown(value)
|
112 |
+
elif hasattr(obj, '__dict__'):
|
113 |
+
breakdown['vars'] = {}
|
114 |
+
for key, value in obj.__dict__.items():
|
115 |
+
if key != 'stat':
|
116 |
+
breakdown['vars'][f'[{key}]'] = get_mem_size_breakdown(value)
|
117 |
+
return breakdown
|
118 |
+
|
119 |
+
def save_obj_graph_img(obj):
|
120 |
+
# Generate and save a graph of objects referencing a specific object
|
121 |
+
graph_filename = f'data/image/obj_graph_{obj.__class__.__name__}.png'
|
122 |
+
objgraph.show_refs([obj], filename=graph_filename)
|
123 |
+
return graph_filename
|
124 |
+
|
125 |
+
def take_mem_growth_snapshot():
|
126 |
+
# Take a snapshot of the objects that have grown since the last snapshot
|
127 |
+
growth_info = objgraph.show_growth()
|
128 |
+
if growth_info:
|
129 |
+
# Convert the growth information to a dictionary
|
130 |
+
growth_dict = {}
|
131 |
+
for line in growth_info.split('\n'):
|
132 |
+
if line.strip(): # Skip empty lines
|
133 |
+
parts = line.split()
|
134 |
+
obj_type = parts[0]
|
135 |
+
prev_count = int(parts[1])
|
136 |
+
growth_count = int(parts[2])
|
137 |
+
growth_dict[obj_type] = {'prev_count': prev_count, 'growth_count': growth_count}
|
138 |
+
#print(format_table(growth_dict))
|
139 |
+
return growth_dict
|
140 |
+
|
141 |
+
|
142 |
+
|
143 |
+
|
144 |
+
|
mew_log/profile_utils.py
ADDED
@@ -0,0 +1,194 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import time
|
2 |
+
|
3 |
+
from ..mew_log.ansi_utils import ansi_color_str
|
4 |
+
from ..mew_log.attr_utils import get_prev_caller_info
|
5 |
+
from ..mew_log.log_utils import log_info, LogSpinner
|
6 |
+
from ..mew_log.table_utils import format_table
|
7 |
+
|
8 |
+
|
9 |
+
time_color = ''
|
10 |
+
|
11 |
+
# helper funcs
|
12 |
+
class ProfileSection:
|
13 |
+
Registry = {} # Class Variable
|
14 |
+
|
15 |
+
def __init__(self, newMsg=None, enable=True, do_print=False):
|
16 |
+
self.msg = newMsg if newMsg else get_prev_caller_info()
|
17 |
+
self.enable = enable
|
18 |
+
self.startTime = 0.0
|
19 |
+
self.endTime = 0.0
|
20 |
+
self.elapsedTime = 0.0
|
21 |
+
self.totalTime = 0.0
|
22 |
+
self.avgTime = 0.0
|
23 |
+
self.numCalls = 0
|
24 |
+
self.do_print = do_print
|
25 |
+
if self.msg in ProfileSection.Registry:
|
26 |
+
p = ProfileSection.Registry[self.msg]
|
27 |
+
self.__dict__ = p.__dict__
|
28 |
+
self.numCalls += 1
|
29 |
+
else:
|
30 |
+
self.numCalls += 1
|
31 |
+
|
32 |
+
if (self.enable):
|
33 |
+
self.start()
|
34 |
+
|
35 |
+
def __del__(self):
|
36 |
+
if (self.enable):
|
37 |
+
self.stop()
|
38 |
+
|
39 |
+
def __enter__(self):
|
40 |
+
# if self.enable:
|
41 |
+
self.start()
|
42 |
+
|
43 |
+
return self
|
44 |
+
|
45 |
+
def __exit__(self, exc_type, exc_value, traceback):
|
46 |
+
# if self.enable:
|
47 |
+
self.stop()
|
48 |
+
|
49 |
+
def start(self):
|
50 |
+
self.startTime = time.time()
|
51 |
+
# if self.do_print : log_info(str(self))
|
52 |
+
|
53 |
+
def stop(self):
|
54 |
+
self.endTime = time.time()
|
55 |
+
self.update_stats()
|
56 |
+
stopMsg = str(self)
|
57 |
+
|
58 |
+
p = create_or_get_profile(self.msg)
|
59 |
+
p.__dict__.update(self.__dict__)
|
60 |
+
if self.do_print: log_info(str(p))
|
61 |
+
|
62 |
+
# st.toast(stopMsg)
|
63 |
+
return stopMsg
|
64 |
+
|
65 |
+
def update_stats(self):
|
66 |
+
self.elapsedTime = self.endTime - self.startTime
|
67 |
+
self.totalTime += self.elapsedTime
|
68 |
+
self.avgTime = self.totalTime / float(self.numCalls)
|
69 |
+
|
70 |
+
def __str__(self, use_color=True):
|
71 |
+
msg_str = self.msg #ansi_color_str(self.msg, fg='green') # Green color for message
|
72 |
+
elapsed_time_str = make_time_str('elapsed', self.elapsedTime)
|
73 |
+
total_time_str = make_time_str('total', self.totalTime)
|
74 |
+
avg_time_str = make_time_str('avg', self.avgTime)
|
75 |
+
calls_str = f'calls={self.numCalls}'
|
76 |
+
if use_color:
|
77 |
+
elapsed_time_str = ansi_color_str(elapsed_time_str, fg='bright_cyan') # Cyan color for elapsed time
|
78 |
+
total_time_str = ansi_color_str(total_time_str, fg='bright_cyan') # Cyan color for total time
|
79 |
+
avg_time_str = ansi_color_str(avg_time_str, fg='bright_cyan') # Cyan color for average time
|
80 |
+
calls_str = ansi_color_str(calls_str, fg='yellow') # Cyan color for calls information
|
81 |
+
|
82 |
+
return f"{msg_str} ~ Elapsed Time: {elapsed_time_str} | Total Time: {total_time_str} | Avg Time: {avg_time_str} | {calls_str}"
|
83 |
+
|
84 |
+
|
85 |
+
from functools import wraps
|
86 |
+
from threading import Thread
|
87 |
+
from rich.progress import Progress, BarColumn, TextColumn, TimeRemainingColumn, TaskProgressColumn
|
88 |
+
from rich.console import Console
|
89 |
+
from rich.style import Style
|
90 |
+
from rich.panel import Panel
|
91 |
+
|
92 |
+
|
93 |
+
def profile_function(func):
|
94 |
+
"""
|
95 |
+
Decorator to track the progress of a function with a spinner.
|
96 |
+
The decorated function should not require explicit progress updates.
|
97 |
+
"""
|
98 |
+
@wraps(func)
|
99 |
+
def wrapper(*args, **kwargs):
|
100 |
+
profile_section = ProfileSection(func.__name__, enable=True, do_print=False)
|
101 |
+
console = Console()
|
102 |
+
with LogSpinner(func.__name__):
|
103 |
+
with profile_section:
|
104 |
+
result = func(*args, **kwargs)
|
105 |
+
|
106 |
+
#profile_section.stop()
|
107 |
+
# Display timing information using rich
|
108 |
+
panel = Panel.fit(
|
109 |
+
f"[bold green]{profile_section.msg}[/bold green]\n"
|
110 |
+
f"[cyan]Elapsed Time:[/cyan] {make_time_str('elapsed', profile_section.elapsedTime)}\n"
|
111 |
+
f"[cyan]Total Time:[/cyan] {make_time_str('total', profile_section.totalTime)}\n"
|
112 |
+
f"[cyan]Avg Time:[/cyan] {make_time_str('avg', profile_section.avgTime)}\n"
|
113 |
+
f"[yellow]Calls:[/yellow] {profile_section.numCalls}",
|
114 |
+
title="Profile Report",
|
115 |
+
border_style="bright_blue"
|
116 |
+
)
|
117 |
+
console.print(panel)
|
118 |
+
#print(str(profile_section))
|
119 |
+
|
120 |
+
|
121 |
+
return result
|
122 |
+
return wrapper
|
123 |
+
|
124 |
+
|
125 |
+
def make_time_str(msg, value):
|
126 |
+
# do something fancy
|
127 |
+
value, time_unit = (value / 60, 'min') if value >= 60 else (value * 1000, 'ms') if value < 0.01 else (value, 's')
|
128 |
+
return f"{msg}={int(value) if value % 1 == 0 else value:.2f} {time_unit}"
|
129 |
+
|
130 |
+
|
131 |
+
def create_or_get_profile(key, enable=False, do_print=False):
|
132 |
+
if key not in ProfileSection.Registry:
|
133 |
+
ProfileSection.Registry[key] = ProfileSection(key, enable, do_print)
|
134 |
+
return ProfileSection.Registry[key]
|
135 |
+
|
136 |
+
|
137 |
+
def profile_start(msg, enable=True, do_print=False):
|
138 |
+
p = create_or_get_profile(msg, enable, do_print)
|
139 |
+
if not enable: p.start()
|
140 |
+
|
141 |
+
|
142 |
+
def profile_stop(msg):
|
143 |
+
if key in ProfileSection.Registry:
|
144 |
+
create_or_get_profile(msg).stop()
|
145 |
+
|
146 |
+
|
147 |
+
def get_profile_registry():
|
148 |
+
return ProfileSection.Registry
|
149 |
+
|
150 |
+
from loguru import logger
|
151 |
+
|
152 |
+
|
153 |
+
def get_profile_reports():
|
154 |
+
reports = [value for value in ProfileSection.Registry.values()]
|
155 |
+
reports.sort(key=lambda x: (x.totalTime, x.avgTime), reverse=True)
|
156 |
+
return reports
|
157 |
+
|
158 |
+
def log_profile_registry(use_color=True):
|
159 |
+
formatted_output = format_profile_registry(use_color=use_color)
|
160 |
+
print(formatted_output)
|
161 |
+
#logger.info(formatted_output)
|
162 |
+
#return formatted_output
|
163 |
+
|
164 |
+
|
165 |
+
|
166 |
+
def allow_curly_braces(original_string):
|
167 |
+
escaped_string = original_string.replace("{", "{{").replace("}", "}}")
|
168 |
+
#print("Escaped String:", escaped_string) # Debug output
|
169 |
+
return escaped_string
|
170 |
+
|
171 |
+
def format_profile_registry(use_color=True):
|
172 |
+
reports = get_profile_reports()
|
173 |
+
out_str = []
|
174 |
+
|
175 |
+
out_str.append('=== Profile Reports ===\n')
|
176 |
+
|
177 |
+
for report in reports:
|
178 |
+
out_str.append(str(report)+'\n')
|
179 |
+
|
180 |
+
out_str.append('===>_<===\n')
|
181 |
+
return ''.join(out_str)
|
182 |
+
|
183 |
+
|
184 |
+
# import random
|
185 |
+
|
186 |
+
# def do_this(y):
|
187 |
+
# p = ProfileSection("do_this", True) #only way to use the auto destruct method
|
188 |
+
# x = random.randint(0, (y+1)*2)
|
189 |
+
# print(f'do_this: {x} - {y}')
|
190 |
+
|
191 |
+
# for index in range(1000):
|
192 |
+
# do_this(index)
|
193 |
+
|
194 |
+
# log_profile_registry()
|
mew_log/table_utils.py
ADDED
@@ -0,0 +1,174 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from tabulate import tabulate
|
2 |
+
|
3 |
+
from ..mew_log.ansi_utils import ansi_color_str
|
4 |
+
from ..mew_log.attr_utils import get_class_attributes
|
5 |
+
|
6 |
+
|
7 |
+
def transpose_dict(d):
|
8 |
+
return {k: [dic[k] for dic in d.values()] for k in d.keys()}
|
9 |
+
|
10 |
+
|
11 |
+
def format_data_as_table(data, headers=None, tablefmt='grid', **kwargs):
|
12 |
+
if kwargs.get('use_color', False):
|
13 |
+
# Apply color to headers if applicable
|
14 |
+
if headers:
|
15 |
+
headers = [ansi_color_str(header, fg="yellow") for header in headers]
|
16 |
+
|
17 |
+
|
18 |
+
return tabulate(data, headers=headers, tablefmt=tablefmt)
|
19 |
+
|
20 |
+
def format_property(name, data, headers=None, tablefmt='simple_grid', use_color=False):
|
21 |
+
colored_name = ansi_color_str(f"===[{name}]<{type(data).__name__}>===", fg="yellow") if use_color else f"===[{name}]<{type(data).__name__}>==="
|
22 |
+
return f'{colored_name}\n{format_table(data, headers, tablefmt, use_color=use_color)}'
|
23 |
+
|
24 |
+
|
25 |
+
|
26 |
+
# note: pretty sure transpose doesn't work correctly
|
27 |
+
def format_table(data, headers=None, tablefmt='simple_grid', **kwargs):
|
28 |
+
if not data:
|
29 |
+
return ""
|
30 |
+
if isinstance(data, dict):
|
31 |
+
return format_dict_as_table(data, headers, tablefmt, **kwargs)
|
32 |
+
elif isinstance(data, (list, set)):
|
33 |
+
# transpose = list(zip(*data))
|
34 |
+
return format_list_as_table(data, headers, tablefmt, **kwargs)
|
35 |
+
elif hasattr(data, '__dict__'):
|
36 |
+
return format_obj_as_table(data, headers, tablefmt, **kwargs)
|
37 |
+
else:
|
38 |
+
use_color = kwargs.get('use_color', False)
|
39 |
+
if use_color:
|
40 |
+
return ansi_color_str(str(data), fg='bright_yellow')
|
41 |
+
else:
|
42 |
+
return str(data)
|
43 |
+
|
44 |
+
|
45 |
+
|
46 |
+
def format_list_as_table(data, headers=None, tablefmt='simple', **kwargs):
|
47 |
+
if not headers:
|
48 |
+
headers = ["Index", "Value"] if kwargs.get('show_indexes', False) else []
|
49 |
+
|
50 |
+
if kwargs.get('show_indexes', False):
|
51 |
+
out_dict = {i: item for i, item in enumerate(data)}
|
52 |
+
return format_table(out_dict, headers, tablefmt, **kwargs)
|
53 |
+
else:
|
54 |
+
formatted_str = ansi_color_str(f" | len={len(data)} | ", fg='bright_cyan') if kwargs.get('use_color', False) else f" | len={len(data)} | "
|
55 |
+
formatted_str += "["
|
56 |
+
for i, item in enumerate(data):
|
57 |
+
formatted_str += format_table(item, [], tablefmt, **kwargs)
|
58 |
+
if i < len(data) - 1: formatted_str += kwargs.get('delimeter', ', ')
|
59 |
+
formatted_str += "]"
|
60 |
+
return formatted_str
|
61 |
+
|
62 |
+
|
63 |
+
def format_dict_as_table(data, headers=None, tablefmt='simple_grid', **kwargs):
|
64 |
+
if kwargs.get('transpose', False):
|
65 |
+
if not headers:
|
66 |
+
headers =list(data.keys())
|
67 |
+
#Transpose the dictionary: each key-value pair becomes a column
|
68 |
+
transposed_data = [list(value) for key, value in zip(list(data.keys()), zip(*data.items()))]
|
69 |
+
#intended for values of the same lenth
|
70 |
+
#transposed_data = [headers] + [list(row) for row in zip(list(data.values()))]
|
71 |
+
return format_data_as_table(transposed_data, headers=headers, tablefmt=tablefmt)
|
72 |
+
else:
|
73 |
+
if not headers:
|
74 |
+
headers = ["Key", "Value"]
|
75 |
+
|
76 |
+
# Convert the dictionary into a list of lists
|
77 |
+
table_data = [[key, format_table(value, [], 'simple',**kwargs)] for key, value in data.items() if value is not None]
|
78 |
+
|
79 |
+
# Format the table
|
80 |
+
return format_data_as_table(table_data, headers=headers, tablefmt=tablefmt)
|
81 |
+
|
82 |
+
|
83 |
+
def format_obj_as_table(data, headers=None, tablefmt='fancy_grid', **kwargs):
|
84 |
+
verbose = kwargs.get('verbose', True)
|
85 |
+
fancy = kwargs.get('fancy', True)
|
86 |
+
use_color = kwargs.get('use_color', True)
|
87 |
+
|
88 |
+
attributes_dict = get_class_attributes(data, verbose=verbose)
|
89 |
+
class_name = data.__class__.__name__
|
90 |
+
variables = [*attributes_dict['variables'].values()]
|
91 |
+
methods = [*attributes_dict['methods'].values()]
|
92 |
+
# Check if headers are provided, if not, construct them
|
93 |
+
if not headers:
|
94 |
+
headers = [class_name]
|
95 |
+
if variables:
|
96 |
+
headers.append('variables')
|
97 |
+
if methods:
|
98 |
+
headers.append('methods')
|
99 |
+
|
100 |
+
# Initialize an empty list to store the formatted table data
|
101 |
+
table_data = []
|
102 |
+
|
103 |
+
# formatted_vars = []
|
104 |
+
# for v in variables:
|
105 |
+
# format_v = format_arg(v, use_color=use_color, fancy=fancy)
|
106 |
+
# formatted_vars.append(format_v)
|
107 |
+
|
108 |
+
# Add variables and methods data to the table data
|
109 |
+
table_data.append([format_table(variables, ['variables'], 'simple', **kwargs) if variables else None,
|
110 |
+
format_table(methods, ['methods'], 'simple', **kwargs) if methods else None])
|
111 |
+
|
112 |
+
table_data = list(zip(*table_data))
|
113 |
+
|
114 |
+
# Return the formatted table
|
115 |
+
return format_data_as_table(table_data, headers, tablefmt, **kwargs)
|
116 |
+
|
117 |
+
|
118 |
+
def print_table(msg, data, headers=None, tablefmt='fancy_grid'):
|
119 |
+
print(f'==={msg}==>\n{format_table(data, headers, tablefmt)}')
|
120 |
+
|
121 |
+
|
122 |
+
def format_square_layout(data):
|
123 |
+
if isinstance(data, (list, set)):
|
124 |
+
result_count = len(data)
|
125 |
+
rows, columns = calc_best_square(result_count)
|
126 |
+
table_data = [data[i:i + columns] for i in range(0, len(data), columns)]
|
127 |
+
return format_table(table_data)
|
128 |
+
elif isinstance(data, dict):
|
129 |
+
items = [(k, v) for k, v in data.items()]
|
130 |
+
result_count = len(items)
|
131 |
+
rows, columns = calc_best_square(result_count)
|
132 |
+
table_data = [items[i:i + columns] for i in range(0, len(items), columns)]
|
133 |
+
return format_table(table_data)
|
134 |
+
elif hasattr(data, '__dict__'):
|
135 |
+
items = [(k, v) for k, v in data.__dict__.items()]
|
136 |
+
result_count = len(items)
|
137 |
+
rows, columns = calc_best_square(result_count)
|
138 |
+
table_data = [items[i:i + columns] for i in range(0, len(items), columns)]
|
139 |
+
return format_table(table_data)
|
140 |
+
else:
|
141 |
+
return format_table(data)
|
142 |
+
|
143 |
+
|
144 |
+
def print_square_layout(msg, data):
|
145 |
+
print(f'==={msg}==>\n{format_square_layout(data)}')
|
146 |
+
|
147 |
+
|
148 |
+
def print_as_square(strings):
|
149 |
+
# Calculate the number of strings in the input list
|
150 |
+
result_count = len(strings)
|
151 |
+
|
152 |
+
# Calculate the best square layout dimensions based on the result count
|
153 |
+
rows, columns = calc_best_square(result_count)
|
154 |
+
|
155 |
+
# Create a grid with empty strings filled in for each cell
|
156 |
+
grid = [[' ' for _ in range(columns)] for _ in range(rows)]
|
157 |
+
|
158 |
+
# Iterate over the strings and populate the grid with them
|
159 |
+
for i, string in enumerate(strings):
|
160 |
+
# Calculate the row and column index for the current string
|
161 |
+
row = i // columns
|
162 |
+
col = i % columns
|
163 |
+
# Ensure the row and column indices are within the valid range of the grid dimensions
|
164 |
+
if row < rows and col < columns:
|
165 |
+
# Place the string in the corresponding cell of the grid
|
166 |
+
grid[row][col] = string
|
167 |
+
|
168 |
+
# Determine the maximum width of each column in the grid
|
169 |
+
max_widths = [max(len(cell) for cell in row) for row in grid]
|
170 |
+
|
171 |
+
# Print the grid, ensuring each cell is left-aligned and padded to its maximum width
|
172 |
+
for row in grid:
|
173 |
+
print(' '.join(cell.ljust(width) for cell, width in zip(row, max_widths)))
|
174 |
+
|