# coding: utf-8

import os
from glob import glob

from setuptools import setup, find_packages

#---------------------------------------------------------------------------
# Basic project information
#---------------------------------------------------------------------------

# release.py contains version, authors, license, url, keywords, etc.
release_file = os.path.join('ElectronPhononCoupling', 'config', 'release.py')

with open(release_file) as f:
    code = compile(f.read(), release_file, 'exec')
    exec(code)


#---------------------------------------------------------------------------
# Find scripts
#---------------------------------------------------------------------------

def get_long_desc():
    with open("README.rst") as f:
        return f.read()

def write_manifest():
    # FIXME need a more flexible formulation for data files to exclude.
    content = """
    include *.rst
    recursive-include ElectronPhononCoupling *.py
    recursive-include ElectronPhononCoupling/scripts *
    recursive-include ElectronPhononCoupling/data *
    prune ElectronPhononCoupling/data/LiF_g2/epc_inputs/output
    graft Examples
    exclude Examples/Calculations/*/odat*
    exclude Examples/Calculations/*/*.out*
    exclude Examples/Calculations/*/*.log*
    exclude Examples/Calculations/*/*fort*
    exclude Examples/Out/*
    """
    with open('MANIFEST.in', 'write') as f:
        for line in content.strip().splitlines():
            f.write(line.strip() + '\n')


def find_package_data(dirname):
    paths = []
    for (path, directories, filenames) in os.walk(dirname):
        for filename in filenames:
            # We need the path relative to the main source directory.
            paths.append(os.path.join('..', path, filename))
    return paths

#---------------------------------------------------------------------------
# Setup
#---------------------------------------------------------------------------

write_manifest()

install_requires = [
    'numpy >=1.8',
    'mpi4py >=2.0',
    'netCDF4 >=1.2',
    ]


my_package_data = {'' :
    find_package_data('ElectronPhononCoupling/data/Psps_for_tests')
  + find_package_data('ElectronPhononCoupling/data/LiF_g2')
  + find_package_data('ElectronPhononCoupling/data/LiF_g4')
    }

my_exclude_package_data = {
    }


setup_args = dict(
      name             = name,
      version          = __version__,
      description      = description,
      long_description = long_description,
      author           = author,
      author_email     = author_email,
      url              = url,
      license          = license,
      packages         = find_packages(),
      package_data     = my_package_data,
      exclude_package_data = my_exclude_package_data,
      install_requires = install_requires,
      )

if __name__ == "__main__":
    setup(**setup_args)