+"""A few things that didn't seem to fit anywhere else."""
+
import cPickle
import errno
import os
import logger
-PID_FILE = '/var/run/node_mgr.pid'
+PID_FILE = '/var/run/nm.pid'
def as_daemon_thread(run):
"""Call function <run> with no arguments in its own thread."""
devnull = os.open(os.devnull, os.O_RDWR)
for fd in range(3): os.dup2(devnull, fd)
-def deepcopy(obj):
- """Return a deep copy of obj."""
- return cPickle.loads(cPickle.dumps(obj, -1))
-
def fork_as(su, function, *args):
"""fork(), cd / to avoid keeping unused directories open, close all nonstandard file descriptors (to avoid capturing open sockets), fork() again (to avoid zombies) and call <function> with arguments <args> in the grandchild process. If <su> is not None, set our group and user ids appropriately in the child process."""
child_pid = os.fork()
write_file(PID_FILE, lambda f: f.write(str(os.getpid())))
return other_pid
-def write_file(filename, do_write):
+def write_file(filename, do_write, **kw_args):
"""Write file <filename> atomically by opening a temporary file, using <do_write> to write that file, and then renaming the temporary file."""
- os.rename(write_temp_file(do_write), filename)
+ os.rename(write_temp_file(do_write, **kw_args), filename)
-def write_temp_file(do_write):
+def write_temp_file(do_write, mode=None, uidgid=None):
fd, temporary_filename = tempfile.mkstemp()
+ if mode: os.chmod(temporary_filename, mode)
+ if uidgid: os.chown(temporary_filename, *uidgid)
f = os.fdopen(fd, 'w')
try: do_write(f)
finally: f.close()