Merge pull request #691 from AnswerDotAI/add-uid-guid-to-mk_write

Add uid and gid to mk_write as optional parameters

Author: jph00

fastcore/xtras.py

@@ -338,10 +338,11 @@ def read_json(self:Path, encoding=None, errors=None):
 
 # %% ../nbs/03_xtras.ipynb
 @patch
-def mk_write(self:Path, data, encoding=None, errors=None, mode=511):
+def mk_write(self:Path, data, encoding=None, errors=None, mode=511, uid=-1, gid=-1):
     "Make all parent dirs of `self`, and write `data`"
     self.parent.mkdir(exist_ok=True, parents=True, mode=mode)
     self.write_text(data, encoding=encoding, errors=errors)
+    if uid!=-1 or gid!=-1: os.chown(self, uid, gid)
 
 # %% ../nbs/03_xtras.ipynb
 @patch

nbs/03_xtras.ipynb

@@ -1167,10 +1167,11 @@
    "source": [
     "#|export\n",
     "@patch\n",
-    "def mk_write(self:Path, data, encoding=None, errors=None, mode=511):\n",
+    "def mk_write(self:Path, data, encoding=None, errors=None, mode=511, uid=-1, gid=-1):\n",
     "    \"Make all parent dirs of `self`, and write `data`\"\n",
     "    self.parent.mkdir(exist_ok=True, parents=True, mode=mode)\n",
-    "    self.write_text(data, encoding=encoding, errors=errors)"
+    "    self.write_text(data, encoding=encoding, errors=errors)\n",
+    "    if uid!=-1 or gid!=-1: os.chown(self, uid, gid)"
    ]
   },
   {