requirements.py

# -*- coding: utf-8 -*-
# This file is part of Invenio.
# Copyright (C) 2013, 2014 CERN.
#
# Invenio is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License as
# published by the Free Software Foundation; either version 2 of the
# License, or (at your option) any later version.
#
# Invenio is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Invenio; if not, write to the Free Software Foundation, Inc.,
# 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.

"""Generate minimal requirements from setup.py."""

from __future__ import print_function

import mock
import pkg_resources
import setuptools
import sys


if __name__ == '__main__':
    with mock.patch.object(setuptools, 'setup') as mock_setup:
        import setup  # pylint: disable=F401

    # called arguments are in `mock_setup.call_args`
    args, kwargs = mock_setup.call_args
    install_requires = kwargs.get('install_requires', [])

    for pkg in pkg_resources.parse_requirements(install_requires):
        if len(pkg.specs):
            if pkg.specs[0][0] == '>=':
                print("{0.project_name}=={0.specs[0][1]}".format(pkg))
            elif pkg.specs[0][0] == '>':
                print(
                    "{0.project_name} specify exact minimal version using "
                    "'>=' instead of '>'.".format(pkg), file=sys.stderr)