FIX: handle better dicom-only conversion

- It won't try to embed files that do not exist
- It won't create directories if output is in bids format

Author: mvdoc

heudiconv/convert.py

@@ -244,7 +244,10 @@ def convert(items, converter, scaninfo_suffix, custom_callable, with_prov,
         lgr.info('Converting %s (%d DICOMs) -> %s . '
                  'Converter: %s . Output types: %s',
                  prefix, len(item_dicoms), prefix_dirname, converter, outtypes)
-        if not op.exists(prefix_dirname):
+        # We want to create this dir only if we are converting it to nifti,
+        # or if we're using BIDS
+        dicom_only = len(outtypes) == 1 and outtypes[0] == 'dicom'
+        if not(dicom_only and bids) and not op.exists(prefix_dirname):
             os.makedirs(prefix_dirname)
 
         for outtype in outtypes:
@@ -254,6 +257,9 @@ def convert(items, converter, scaninfo_suffix, custom_callable, with_prov,
 
             seqtype = op.basename(op.dirname(prefix)) if bids else None
 
+            # set empty outname and scaninfo in case we only want dicoms
+            outname = ''
+            scaninfo = ''
             if outtype == 'dicom':
                 convert_dicom(item_dicoms, bids, prefix,
                               outdir, tempdirs, symlink, overwrite)
@@ -299,17 +305,17 @@ def convert(items, converter, scaninfo_suffix, custom_callable, with_prov,
                         "multiple files")
         elif not bids_outfiles:
             lgr.debug("No BIDS files were produced, nothing to embed to then")
-        else:
+        elif outname:
             embed_metadata_from_dicoms(bids, item_dicoms, outname, outname_bids,
                                        prov_file, scaninfo, tempdirs, with_prov,
                                        min_meta)
-        if op.exists(scaninfo):
+        if scaninfo and op.exists(scaninfo):
             lgr.info("Post-treating %s file", scaninfo)
             treat_infofile(scaninfo)
 
         # this may not always be the case: ex. fieldmap1, fieldmap2
         # will address after refactor
-        if op.exists(outname):
+        if outname and op.exists(outname):
             set_readonly(outname)
 
     if custom_callable is not None:

tests/test_heuristics.py

@@ -1,6 +1,7 @@
 from heudiconv.cli.run import main as runner
 
 import os
+import os.path as op
 from mock import patch
 from six.moves import StringIO
 
@@ -98,7 +99,7 @@ def test_scans_keys_reproin(tmpdir, invocation):
     args += invocation
     runner(args.split())
     # for now check it exists
-    scans_keys = glob(pjoin(tmpdir.strpath, '*/*/*/*/*.tsv'))
+    scans_keys = glob(pjoin(tmpdir.strpath, '*/*/*/*/*/*.tsv'))
     assert(len(scans_keys) == 1)
     with open(scans_keys[0]) as f:
         reader = csv.reader(f, delimiter='\t')
@@ -123,3 +124,23 @@ def test_ls(stdout):
     out = stdout.getvalue()
     assert 'StudySessionInfo(locator=' in out
     assert 'Halchenko/Yarik/950_bids_test4' in out
+
+
+def test_scout_conversion(tmpdir):
+    args = (
+        "-b -f reproin --files %s -o %s"
+        % (TESTS_DATA_PATH, tmpdir)
+    ).split(' ')
+    runner(args)
+    tmppath = tmpdir.strpath
+
+    assert not op.exists(pjoin(
+        tmppath,
+        'Halchenko/Yarik/950_bids_test4/sub-phantom1sid1/ses-localizer/anat'))
+
+    assert op.exists(pjoin(
+        tmppath,
+        'Halchenko/Yarik/950_bids_test4/sourcedata/sub-phantom1sid1/'
+        'ses-localizer/anat/sub-phantom1sid1_ses-localizer_scout.dicom.tgz'
+    )
+    )