nodeocc 1.0.11__tar.gz → 1.0.12__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nodeocc
-Version: 1.0.11
+Version: 1.0.12
 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.12}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "nodeocc"
-version = "1.0.11"
+version = "1.0.12"
 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.12}/src/nodeocc.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nodeocc
-Version: 1.0.11
+Version: 1.0.12
 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.12}/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,6 +36,14 @@ 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():
         raise Exception("No master running")
@@ -89,7 +98,7 @@ class Singleton:
 
     def clean_port_files(self):
         for f in os.listdir(self.basepath):
-            if f.endswith('.port'):
+            if f.endswith('.port') and is_file_writable_byall(os.path.join(self.basepath, f)):
                 os.system(f"rm {os.path.join(self.basepath, f)}")
 
     def __init__(self, args=None):
@@ -158,8 +167,11 @@ class Singleton:
 
     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)
+            portfiles = [f for f in os.listdir(self.basepath) if f.endswith('.port')]
+            portfiles = [(f, os.path.join(self.basepath, f)) for f in portfiles]
+            portfiles = [(fname, fpath) for fname, fpath in portfiles if is_file_readable_byall(fpath) and is_file_writable_byall(fpath)][0]
+
+            return portfiles[0], portfiles[1]
         return None
 
     def check_existing_master_running(self):
@@ -171,7 +183,15 @@ class Singleton:
         return False
 
     def port_file_exists(self):
-        return len([f for f in os.listdir(self.basepath) if f.endswith('.port')]) > 0
+        """
+        Check if a port file exists in the basepath and the file has 666 permissions
+        """
+        portfiles = [f for f in os.listdir(self.basepath) if f.endswith('.port')]
+        if len(portfiles) == 0:
+            return False
+        portfiles = [os.path.join(self.basepath, f) for f in portfiles]
+        portfiles = [f for f in portfiles if is_file_readable_byall(f) and is_file_writable_byall(f)]
+        return len(portfiles) > 0
 
     def create_socket_as_master(self):
         # create udp socket for broadcasting
@@ -191,8 +211,10 @@ 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))
+        # create file to store port with 666 permissions to file
+        filepath = Path(self.basepath, f'{self.port}.port')
+        filepath.write_text(str(self.pid))
+        os.chmod(filepath, 0o666)
 
         return self.port