Source code for soma.config

# -*- coding: utf-8 -*-

""" Config module that sets up version variable and finds the
`BrainVISA <http://brainvisa.info>`_ `brainvisa-share` data directory.

Attributes
----------
fullVersion
shortVersion
full_version
short_version
BRAINVISA_SHARE
"""

import os
import soma.info

full_version = ".".join([str(soma.info.version_major),
                        str(soma.info.version_minor),
                        str(soma.info.version_micro)])
short_version = ".".join([str(soma.info.version_major),
                         str(soma.info.version_minor)])

fullVersion = full_version
shortVersion = short_version




def path_up(path, n=1):
    ''' Move up n directories
    (in other words, apply dirname the given number of times)
    '''
    for i in range(n):
        path = os.path.dirname(path)
    return path



def _init_default_brainvisa_share():
    try:
        import brainvisa_share.config
        bv_share_dir = brainvisa_share.config.share
        has_config = True
    except ImportError:
        bv_share_dir = "brainvisa-share-%s" % short_version
        has_config = False

    if bv_share_dir and os.path.exists(bv_share_dir):
        return bv_share_dir

    share = os.getenv('BRAINVISA_SHARE')
    if share:
        # share is the base share/ directory: we must find the brainvisa-share
        # directory in it.
        share = os.path.join(share, bv_share_dir)
    if not share or not os.path.exists(share):
        if 'CONDA_PREFIX' in os.environ:
            share = os.path.join(
                os.environ['CONDA_PREFIX'],
                'share', bv_share_dir)
            if not os.path.exists(share):
                # build dir config, we are in
                # <root>/lib/pythonx.x/site_packages/soma/config.py
                share = os.path.join(path_up(__file__, 5), 'share',
                                     bv_share_dir)
        if (not share or not os.path.exists(share)) and has_config:
            share = os.path.join(os.path.dirname(os.path.dirname(
                os.path.dirname(
                    brainvisa_share.config.__file__))), 'share',
                    brainvisa_share.config.share)
        if not share or not os.path.exists(share):
            share = os.path.join(path_up(__file__, 3), 'share', bv_share_dir)
    return share

BRAINVISA_SHARE = _init_default_brainvisa_share()
""" share directory used for all BrainVisa tools """

INSTALL_ROOT = path_up(BRAINVISA_SHARE, 2)
""" Install root directory of BrainVisa """