setup.py

# :coding: utf-8
# :copyright: Copyright (c) 2014 ftrack

import os
import re
import shutil

from setuptools.command.test import test as TestCommand
from setuptools import setup, find_packages, Command
from pkg_resources import parse_version
import pip

from pip._internal import main as pip_main

# Define paths

PLUGIN_NAME = 'ftrack-connect-nuke-{0}'

ROOT_PATH = os.path.dirname(
    os.path.realpath(__file__)
)

ROOT_PATH = os.path.dirname(os.path.realpath(__file__))

RESOURCE_PATH = os.path.join(ROOT_PATH, 'resource')

SOURCE_PATH = os.path.join(ROOT_PATH, 'source')

README_PATH = os.path.join(ROOT_PATH, 'README.rst')

BUILD_PATH = os.path.join(ROOT_PATH, 'build')

STAGING_PATH = os.path.join(BUILD_PATH, PLUGIN_NAME)

NUKE_PLUGIN_PATH = os.path.join(RESOURCE_PATH, 'nuke_path')

FTRACK_CONNECT_NUKE_PLUGIN_PATH = os.path.join(
    RESOURCE_PATH, 'ftrack_connect_nuke_plugin'
)

HOOK_PATH = os.path.join(RESOURCE_PATH, 'hook')


with open(os.path.join(
    SOURCE_PATH, 'ftrack_connect_nuke', '_version.py')
) as _version_file:
    VERSION = re.match(
        r'.*__version__ = \'(.*?)\'', _version_file.read(), re.DOTALL
    ).group(1)

# Update staging path with the plugin version
STAGING_PATH = STAGING_PATH.format(VERSION)


# Custom commands.
class PyTest(TestCommand):
    '''Pytest command.'''

    def finalize_options(self):
        TestCommand.finalize_options(self)
        self.test_args = []
        self.test_suite = True

    def run_tests(self):
        '''Import pytest and run.'''
        import pytest
        errno = pytest.main(self.test_args)
        raise SystemExit(errno)



class BuildPlugin(Command):
    '''Build plugin.'''

    description = 'Download dependencies and build plugin .'

    user_options = []

    def initialize_options(self):
        pass

    def finalize_options(self):
        pass

    def run(self):
        '''Run the build step.'''
        # Clean staging path
        shutil.rmtree(STAGING_PATH, ignore_errors=True)

        # Copy plugin files
        shutil.copytree(
            NUKE_PLUGIN_PATH,
            os.path.join(STAGING_PATH, 'resource', 'nuke_path')
        )

        # Copy plugin files
        shutil.copytree(
            FTRACK_CONNECT_NUKE_PLUGIN_PATH,
            os.path.join(STAGING_PATH, 'resource', 'ftrack_connect_nuke_plugin')
        )

        # Copy hook files
        shutil.copytree(
            HOOK_PATH,
            os.path.join(STAGING_PATH, 'hook')
        )

        # Install local dependencies
        pip_main(
            [
                'install',
                '.',
                '--target',
                os.path.join(STAGING_PATH, 'dependencies')
            ]
        )

        # Generate plugin zip
        shutil.make_archive(
            os.path.join(
                BUILD_PATH,
                PLUGIN_NAME.format(VERSION)
            ),
            'zip',
            STAGING_PATH
        )

# Configuration.
setup(
    name='ftrack-connect-nuke',
    version=VERSION,
    description='Repository for ftrack connect nuke.',
    long_description=open(README_PATH).read(),
    keywords='ftrack, connect, connector, nuke, foundry',
    url='https://bitbucket.org/ftrack/ftrack-connect-nuke',
    author='ftrack',
    author_email='support@ftrack.com',
    license='Apache License (2.0)',
    packages=find_packages(SOURCE_PATH),
    package_dir={
        '': 'source'
    },
    setup_requires=[
        'sphinx >= 1.2.2, < 2',
        'sphinx_rtd_theme >= 0.1.6, < 2',
        'lowdown >= 0.1.0, < 1'
    ],
    tests_require=[
        'pytest >= 2.3.5, < 3'
    ],
    install_requires=[
        'appdirs == 1.4.0',
        'ftrack-connect-foundry @ https://bitbucket.org/ftrack/ftrack-connect-foundry/get/1.2.1.zip#egg=ftrack-connect-foundry-1.2.1',
        'qtext @ git+https://bitbucket.org/ftrack/qtext/get/0.2.2.zip#egg=QtExt-0.2.2'
    ],
    cmdclass={
        'test': PyTest,
        'build_plugin': BuildPlugin,
    }
)