setup.py

from setuptools import setup, find_packages
from os import path, listdir
from functools import partial
from itertools import imap, ifilter
from ast import parse
from pip import __file__ as pip_loc

if __name__ == '__main__':
    package_name = 'yq'

    f_for = partial(path.join, path.dirname(__file__), package_name)
    d_for = partial(path.join, path.dirname(path.dirname(pip_loc)), package_name)

    _data_join = partial(path.join, f_for('data'))
    _data_install_dir = partial(path.join, d_for('data'))

    get_vals = lambda var0, var1: imap(lambda buf: next(imap(lambda e: e.value.s, parse(buf).body)),
                                       ifilter(lambda line: line.startswith(var0) or line.startswith(var1), f))

    with open(path.join(package_name, '__init__.py')) as f:
        __author__, __version__ = get_vals('__version__', '__author__')

    setup(
            name=package_name,
            author=__author__,
            version=__version__,
            description='Pure Python implementation of a subset of the features of jq for YAML documents',
            test_suite=package_name + '.tests',
            packages=find_packages(),
            package_dir={package_name: package_name},
            data_files=[
                (_data_install_dir(), map(_data_join, listdir(_data_join()))),
            ]
    )