ENH: Get version from file (e.g. FreeSurfer)

Author: effigies

nipype/interfaces/base.py

@@ -1927,15 +1927,27 @@ def _format_arg(self, name, spec, value):
 class PackageInfo(object):
     _version = None
     version_cmd = None
+    version_file = None
 
     @classmethod
     def version(klass):
         if klass._version is None:
-            try:
-                clout = CommandLine(command=klass.version_cmd,
-                                    resource_monitor=False,
-                                    terminal_output='allatonce').run()
-            except OSError:
+            if klass.version_cmd is not None:
+                try:
+                    clout = CommandLine(command=klass.version_cmd,
+                                        resource_monitor=False,
+                                        terminal_output='allatonce').run()
+                except OSError:
+                    return None
+
+                raw_info = clout.runtime.stdout
+            elif klass.version_file is not None:
+                try:
+                    with open(klass.version_file, 'rt') as fobj:
+                        raw_info = fobj.read()
+                except OSError:
+                    return None
+            else:
                 return None
 
         klass._version = klass.parse_version(raw_info)

nipype/interfaces/freesurfer/base.py

@@ -23,12 +23,13 @@
 from ...utils.filemanip import fname_presuffix
 from ..base import (CommandLine, Directory,
                     CommandLineInputSpec, isdefined,
-                    traits, TraitedSpec, File)
+                    traits, TraitedSpec, File,
+                    PackageInfo)
 
 __docformat__ = 'restructuredtext'
 
 
-class Info(object):
+class Info(PackageInfo):
     """ Freesurfer subject directory and version information.
 
     Examples
@@ -39,32 +40,13 @@ class Info(object):
     >>> Info.subjectsdir()  # doctest: +SKIP
 
     """
+    if os.getenv('FREESURFER_HOME'):
+        version_file = os.path.join(os.getenv('FREESURFER_HOME'),
+                                    'build-stamp.txt')
 
     @staticmethod
-    def version():
-        """Check for freesurfer version on system
-
-        Find which freesurfer is being used....and get version from
-        /path/to/freesurfer/build-stamp.txt
-
-        Returns
-        -------
-
-        version : string
-           version number as string
-           or None if freesurfer version not found
-
-        """
-        fs_home = os.getenv('FREESURFER_HOME')
-        if fs_home is None:
-            return None
-        versionfile = os.path.join(fs_home, 'build-stamp.txt')
-        if not os.path.exists(versionfile):
-            return None
-        fid = open(versionfile, 'rt')
-        version = fid.readline()
-        fid.close()
-        return version
+    def parse_version(raw_info):
+        return raw_info.splitlines()[0]
 
     @classmethod
     def looseversion(cls):