view python/dependencies.py @ 258:2862f4453296

add a hack to override setuptools
author Jeff Hammel <jhammel@mozilla.com>
date Wed, 02 Jan 2013 15:41:11 -0800
parents
children a529f0d3c259
line wrap: on
line source

#!/usr/bin/env python

import imp
import os
import pprint
import sys

current_module = None
info = {}

def setup(**kwargs):
    assert current_module
    info[current_module] = kwargs

def main(args=sys.argv[1:]):

    global current_module
    current_module = None

    setuptools = sys.modules.get('setuptools')
    sys.modules['setuptools'] = sys.modules[__name__]

    try:
        for setup_py in args:
            current_module = setup_py
            assert os.path.exists(setup_py)
            module = imp.load_source('setup', setup_py)
    except:
        sys.modules.pop('setuptools')
        if setuptools:
            sys.modules['setuptools'] = setuptoools

    pprint.pprint(info)

if __name__ == '__main__':
    main()