sty: cleaner imports

Author: mgxd

heudiconv/bids.py

@@ -14,8 +14,14 @@
 import dcmstack as ds
 
 from .parser import find_files
-from .utils import (load_json, save_json, create_file_if_missing,
-                    json_dumps_pretty, set_readonly, is_readonly)
+from .utils import (
+    load_json,
+    save_json,
+    create_file_if_missing,
+    json_dumps_pretty,
+    set_readonly,
+    is_readonly,
+)
 
 lgr = logging.getLogger(__name__)
 

heudiconv/cli/run.py

@@ -5,7 +5,7 @@
 
 from .. import __version__, __packagename__
 from ..parser import get_study_sessions
-from ..utils import (load_heuristic, anonymize_sid, treat_infofile, SeqInfo)
+from ..utils import load_heuristic, anonymize_sid, treat_infofile, SeqInfo
 from ..convert import prep_conversion
 
 import inspect
@@ -201,7 +201,8 @@ def process_args(args):
     #
     # Load heuristic -- better do it asap to make sure it loads correctly
     #
-    heuristic = load_heuristic(op.realpath(args.heuristic_file))
+    heuristic = (load_heuristic(op.realpath(args.heuristic_file))
+                if args.heuristic_file else gen_heuristic)
 
     study_sessions = get_study_sessions(args.dicom_dir_template, args.files,
                                         heuristic, outdir, args.session,

heudiconv/convert.py

@@ -3,13 +3,30 @@
 import logging
 import shutil
 
-from .utils import (read_config, load_json, save_json, write_config, TempDirs,
-                    safe_copyfile, treat_infofile, set_readonly,
-                    clear_temp_dicoms)
-from .bids import (convert_sid_bids, populate_bids_templates, save_scans_key,
-                   tuneup_bids_json_files, add_participant_record)
-from .dicoms import (group_dicoms_into_seqinfos, embed_metadata_from_dicoms,
-                     compress_dicoms)
+from .utils import (
+    read_config,
+    load_json,
+    save_json,
+    write_config,
+    TempDirs,
+    safe_copyfile,
+    treat_infofile,
+    set_readonly,
+    clear_temp_dicoms,
+    seqinfo_fields,
+)
+from .bids import (
+    convert_sid_bids,
+    populate_bids_templates,
+    save_scans_key,
+    tuneup_bids_json_files,
+    add_participant_record,
+)
+from .dicoms import (
+    group_dicoms_into_seqinfos,
+    embed_metadata_from_dicoms,
+    compress_dicoms
+)
 
 lgr = logging.getLogger(__name__)
 
@@ -115,6 +132,7 @@ def prep_conversion(sid, dicoms, outdir, heuristic, converter, anon_sid,
         filegroup = {si.series_id: x for si, x in seqinfo.items()}
         dicominfo_file = op.join(idir, 'dicominfo%s.tsv' % ses_suffix)
         with open(dicominfo_file, 'wt') as fp:
+            fp.write('\t'.join([val for val in seqinfo_fields]) + '\n')
             for seq in seqinfo_list:
                 fp.write('\t'.join([str(val) for val in seq]) + '\n')
         lgr.debug("Calling out to %s.infodict", heuristic)

heudiconv/external/dlad.py

@@ -2,7 +2,7 @@
 import os.path as op
 import logging
 
-from ..utils import (create_file_if_missing, mark_sensitive)
+from ..utils import create_file_if_missing, mark_sensitive
 
 lgr = logging.getLogger(__name__)
 

heudiconv/parser.py

@@ -10,8 +10,14 @@
 from tempfile import mkdtemp
 
 from .dicoms import group_dicoms_into_seqinfos
-from .utils import (docstring_parameter, StudySessionInfo, load_json,
-                    save_json, create_file_if_missing, json_dumps_pretty)
+from .utils import (
+    docstring_parameter,
+    StudySessionInfo,
+    load_json,
+    save_json,
+    create_file_if_missing,
+    json_dumps_pretty,
+)
 
 lgr = logging.getLogger(__name__)