Allow callers to specify desired uid/gid

jdegges · jdegges · commit 3ea7407da0ec · 2021-08-18T17:25:20.000-07:00
It can be useful for a file to be created atomically with the desired
user and group. Let's allow callers to specify their desired uid/gid and
change ownership of the intermediate temp file before it is moved into
place.
diff --git a/atomicwrites/__init__.py b/atomicwrites/__init__.py
@@ -182,6 +182,13 @@ def get_fileobject(self, suffix="", prefix=tempfile.gettempprefix(),
             dir = os.path.normpath(os.path.dirname(self._path))
         descriptor, name = tempfile.mkstemp(suffix=suffix, prefix=prefix,
                                             dir=dir)
+
+        # If specified, change user and group.
+        uid = kwargs.get('uid', default=-1)
+        gid = kwargs.get('gid', default=-1)
+        if uid != -1 or gid != -1:
+            os.fchown(descriptor, uid, gid)
+
         # io.open() will take either the descriptor or the name, but we need
         # the name later for commit()/replace_atomic() and couldn't find a way
         # to get the filename from the descriptor.
