import setuptools
import subprocess
import os
import shutil

try:
    git_describe = subprocess.check_output(
        ['git', 'describe', '--tags', '--long']).decode('utf-8').strip()
    git_branch = subprocess.check_output(
        ['git', 'rev-parse', '--abbrev-ref', 'HEAD']).decode('utf-8').strip()
    version = f'{git_describe}-{git_branch}'

    with open('petrel_client/version.py', 'w') as f:
        f.write(f"version = '{version}'\n")
        f.truncate()
except Exception:
    from importlib.machinery import SourceFileLoader
    version_module = SourceFileLoader(
        'version_module', 'petrel_client/version.py').load_module()
    version = version_module.version

dist_path = 'dist'
if os.path.exists(dist_path):
    shutil.rmtree(dist_path)

setuptools.setup(
    name='petrel-oss-sdk',
    version=version,
    description='Ceph S3 storage API for Pytorch, Parrots',
    url="http://gitlab.bj.sensetime.com/platform/StorageSystem/petrel-oss-python-sdk",
    packages=setuptools.find_packages(),
    package_data={'': ['**/*.so']},
    install_requires=['boto3', 'environs', 'coloredlogs',
                      'humanize', 'multiprocessing-logging'],
    python_requires='>=3.6',
    zip_safe=False,
)