nodeocc 1.0.25__tar.gz → 1.0.27__tar.gz

{nodeocc-1.0.25 → nodeocc-1.0.27}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nodeocc
-Version: 1.0.25
+Version: 1.0.27
 Summary: TUI application for viewing the status of GPU allocations on a Slurm cluster
 Author: Matteo Boschini
 Maintainer: Lorenzo Bonicelli, Aniello Panariello
@@ -31,4 +31,8 @@ Contents:
 
 Run it with: `python controller/controller.py`
 
+Press `D` in the interactive view to open or close the debug screen. It shows
+the local role, master PID and port, snapshot status, and local package
+versions. Use the up and down keys to scroll.
+
 Requirements: pandas, pycurses.

{nodeocc-1.0.25 → nodeocc-1.0.27}/README.md

@@ -13,4 +13,8 @@ Contents:
 
 Run it with: `python controller/controller.py`
 
+Press `D` in the interactive view to open or close the debug screen. It shows
+the local role, master PID and port, snapshot status, and local package
+versions. Use the up and down keys to scroll.
+
 Requirements: pandas, pycurses.

{nodeocc-1.0.25 → nodeocc-1.0.27}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "nodeocc"
-version = "1.0.25"
+version = "1.0.27"
 authors = [{ name = "Matteo Boschini" }]
 maintainers = [{ name = "Lorenzo Bonicelli" }, { name = "Aniello Panariello" }]
 description = "TUI application for viewing the status of GPU allocations on a Slurm cluster"

{nodeocc-1.0.25 → nodeocc-1.0.27}/src/controller/controller.py

@@ -24,43 +24,38 @@ import json
 import argparse
 import traceback
 import socket
+import re
+from urllib.request import urlopen
 from model.infrastructure import Infrastructure
 from model.job import Job
 import asyncio
 import importlib.metadata
+from view.transport import send as send_datagram, receive as receive_datagram
 
 program_name = 'nodeocc'
 version_number = importlib.metadata.version(program_name)
-updated = False
 
-BBLUE = '\033[1;34m'
-END = '\033[0m'
-try:
-    url = "https://pypi.python.org/pypi/nodeocc/json"
-    import requests
-    newest_version = json.loads(requests.get(url).text)['info']['version']
-    print(BBLUE, end='')
-    if int(newest_version.split('.')[0]) > int(version_number.split('.')[0]):
-        print(f"New major version available: {newest_version} (current: {version_number})")
-        print(f"Please update with 'pip install {program_name} --upgrade'")
-        time.sleep(7)
-    elif int(newest_version.split('.')[1]) > int(version_number.split('.')[1]):
-        print(f"New minor version available: {newest_version} (current: {version_number})")
-        print(f"Please update with 'pip install {program_name} --upgrade'")
-        time.sleep(7)
-    elif int(newest_version.split('.')[2]) > int(version_number.split('.')[2]):
-        print(f"New patch version available: {newest_version} (current: {version_number})")
-        print(f"Please update with 'pip install {program_name} --upgrade'")
-        time.sleep(7)
-    else:
-        print(f"Up to date: {version_number}")
-        updated = True
-        time.sleep(3)
-    print(END, end='')
-except Exception as e:
-    print(f"{BBLUE}Could not get newest version from pip, check if {program_name} is up to date{END}")
-    newest_version = '?.?.?'
-    time.sleep(5)
+
+def _version_tuple(v):
+    return tuple(int(x) for x in re.findall(r'\d+', str(v)))
+
+
+def check_latest_version(cur_version, timeout_sec=1.2):
+    """
+    Fast, non-blocking-ish version check:
+    - short timeout
+    - no startup sleeps/pauses
+    - graceful fallback to current version on errors
+    """
+    try:
+        url = f"https://pypi.python.org/pypi/{program_name}/json"
+        with urlopen(url, timeout=timeout_sec) as response:
+            payload = json.loads(response.read().decode('utf-8'))
+        newest = payload['info']['version']
+        is_updated = _version_tuple(cur_version) >= _version_tuple(newest)
+        return newest, is_updated
+    except Exception:
+        return cur_version, True
 
 BEGIN_DELIM = "!{$"
 END_DELIM_ENCODED = "!}$".encode('utf-8')
@@ -81,6 +76,12 @@ args = parser.parse_args()
 # check basepath is readable
 assert os.access(args.basepath, os.R_OK), f"Base path {args.basepath} not readable"
 
+# Keep startup snappy for daemon/diagnostics; check updates only in interactive mode.
+if args.master_only or args.diag:
+    newest_version, updated = version_number, True
+else:
+    newest_version, updated = check_latest_version(version_number)
+
 # export args
 orig_instance = Singleton.getInstance(args)
 
@@ -121,7 +122,7 @@ def send_msg(instance: Singleton, msg, n_extra_msgs=0, is_extra=False):
     msg = (str(msg_len) + begin_delim).encode('utf-8') + msg + END_DELIM_ENCODED
     instance.timeme(f"- msg encoded with len {msg_len}")
 
-    instance.sock.sendto(msg, ('<broadcast>', instance.port))
+    send_datagram(instance.sock, instance.port, msg)
 
 
 def update_data_master(instance: Singleton):
@@ -139,24 +140,11 @@ def update_data_master(instance: Singleton):
     running_jobs = [j.to_nested_dict() for j in jobs if j.state == 'CG' or j.state == 'R']
     queue_jobs = [j.to_nested_dict() for j in jobs if j.state != 'R' and j.state != 'CG']
     queue_jobs = sorted(queue_jobs, key=lambda x: x['priority'], reverse=True)
-    maxlen = min(JOB_LIMIT_PER_MSG, len(running_jobs) + len(queue_jobs))
-
-    queue_jobs, extra_queue_jobs = queue_jobs[:maxlen - len(running_jobs)], queue_jobs[maxlen - len(running_jobs):]
     cur_timestamp = time.time()
     msg = json.dumps({'inf': inf.to_nested_dict(), 'jobs': running_jobs + queue_jobs, 'ts': str(cur_timestamp)})
-
-    N_EXTRA_MSGS = (len(extra_queue_jobs) // JOB_LIMIT_PER_MSG) + 1 if len(extra_queue_jobs) > 0 else 0
-    send_msg(instance, msg, n_extra_msgs=N_EXTRA_MSGS)
+    send_msg(instance, msg)
     instance.timeme("- main broadcast")
 
-    if extra_queue_jobs:
-        # get batches of JOB_LIMIT_PER_MSG jobs
-        for job_idx in range(N_EXTRA_MSGS):
-            i = job_idx * JOB_LIMIT_PER_MSG
-            msg = json.dumps({'inf': inf.to_nested_dict(), 'jobs': extra_queue_jobs[i:i + JOB_LIMIT_PER_MSG], 'ts': str(cur_timestamp)})
-            send_msg(instance, msg, is_extra=True, n_extra_msgs=N_EXTRA_MSGS - job_idx - 1)
-            instance.timeme(f"- extra broadcast #{job_idx+1}")
-
     return inf, jobs
 
 
@@ -178,7 +166,7 @@ async def get_data_slave(instance):
     try:
         # listen for data from master
         # instance.sock.settimeout(6.5)
-        data, _ = instance.sock.recvfrom(MAX_BUF)
+        data = receive_datagram(instance.sock, MAX_BUF)
         decoded_data = data.decode('utf-8', errors='replace')
         if BEGIN_DELIM not in decoded_data:
             if EXTRA_MSG_BEGIN_DELIM in decoded_data:
@@ -198,7 +186,7 @@ async def get_data_slave(instance):
         total_msg_n = n_msgs_remaining
         msg_store = {i: [] for i in range(total_msg_n)}
         while n_msgs_remaining > 0:
-            data, _ = instance.sock.recvfrom(MAX_BUF)
+            data = receive_datagram(instance.sock, MAX_BUF)
             decoded_extra = data.decode('utf-8', errors='replace')
             if EXTRA_MSG_BEGIN_DELIM not in decoded_extra:
                 instance.log("Received non-extra packet while waiting for extra packets, skipping")
@@ -239,8 +227,6 @@ async def get_data_slave(instance):
         if n_timeout > TIMEOUT_MAX:
             try_open_socket_as_slave(instance, force=True)
 
-        time.sleep(random.randint(100, 10000) / 1000)
-
     except Exception as e:
         instance.err(f"Exception while receiving data from master: {e}")
         instance.err(traceback.format_exc())
@@ -272,7 +258,6 @@ async def get_all():
         if orig_instance.is_master:
             inf, jobs = update_data_master(orig_instance)
 
-            time.sleep(5)
         else:
             # loop = asyncio.get_event_loop()
             orig_instance.timeme(f"- listening for data")
@@ -345,6 +330,7 @@ def _main():
         orig_instance.log(f"Starting master daemon")
 
         while True:
+            cycle_start = time.monotonic()
             orig_instance.timeme(f"Updating...")
             try:
                 update_data_master(orig_instance)
@@ -352,7 +338,7 @@ def _main():
                 orig_instance.err(f"Exception: {e}")
                 orig_instance.err(traceback.format_exc())
 
-            time.sleep(5)
+            time.sleep(max(0, 5 - (time.monotonic() - cycle_start)))
 
     else:
         # configure singleton

{nodeocc-1.0.25 → nodeocc-1.0.27}/src/nodeocc.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nodeocc
-Version: 1.0.25
+Version: 1.0.27
 Summary: TUI application for viewing the status of GPU allocations on a Slurm cluster
 Author: Matteo Boschini
 Maintainer: Lorenzo Bonicelli, Aniello Panariello
@@ -31,4 +31,8 @@ Contents:
 
 Run it with: `python controller/controller.py`
 
+Press `D` in the interactive view to open or close the debug screen. It shows
+the local role, master PID and port, snapshot status, and local package
+versions. Use the up and down keys to scroll.
+
 Requirements: pandas, pycurses.

{nodeocc-1.0.25 → nodeocc-1.0.27}/src/nodeocc.egg-info/SOURCES.txt

@@ -18,4 +18,5 @@ src/view/curses_multiwindow.py
 src/view/slurm_list.py
 src/view/slurm_viz.py
 src/view/styles.py
+src/view/transport.py
 src/view/utils.py

{nodeocc-1.0.25 → nodeocc-1.0.27}/src/readers/slurmreader.py

@@ -1,5 +1,6 @@
 import os
 import re
+from collections import defaultdict
 from io import StringIO
 
 import pandas as pd
@@ -57,9 +58,16 @@ def _parse_gpu_count(gres):
         token = token.strip()
         if not token.lower().startswith('gpu'):
             continue
-        last_field = token.split(':')[-1].split('(')[0]
-        match = re.search(r'\d+', last_field)
-        n_gpu += int(match.group(0)) if match else 1
+        token_core = token.split('(')[0]
+        parts = token_core.split(':')
+        count = None
+
+        if len(parts) >= 2 and re.fullmatch(r'\d+', parts[-1]):
+            count = int(parts[-1])
+        elif len(parts) >= 3 and re.fullmatch(r'\d+', parts[-2]):
+            count = int(parts[-2])
+
+        n_gpu += count if count is not None else 1
     return n_gpu
 
 
@@ -75,10 +83,13 @@ def _parse_gpu_models(gres):
         token = token.strip()
         if not token.lower().startswith('gpu'):
             continue
-        parts = token.split(':')
-        if len(parts) >= 3:
+        token_core = token.split('(')[0]
+        parts = token_core.split(':')
+        if len(parts) >= 3 and not re.fullmatch(r'\d+', parts[1]):
+            model = parts[1]
+        elif len(parts) == 2 and not re.fullmatch(r'\d+', parts[1]):
             model = parts[1]
-        elif len(parts) == 2:
+        elif len(parts) == 2 and re.fullmatch(r'\d+', parts[1]):
             model = 'gpu'
         else:
             model = 'n/a'
@@ -367,8 +378,11 @@ def read_jobs():
         line['NICE']
     ), axis=1).tolist()
 
+    joblets_by_job = defaultdict(list)
+    for joblet in joblets:
+        joblets_by_job[joblet.jobid].append(joblet)
     for j in jobs:
-        j.joblets = [x for x in joblets if x.jobid == j.jobid]
+        j.joblets = joblets_by_job[j.jobid]
     return jobs, joblets
 
 

{nodeocc-1.0.25 → nodeocc-1.0.27}/src/view/__init__.py

@@ -13,5 +13,11 @@ def update_views(stdscr, instance, filter):
             instance, 'job_id_type') else 'agg')
     instance.timeme(f"- print job list")
 
-    instance.nocc = slurm_viz.view_viz(instance.inf, instance.jobs, stylefn=styles.crsstyler, mode=instance.view_mode if hasattr(instance, 'view_mode') else 'gpu')
+    instance.nocc = slurm_viz.view_viz(
+        instance.inf,
+        instance.jobs,
+        stylefn=styles.crsstyler,
+        mode=instance.view_mode if hasattr(instance, 'view_mode') else 'gpu',
+        panel_width=instance.right_width if hasattr(instance, 'right_width') else None,
+    )
     instance.timeme(f"- view nodeocc")

{nodeocc-1.0.25 → nodeocc-1.0.27}/src/view/curses_multiwindow.py

@@ -14,6 +14,9 @@ from logging.handlers import RotatingFileHandler
 from pathlib import Path
 from view import update_views
 import stat
+import sys
+from datetime import datetime
+from importlib import metadata
 
 a_filter_values = [None, 'me', 'prod', 'stud', 'cvcs']
 
@@ -33,8 +36,14 @@ BUTTON_ACTIONS = {
     'show_prio': [ord('p')],
     'refresh': [ord('y')],
     'info': [ord('i')],
+    'debug_view': [ord('D')],
 }
 
+RIGHT_WIDTH_GPU = 33
+RIGHT_WIDTH_INFO = 39
+# Keep a single stable width across modes; use the widest one so info fields are not cut.
+RIGHT_WIDTH_UNIFIED = max(RIGHT_WIDTH_GPU, RIGHT_WIDTH_INFO)
+
 
 def is_file_writable_byall(file_path):
     return os.stat(file_path).st_mode & stat.S_IWOTH
@@ -48,7 +57,10 @@ def is_nodeocc_process(pid):
     try:
         cmdline = ' '.join(psutil.Process(int(pid)).cmdline())
         return ('nodeocc' in cmdline) or ('controller.py' in cmdline)
-    except Exception:
+    except psutil.AccessDenied:
+        # Other users' command lines may be hidden on shared login nodes.
+        return psutil.pid_exists(int(pid))
+    except (psutil.Error, ValueError):
         return False
 
 
@@ -67,7 +79,10 @@ def process_has_udp_port(pid, port):
         for conn in proc.connections(kind='udp'):
             if conn.laddr and conn.laddr.port == int(port):
                 return True
-    except Exception:
+    except psutil.AccessDenied:
+        # Socket ownership is commonly visible only to the process owner.
+        return psutil.pid_exists(int(pid))
+    except (psutil.Error, ValueError):
         return False
     return False
 
@@ -169,6 +184,11 @@ class Singleton:
         self.rens = ""
 
         self.fetch_fn = None
+        self.debug_view = False
+        self.debug_offset = 0
+        self.last_success = None
+        self.last_fetch_duration = None
+        self.last_error = None
         self.view_mode = 'gpu'
         self.job_id_type = 'agg'
 
@@ -319,6 +339,7 @@ class Singleton:
             self._ctime = _ntime
 
     def err(self, msg):
+        self.last_error = str(msg)
         if self.args.debug or self.args.master_only:
             logging.error(msg)
 
@@ -332,11 +353,14 @@ class Singleton:
         if self.fetch_fn is not None:
             inf, jobs, avg_wait_time = await self.fetch_fn()  # a_filter_values[self.a_filter])
             master_dead = inf is None
+            if not master_dead:
+                self.last_success = time.time()
             self.inf = inf if inf is not None else self.inf
             self.jobs = jobs if jobs is not None else self.jobs
             self.avg_wait_time = avg_wait_time if avg_wait_time is not None else self.avg_wait_time
 
         _delta_t = time.time() - _ctime
+        self.last_fetch_duration = _delta_t
         self.log(f"Fetch took {_delta_t:.2f} seconds")
         return master_dead
 
@@ -454,6 +478,18 @@ def handle_keys(stdscr, instance):
 
     instance.mouse_state = {}
 
+    if k in BUTTON_ACTIONS['debug_view']:
+        instance.debug_view = not instance.debug_view
+        instance.debug_offset = 0
+        stdscr.clear()
+        return
+    if instance.debug_view:
+        if k in BUTTON_ACTIONS['down_1']:
+            instance.debug_offset += 1
+        elif k in BUTTON_ACTIONS['up_1']:
+            instance.debug_offset = max(0, instance.debug_offset - 1)
+        return
+
     # process input
     # RIGHT
     if k in BUTTON_ACTIONS['right_1']:
@@ -476,7 +512,7 @@ def handle_keys(stdscr, instance):
         instance.sort_by_prio = not instance.sort_by_prio
     elif k in BUTTON_ACTIONS['res_view_mode']:
         instance.view_mode = {"gpu": "ram", "ram": "cpu", "cpu": "gpu", "info": "gpu"}[instance.view_mode]
-        instance.right_width = 33 if instance.view_mode != 'info' else 39
+        instance.right_width = RIGHT_WIDTH_UNIFIED
         stdscr.clear()
     elif k in BUTTON_ACTIONS['job_id_type']:
         instance.job_id_type = "true" if instance.job_id_type == "agg" else "agg"
@@ -489,7 +525,7 @@ def handle_keys(stdscr, instance):
         instance.show_prio = not instance.show_prio
     elif k in BUTTON_ACTIONS['info']:
         instance.view_mode = 'info' if instance.view_mode != 'info' else 'gpu'
-        instance.right_width = 33 if instance.view_mode != 'info' else 39
+        instance.right_width = RIGHT_WIDTH_UNIFIED
         stdscr.clear()
     elif k in BUTTON_ACTIONS['up_tab']:
         # get screen size
@@ -502,7 +538,70 @@ def handle_keys(stdscr, instance):
         instance.voff -= ((height - 4) // 2) + 1
 
 
+def debug_lines(instance):
+    """Collect local diagnostics without querying Slurm or the network."""
+    if not hasattr(instance, '_package_versions'):
+        instance._package_versions = {}
+        for name in ('nodeocc', 'pandas', 'numpy', 'psutil'):
+            try:
+                instance._package_versions[name] = metadata.version(name)
+            except metadata.PackageNotFoundError:
+                instance._package_versions[name] = 'not installed'
+    port = getattr(instance, 'port', None)
+    master_pid = str(os.getpid()) if instance.is_master else 'unknown'
+    try:
+        for path in instance.get_port_files():
+            if port_from_portfile(path) == port:
+                master_pid = Path(path).read_text().strip() or 'unknown'
+                break
+    except OSError:
+        pass
+    try:
+        master_owner = psutil.Process(int(master_pid)).username()
+    except (ValueError, psutil.Error, OSError, KeyError):
+        master_owner = 'unknown'
+    last_success = (datetime.fromtimestamp(instance.last_success).strftime('%Y-%m-%d %H:%M:%S')
+                    if instance.last_success else 'never')
+    duration = (f'{instance.last_fetch_duration:.2f} s'
+                if instance.last_fetch_duration is not None else 'not available')
+    lines = [
+        'Connection',
+        f'  Host: {socket.getfqdn()}',
+        f'  Role: {"master" if instance.is_master else "client"}',
+        f'  Master PID: {master_pid}',
+        f'  Master owner: {master_owner}',
+        f'  Master UDP port: {port if port is not None else "unknown"}',
+        f'  Base path: {instance.basepath}',
+        '', 'Snapshot',
+        f'  Last successful update: {last_success}',
+        f'  Last {"update" if instance.is_master else "receive wait"}: {duration}',
+        f'  Jobs: {len(instance.jobs) if instance.jobs is not None else "unknown"}',
+        f'  Nodes: {len(instance.inf.nodes) if instance.inf is not None else "unknown"}',
+        f'  Last error: {instance.last_error or "none recorded"}',
+        '', 'Environment',
+        f'  Python: {sys.version.split()[0]}',
+        f'  Executable: {sys.executable}',
+        f'  Process PID: {os.getpid()}',
+        '  Package versions (local environment):',
+    ]
+    lines.extend(f'    {name}: {version}' for name, version in instance._package_versions.items())
+    return lines
+
+
 def update_screen(stdscr, instance):
+    if instance.debug_view:
+        lines, columns = stdscr.getmaxyx()
+        content = debug_lines(instance)
+        instance.debug_offset = min(instance.debug_offset, max(0, len(content) - max(1, lines - 2)))
+        stdscr.erase()
+        try:
+            stdscr.addnstr(0, 0, 'DEBUG  [D: close]  [up/down: scroll]  [Q: quit]', max(0, columns - 1))
+            for row, line in enumerate(content[instance.debug_offset:instance.debug_offset + max(0, lines - 2)], 1):
+                stdscr.addnstr(row, 0, line, max(0, columns - 1))
+        except curses.error:
+            pass
+        stdscr.refresh()
+        return
     update_views(stdscr, instance, a_filter_values[instance.a_filter])
     xoffset = 0
     instance.xoffset = xoffset
@@ -513,7 +612,7 @@ def update_screen(stdscr, instance):
     if instance.k in BUTTON_ACTIONS['refresh']:
         stdscr.clear()
 
-    totsize = 106
+    totsize = 106 + max(0, RIGHT_WIDTH_UNIFIED - RIGHT_WIDTH_GPU)
     if instance.show_account:
         totsize += 10
     if instance.show_prio:
@@ -579,7 +678,9 @@ def update_screen(stdscr, instance):
     stdscr.addstr(0, left_width + 5, 'INFO', curses.color_pair(2) | (curses.A_REVERSE if instance.view_mode == 'info' else 0))
     stdscr.addstr(0, left_width + 9, ']', curses.color_pair(2))
     instance.add_button(0, left_width + 2, '[I:INFO]', BUTTON_ACTIONS['info'][0])
-    stdscr.addstr(0, left_width + 10, '─' * (columns - 15 - left_width - 3), curses.color_pair(2))
+    stdscr.addstr(0, left_width + 11, '[D:DEBUG]', curses.color_pair(2))
+    instance.add_button(0, left_width + 11, '[D:DEBUG]', BUTTON_ACTIONS['debug_view'][0])
+    stdscr.addstr(0, left_width + 20, '─' * max(0, columns - 15 - left_width - 20), curses.color_pair(2))
     stdscr.addstr(0, columns - 15, '[G:', curses.color_pair(2))
     stdscr.addstr(0, columns - 12, 'GPU', curses.color_pair(2) | (curses.A_REVERSE if instance.view_mode == 'gpu' else 0))
     stdscr.addstr(0, columns - 12 + 3, 'RAM', curses.color_pair(2) | (curses.A_REVERSE if instance.view_mode == 'ram' else 0))
@@ -606,8 +707,14 @@ def update_screen(stdscr, instance):
     # get slurm user partition
     # stdscr.addstr(lines - 1, xoffset + 62 + 2, f'(Avg time {instance.avg_wait_time})', curses.color_pair(2))
 
-    signature = instance.signature if instance.updated else f'{instance.version} -> {instance.newest_version}'
-    stdscr.addstr(lines - 1, columns - 2 - len(signature), signature, curses.color_pair(2) if instance.updated else curses.color_pair(3))
+    if instance.updated:
+        signature = instance.signature
+        sig_style = curses.color_pair(2)
+    else:
+        signature = f'{instance.signature} (new {instance.newest_version})'
+        sig_style = curses.color_pair(4)
+
+    stdscr.addstr(lines - 1, columns - 2 - len(signature), signature, sig_style)
 
     stdscr.refresh()
     curses.doupdate()
@@ -645,6 +752,8 @@ async def update_screen_info(stdscr, instance: Singleton):
                 try_open_socket_as_slave(instance)
 
         update_screen(stdscr, instance)
+        if instance.is_master:
+            await asyncio.sleep(max(0, 5 - (instance.last_fetch_duration or 0)))
 
 
 async def wait_first(futures, instance):
@@ -712,7 +821,7 @@ async def main(stdscr):
     # status
 
     instance.a_filter = 0
-    instance.right_width = 33
+    instance.right_width = RIGHT_WIDTH_UNIFIED
 
     stdscr.clear()
 

{nodeocc-1.0.25 → nodeocc-1.0.27}/src/view/slurm_list.py

@@ -44,7 +44,7 @@ def _joblet_format(instance, job, width=74, jobid_type='agg'):
             mem = round(joblet.mem / 1024)
         joblet_repr += _format_to(f'{mem}G', 4, 'right')
         joblet_repr += ' '
-        if instance.view_mode in ('gpu', 'ram'):
+        if instance.view_mode in ('gpu', 'ram', 'info'):
             joblet_repr += _format_to((f'{joblet.n_gpus}gp' if joblet.n_gpus > 0 else " - "), 3, 'left')
         elif instance.view_mode == 'cpu':
             joblet_repr += _format_to((f'{joblet.cpus}cp' if joblet.cpus > 0 else " - "), 3, 'left')

{nodeocc-1.0.25 → nodeocc-1.0.27}/src/view/slurm_viz.py

@@ -101,6 +101,20 @@ cpu_paused = '▚'
 cpu_down = '⨯'
 
 GLOBAL_NODES = None
+BASE_PANEL_WIDTH = 33
+
+
+def _center_panel_content(text, panel_width, base_width=BASE_PANEL_WIDTH):
+    if panel_width is None or panel_width <= base_width:
+        return text
+
+    pad = max(0, (panel_width - base_width) // 2)
+    if pad == 0:
+        return text
+
+    lines = text.split('\n')
+    lines = [(' ' * pad + line) if line else line for line in lines]
+    return '\n'.join(lines)
 
 
 def get_ram_block(megs):
@@ -138,15 +152,15 @@ def get_node_status(node, jobs):
         return None  # white?
 
 
-def view_viz(infrastructure, jobs, work=True, stylefn=cmdstyle, current_user=None, mode='gpu'):
+def view_viz(infrastructure, jobs, work=True, stylefn=cmdstyle, current_user=None, mode='gpu', panel_width=None):
     if mode == 'gpu':
-        return view_viz_gpu(infrastructure, jobs, work, stylefn, current_user)
+        return _center_panel_content(view_viz_gpu(infrastructure, jobs, work, stylefn, current_user), panel_width)
     elif mode == 'ram':
-        return view_viz_ram(infrastructure, jobs, work, stylefn, current_user)
+        return _center_panel_content(view_viz_ram(infrastructure, jobs, work, stylefn, current_user), panel_width)
     elif mode == 'cpu':
-        return view_viz_cpu(infrastructure, jobs, work, stylefn, current_user)
+        return _center_panel_content(view_viz_cpu(infrastructure, jobs, work, stylefn, current_user), panel_width)
     elif mode == 'info':
-        return view_viz_info(infrastructure, jobs, work, stylefn, current_user)
+        return view_viz_info(infrastructure, jobs, work, stylefn, current_user, panel_width)
 
 
 def view_viz_ram(infrastructure, jobs, work=True, stylefn=cmdstyle, current_user=None):
@@ -519,7 +533,7 @@ def view_viz_cpu(infrastructure, jobs, work=True, stylefn=cmdstyle, current_user
     return RetScope.return_string
 
 
-def view_viz_info(infrastructure, jobs, work=True, stylefn=cmdstyle, current_user=None):
+def view_viz_info(infrastructure, jobs, work=True, stylefn=cmdstyle, current_user=None, panel_width=None):
     # this is for hot reload
     if not work:
         return "UPDATE IN PROGRESS - PLZ W8 M8 B8"
@@ -555,6 +569,7 @@ def view_viz_info(infrastructure, jobs, work=True, stylefn=cmdstyle, current_use
         # max_ram_len = max([len(str(n.mem // 1024)) for n in nodes])
         feat_str = stylefn("GREEN", "FEATURES")
         RetScope.return_string += f'{"":{gpu_name_chars}} GPU {feat_str}\n'
+
         for i, n in GLOBAL_NODES.iterrows():
             gpu_model = n.m_gpu  # if n.m_gpu is not None else 'N/A'
             # gpu_mem = n.gpu_mem if n.gpu_mem is not None else 'N/A'

nodeocc-1.0.27/src/view/transport.py

@@ -0,0 +1,46 @@
+"""Fragment large broadcast messages to fit in UDP datagrams."""
+import uuid
+import zlib
+
+PREFIX = b'nodeocc-fragment:'
+CHUNK_SIZE = 60000
+COMPRESSED = b'nodeocc-zlib:'
+
+
+def send(sock, port, data):
+    compressed = COMPRESSED + zlib.compress(data, level=1)
+    if len(compressed) < len(data):
+        data = compressed
+    if len(data) <= CHUNK_SIZE:
+        sock.sendto(data, ('<broadcast>', port))
+        return
+    message_id = uuid.uuid4().hex.encode('ascii')
+    count = (len(data) + CHUNK_SIZE - 1) // CHUNK_SIZE
+    for index in range(count):
+        header = PREFIX + message_id + b':' + str(index).encode() + b':' + str(count).encode() + b':'
+        sock.sendto(header + data[index * CHUNK_SIZE:(index + 1) * CHUNK_SIZE], ('<broadcast>', port))
+
+
+def receive(sock, max_buf):
+    fragments = {}
+    while True:
+        data, _ = sock.recvfrom(max_buf)
+        if not data.startswith(PREFIX):
+            return zlib.decompress(data[len(COMPRESSED):]) if data.startswith(COMPRESSED) else data
+        try:
+            message_id, index, count, chunk = data[len(PREFIX):].split(b':', 3)
+            index, count = int(index), int(count)
+            if count < 2 or count > 10000 or not 0 <= index < count:
+                continue
+        except ValueError:
+            continue
+        # Retain only one message, so lost fragments cannot accumulate forever.
+        if message_id not in fragments:
+            fragments = {message_id: (count, {})}
+        expected, parts = fragments[message_id]
+        if count != expected:
+            continue
+        parts[index] = chunk
+        if len(parts) == count:
+            data = b''.join(parts[i] for i in range(count))
+            return zlib.decompress(data[len(COMPRESSED):]) if data.startswith(COMPRESSED) else data