- (reverted from commit dd8f677) (reverted from commit c608bf4)

cool-RR · cool-RR · commit 276cd1c8479d · 2016-01-12T20:54:54.000+02:00
diff --git a/source_py3/python_toolbox/temp_file_tools.py b/source_py3/python_toolbox/temp_file_tools.py
@@ -15,7 +15,8 @@
 
 
 @context_management.ContextManagerType
-def create_temp_folder(*, prefix=tempfile.template, suffix='', dir=None):
+def create_temp_folder(*, prefix=tempfile.template, suffix='',
+                       parent_folder=None, chmod=None):
     '''
     Context manager that creates a temporary folder and deletes it after usage.
     
@@ -37,10 +38,18 @@ def create_temp_folder(*, prefix=tempfile.template, suffix='', dir=None):
        
     Use the `prefix` and `suffix` string arguments to dictate a prefix and/or a
     suffix to the temporary folder's name in the filesystem.
+    
+    If you'd like to set the permissions of the temporary folder, pass them to
+    the optional `chmod` argument, like this:
+    
+        create_temp_folder(chmod=0o550)
+    
     '''
     temp_folder = pathlib.Path(tempfile.mkdtemp(prefix=prefix, suffix=suffix, 
-                                                dir=dir))
+                                                dir=parent_folder))
     try:
+        if chmod is not None:
+            temp_folder.chmod(chmod)
         yield temp_folder
     finally:
         shutil.rmtree(str(temp_folder))
diff --git a/source_py3/test_python_toolbox/test_temp_file_tools/test_create_temp_folder.py b/source_py3/test_python_toolbox/test_temp_file_tools/test_create_temp_folder.py
@@ -90,9 +90,17 @@ def test_prefix_suffix():
         assert tf1.name.startswith('hocus')
         assert tf1.name.endswith('pocus')
 
-def test_dir():
+def test_parent_folder():
     with create_temp_folder() as tf1:
-        with create_temp_folder(dir=str(tf1)) as tf2:
+        with create_temp_folder(parent_folder=str(tf1)) as tf2:
             assert isinstance(tf2, pathlib.Path)
             assert str(tf2).startswith(str(tf1))
+    
+def test_chmod():
+    with create_temp_folder(chomd=0o777) as liberal_temp_folder, \
+                   create_temp_folder(chomd=0o000) as conservative_temp_folder:
+        liberal_temp_folder.stat().st_mode & 0o777
+        1 / 0 # blocktodo write this test
+        
+        
     
