nodeocc 1.0.11__tar.gz → 1.0.13__tar.gz

{nodeocc-1.0.11 → nodeocc-1.0.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nodeocc
-Version: 1.0.11
+Version: 1.0.13
 Summary: TUI application for viewing the status of GPU allocations on a Slurm cluster
 Author: Matteo Boschini
 Maintainer: Lorenzo Bonicelli, Aniello Panariello

{nodeocc-1.0.11 → nodeocc-1.0.13}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "nodeocc"
-version = "1.0.11"
+version = "1.0.13"
 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.11 → nodeocc-1.0.13}/src/controller/controller.py

@@ -70,7 +70,6 @@ JOB_LIMIT_PER_MSG = 60
 parser = argparse.ArgumentParser(description='Visualize slurm jobs')
 parser.add_argument('--debug', action='store_true', help='Enable logging')
 parser.add_argument('--master_only', action='store_true', help='Disable all prints - only run in background')
-parser.add_argument('--force_override', action='store_true', help='Force override of port file, kill previous instance')
 parser.add_argument('--basepath', type=str, default='/nas/softechict-nas-2/mboschini/cool_scripts/new_nodeocc/', help='Base path for nodeocc. Must be readable by all users')
 args = parser.parse_args()
 
@@ -123,7 +122,6 @@ def send_msg(instance: Singleton, msg, n_extra_msgs=0, is_extra=False):
 def update_data_master(instance: Singleton):
     if instance.check_existing_master_running():
         instance.log(f"Master already running, killing current instance")
-        instance.cleanup()
         exit(0)
 
     inf = readers.slurmreader.read_infrastructure()
@@ -271,16 +269,6 @@ def _main():
         assert orig_instance.is_master, "Instance is not `master`, run with `force_override` to override"
 
         orig_instance.log(f"Starting master daemon")
-        # register atexit
-        import atexit
-
-        def exit_handler():
-            orig_instance.log(f"Exiting...")
-            orig_instance.sock.close()
-            # remove .port file
-            if orig_instance.port_file_exists():
-                Path(orig_instance.get_port_file_name()[1]).unlink()
-        atexit.register(exit_handler)
 
         while True:
             orig_instance.timeme(f"Updating...")
@@ -306,5 +294,4 @@ def _main():
 
 _main()
 
-orig_instance.cleanup()
 exit(0)

{nodeocc-1.0.11 → nodeocc-1.0.13}/src/nodeocc.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nodeocc
-Version: 1.0.11
+Version: 1.0.13
 Summary: TUI application for viewing the status of GPU allocations on a Slurm cluster
 Author: Matteo Boschini
 Maintainer: Lorenzo Bonicelli, Aniello Panariello

{nodeocc-1.0.11 → nodeocc-1.0.13}/src/view/curses_multiwindow.py

@@ -13,6 +13,7 @@ import traceback
 from logging.handlers import RotatingFileHandler
 from pathlib import Path
 from view import update_views
+import stat
 
 a_filter_values = [None, 'me', 'prod', 'stud', 'cvcs']
 
@@ -35,11 +36,21 @@ BUTTON_ACTIONS = {
 }
 
 
+def is_file_writable_byall(file_path):
+    return os.stat(file_path).st_mode & stat.S_IWOTH
+
+
+def is_file_readable_byall(file_path):
+    return os.stat(file_path).st_mode & stat.S_IROTH
+
+
 def try_open_socket_as_slave(instance):
-    if not instance.port_file_exists():
+    if len(instance.get_port_files()) == 0:
         raise Exception("No master running")
 
-    cur_port = int(instance.get_port_file_name()[0].split('.')[0])
+    cur_port = [int(f.split('.')[0].split('master_')[1]) for f in instance.get_port_files()
+                if (pid := Path(f).read_text()) and psutil.pid_exists(int(pid))][0]
+
     if hasattr(instance, 'port'):
         # check if port file has same name
         if instance.port != cur_port:
@@ -67,14 +78,6 @@ class Singleton:
             Singleton.__instance.args = args
         return Singleton.__instance
 
-    # destructor
-    def cleanup(self):
-        if self.is_master:
-            if self.port_file_exists():
-                os.system(f"rm {self.get_port_file_name()[1]}")
-        if hasattr(self, 'sock') and self.sock is not None:
-            self.sock.close()
-
     def setup_logging(self):
         if self.args.master_only:
             handler = RotatingFileHandler(os.path.join(self.basepath, '.master_log.txt'), maxBytes=5 * 1024 * 1024, backupCount=2, mode='w')
@@ -88,9 +91,18 @@ class Singleton:
                                 handlers=[handler])
 
     def clean_port_files(self):
+        nodename = socket.getfqdn().split('.')[0]
         for f in os.listdir(self.basepath):
-            if f.endswith('.port'):
+            if nodename in f and 'master_' in f and f.endswith('.portfile') and is_file_writable_byall(os.path.join(self.basepath, f)):
+                # read pid from file
+                pid = Path(os.path.join(self.basepath, f)).read_text()
+                # assert no process is running
+                if psutil.pid_exists(int(pid)):
+                    self.log(f"Process {pid} is still running, killing self")
+                    return False
+
                 os.system(f"rm {os.path.join(self.basepath, f)}")
+        return True
 
     def __init__(self, args=None):
         """ Virtually private constructor. """
@@ -131,49 +143,36 @@ class Singleton:
 
         self.basepath = self.args.basepath
 
-        # check if any .port file exists
-        if args.force_override and self.port_file_exists():
-            _, port_filepath = self.get_port_file_name()
-
-            # read pid from file
-            pid = Path(port_filepath).read_text()
-            # kill process
-            try:
-                os.kill(int(pid), signal.SIGTERM)
-            except Exception as e:
-                self.log("Could not kill process, deleting port file")
-                self.log(e)
-
-            # delete port file
-            Path(port_filepath).unlink()
-
         self.is_master = not self.check_existing_master_running()
 
         if self.is_master:
-            # clean up old port files
-            self.clean_port_files()
             self.port = self.create_socket_as_master()
         else:
             try_open_socket_as_slave(self)
 
-    def get_port_file_name(self):
-        if self.port_file_exists():
-            fname = [f for f in os.listdir(self.basepath) if f.endswith('.port')][0]
-            return fname, os.path.join(self.basepath, fname)
-        return None
-
     def check_existing_master_running(self):
-        if self.port_file_exists():
-            # check if process is still running
-            pid = Path(self.get_port_file_name()[1]).read_text()
-            if pid and int(pid) != os.getpid():
-                return psutil.pid_exists(int(pid))
+        portfiles = self.get_port_files()
+        if len(portfiles) > 0:
+            for f in portfiles:
+                # check if process is still running
+                pid = Path(f).read_text()
+                if pid and int(pid) != os.getpid():
+                    return psutil.pid_exists(int(pid))
         return False
 
-    def port_file_exists(self):
-        return len([f for f in os.listdir(self.basepath) if f.endswith('.port')]) > 0
+    def get_port_files(self):
+        """
+        Check if a port file exists in the basepath and the file has 666 permissions
+        """
+        nodename = socket.getfqdn().split('.')[0]
+        portfiles = [os.path.join(self.basepath, f) for f in os.listdir(self.basepath)
+                     if f.endswith('.portfile') and 'master_' in f and nodename in f]
+        portfiles = [f for f in portfiles if is_file_readable_byall(f) and is_file_writable_byall(f)]
+        return portfiles
 
     def create_socket_as_master(self):
+        nodename = socket.getfqdn().split('.')[0]
+
         # create udp socket for broadcasting
         self.sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
 
@@ -191,8 +190,26 @@ class Singleton:
         # get pid of current process
         self.pid = os.getpid()
 
-        # create file to store port
-        Path(self.basepath, f'{self.port}.port').write_text(str(self.pid))
+        # clean up old port files
+        kill_ok = self.clean_port_files()
+        if not kill_ok:
+            return None
+
+        try:
+            # create file to store port with 666 permissions to file
+            filepath = Path(self.basepath, f'{nodename}_master_{self.port}.portfile')
+            filepath.write_text(str(self.pid))
+            os.chmod(filepath, 0o666)
+
+            self.port_filepath = filepath
+        except Exception as e:
+            self.err(f"Could not create port file: {e}")
+            self.err(traceback.format_exc())
+            # check if file exists
+            if filepath.exists():
+                # delete file
+                os.system(f"rm {filepath}")
+            return None
 
         return self.port
 
@@ -548,9 +565,12 @@ async def update_screen_info(stdscr, instance: Singleton):
                 if not instance.check_existing_master_running():
                     instance.log("MASTER STILL DEAD, TRYING TO BECOME MASTER")
                     instance.setup_logging()
-                    instance.clean_port_files()
                     instance.port = instance.create_socket_as_master()
                     instance.is_master = True
+                else:
+                    try_open_socket_as_slave(instance)
+            else:
+                try_open_socket_as_slave(instance)
 
         update_screen(stdscr, instance)
 
@@ -626,7 +646,13 @@ async def main(stdscr):
 
     def exit_handler(sig, frame):
         instance.log(f"FORCED EXIT...")
-        instance.cleanup()
+        # instance.cleanup()
+        if hasattr(instance, 'port_filepath'):
+            os.system(f"rm {instance.port_filepath}")
+
+        if hasattr(instance, 'sock'):
+            instance.sock.close()
+
         exit(0)
     signal.signal(signal.SIGINT, exit_handler)