Merge pull request #201 from minrk/sticky-connection-file

set sticky bit on connection files

Author: takluyver

jupyter_client/connect.py

@@ -14,6 +14,7 @@
 import json
 import os
 import socket
+import stat
 import tempfile
 import warnings
 from getpass import getpass
@@ -135,6 +136,27 @@ def write_connection_file(fname=None, shell_port=0, iopub_port=0, stdin_port=0,
     with open(fname, 'w') as f:
         f.write(json.dumps(cfg, indent=2))
 
+    if hasattr(stat, 'S_ISVTX'):
+        # set the sticky bit on the file and its parent directory
+        # to avoid periodic cleanup
+        paths = [fname]
+        runtime_dir = os.path.dirname(fname)
+        if runtime_dir:
+            paths.append(runtime_dir)
+        for path in paths:
+            permissions = os.stat(path).st_mode
+            new_permissions = permissions | stat.S_ISVTX
+            if new_permissions != permissions:
+                try:
+                    os.chmod(path, permissions)
+                except OSError as e:
+                    # failed to set sticky bit,
+                    # probably not a big deal
+                    warnings.warn(
+                        "Failed to set sticky bit on %r: %s" % (path, e),
+                        RuntimeWarning,
+                    )
+
     return fname, cfg