BF+ENH: set dist-restrictions metadata correctly

Author: yarikoptic

heudiconv/external/dlad.py

@@ -1,4 +1,6 @@
+import inspect
 import os
+
 import os.path as op
 import logging
 from glob import glob
@@ -155,7 +157,12 @@ def mark_sensitive(ds, path_glob):
     paths = glob(op.join(ds.path, path_glob))
     if not paths:
         return
-    ds.repo.set_metadata(
+    lgr.debug("Marking %d files with distribution-restrictions field",
+              len(paths))
+    # set_metadata can be a bloody generator
+    res = ds.repo.set_metadata(
         paths,
-        init=[('distribution-restrictions', 'sensitive')],
-        recursive=True)
+        init=dict([('distribution-restrictions', 'sensitive')]),
+        recursive=True)
+    if inspect.isgenerator(res):
+        res = list(res)

heudiconv/external/tests/__init__.py

@@ -0,0 +1,23 @@
+from ..dlad import mark_sensitive
+from datalad.api import Dataset
+from ...utils import create_tree
+
+
+def test_mark_sensitive(tmpdir):
+    ds = Dataset(str(tmpdir)).create(force=True)
+    create_tree(
+        str(tmpdir),
+        {
+            'f1': 'd1',
+            'f2': 'd2',
+            'g1': 'd3',
+            'g2': 'd1',
+         }
+    )
+    ds.add('.')
+    mark_sensitive(ds, 'f*')
+    all_meta = dict(ds.repo.get_metadata('.'))
+    target_rec = {'distribution-restrictions': ['sensitive']}
+    # g2 since the same content
+    assert not all_meta.pop('g1', None)  # nothing or empty record
+    assert all_meta == {'f1': target_rec, 'f2': target_rec, 'g2': target_rec}

heudiconv/external/tests/test_dlad.py

@@ -388,3 +388,63 @@ def clear_temp_dicoms(item_dicoms):
 def file_md5sum(filename):
     with open(filename, 'rb') as f:
         return hashlib.md5(f.read()).hexdigest()
+
+
+# Borrowed from DataLad (MIT license), with "archives" functionality commented
+# out
+class File(object):
+    """Helper for a file entry in the create_tree/@with_tree
+
+    It allows to define additional settings for entries
+    """
+    def __init__(self, name, executable=False):
+        """
+
+        Parameters
+        ----------
+        name : str
+          Name of the file
+        executable: bool, optional
+          Make it executable
+        """
+        self.name = name
+        self.executable = executable
+
+    def __str__(self):
+        return self.name
+
+
+def create_tree(path, tree, archives_leading_dir=True):
+    """Given a list of tuples (name, load) or a dict create such a tree
+
+    if load is a tuple or a dict itself -- that would create either a subtree
+    or an archive with that content and place it into the tree if name ends
+    with .tar.gz
+    """
+    lgr.log(5, "Creating a tree under %s", path)
+    if not op.exists(path):
+        os.makedirs(path)
+
+    if isinstance(tree, dict):
+        tree = tree.items()
+
+    for file_, load in tree:
+        if isinstance(file_, File):
+            executable = file_.executable
+            name = file_.name
+        else:
+            executable = False
+            name = file_
+        full_name = op.join(path, name)
+        if isinstance(load, (tuple, list, dict)):
+            # if name.endswith('.tar.gz') or name.endswith('.tar') or name.endswith('.zip'):
+            #     create_tree_archive(path, name, load, archives_leading_dir=archives_leading_dir)
+            # else:
+            create_tree(full_name, load, archives_leading_dir=archives_leading_dir)
+        else:
+            with open(full_name, 'w') as f:
+                if sys.version_info[0] == 2 and not isinstance(load, str):
+                    load = load.encode('utf-8')
+                f.write(load)
+        if executable:
+            os.chmod(full_name, os.stat(full_name).st_mode | stat.S_IEXEC)