profiler-runit 0.1.3.post1.dev1__tar.gz → 0.1.4__tar.gz

{profiler_runit-0.1.3.post1.dev1 → profiler_runit-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: profiler-runit
-Version: 0.1.3.post1.dev1
+Version: 0.1.4
 Summary: Execute and profile cli commands
 Author-email: NotesOfCliff <notesofcliff@gmail.com>
 License-Expression: GPL-3.0-only

{profiler_runit-0.1.3.post1.dev1 → profiler_runit-0.1.4}/profiler_runit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: profiler-runit
-Version: 0.1.3.post1.dev1
+Version: 0.1.4
 Summary: Execute and profile cli commands
 Author-email: NotesOfCliff <notesofcliff@gmail.com>
 License-Expression: GPL-3.0-only

{profiler_runit-0.1.3.post1.dev1 → profiler_runit-0.1.4}/runit/cli.py

@@ -28,7 +28,7 @@ def run_cli():
     stats = monitor_process(args.command)
     report = format_report(stats)
     if args.plot:
-        plot_charts(stats)
+        plot_charts(stats, args.plot_width, args.plot_height)
     print(report)
     if args.out_file:
         sys.stdout = orig_stdout
@@ -69,6 +69,18 @@ def create_parser():
         action='store_true',
         help='Show resource usage charts (requires plotext).'
     )
+    parser.add_argument(
+        '--plot-width',
+        type=int,
+        default=80,
+        help='Set the plot width (default: 80)'
+    )
+    parser.add_argument(
+        '--plot-height',
+        type=int,
+        default=20,
+        help='Set the plot height (default: 20)'
+    )
     parser.add_argument(
         'command',
         nargs=argparse.REMAINDER,

{profiler_runit-0.1.3.post1.dev1 → profiler_runit-0.1.4}/runit/plotting.py

@@ -6,7 +6,7 @@ try:
 except ImportError:
     plt = None
 
-def plot_charts(stats):
+def plot_charts(stats, width=80, height=20):
     if plt is None:
         log.warning("plotext is not installed. Skipping plotting.")
         return
@@ -19,6 +19,7 @@ def plot_charts(stats):
             from .utils import extract_memory_rss
             rss = extract_memory_rss(stats['memory_info'])
             plt.clear_figure()
+            plt.plot_size(width=width, height=height)
             plt.subplots(2, 1)
             plt.subplot(1)
             plt.title("CPU % over time")