setup.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @author: vuolter
#      ____________
#   _ /       |    \ ___________ _ _______________ _ ___ _______________
#  /  |    ___/    |   _ __ _  _| |   ___  __ _ __| |   \\    ___  ___ _\
# /   \___/  ______/  | '_ \ || | |__/ _ \/ _` / _` |    \\  / _ \/ _ `/ \
# \       |   o|      | .__/\_, |____\___/\__,_\__,_|    // /_//_/\_, /  /
#  \______\    /______|_|___|__/________________________//______ /___/__/
#          \  /
#           \/

from __future__ import absolute_import

import codecs
import os
import re
import shutil
import subprocess

from itertools import chain

from setuptools import Command, find_packages, setup
from setuptools.command.bdist_egg import bdist_egg
from setuptools.command.build_py import build_py


def read_text(path):
    with codecs.open(path, encoding='utf-8', errors='ignore') as fp:
        return fp.read().strip()


def write_text(path, text):
    with codecs.open(path, mode='w', encoding='utf-8') as fp:
        fp.write(text.strip() + '\n')


_RE_SECTION = re.compile(
    r'^\[([^\s]+)\]\s+([^[\s].+?)(?=^\[|\Z)', flags=re.M | re.S)
_RE_ENTRY = re.compile(r'^\s*(?![#;\s]+)([^\s]+)', flags=re.M)

def _parse_requires(text):
    deps = list(set(
        _RE_ENTRY.findall(_RE_SECTION.split(text, maxsplit=1)[0])))
    extras = {}
    for name, rawdeps in _RE_SECTION.findall(text):
        extras[name] = list(set(pack for pack in _RE_ENTRY.findall(rawdeps)))
    return deps, extras


def get_requires(name):
    path = os.path.join('requirements', name + '.txt')
    text = read_text(path)
    deps, extras = _parse_requires(text)
    if name.startswith('extra'):
        extras['full'] = list(set(chain(*list(extras.values()))))
        return extras
    return deps

    
# class BuildLocale(Command):
    # """
    # Build the locales
    # """
    # description = 'build locales'
    # user_options = []
    
    # def run(self):
        # self.run_command('extract_messages')
        # self.run_command('init_catalog')
        # # self.run_command('download_catalog')
        # self.run_command('compile_catalog')

        
class BuildNode(Command):
    """
    BuildPy the nodejs app
    """
    description = 'build the nodejs app'
    user_options = []

    def run(self):
        if os.path.isdir('pyload/webui/node_modules'):
            return
            
        try:
            # if os.name == 'nt':
                # # : Required by npm package `grunt-contrib-compress` under Windows
                # subprocess.check_call(
                    # 'npm install --global windows-build-tools', shell=True)
            subprocess.check_call(
                'cd {0} && npm install --only=dev'.format('pyload/webui'),
                shell=True)
                
            subprocess.check_call(
                'cd {0} && node node_modules/grunt-cli/bin/grunt build'.format(
                    'pyload/webui'),
                shell=True)
                
        except subprocess.CalledProcessError:
            distutils.log.warn("Failed to build the nodejs app")
            
        shutil.rmtree('pyload/webui/node_modules', ignore_errors=True)
        
        return subprocess.call(
            'cd {0} && npm install --production'.format('pyload/webui'),
            shell=True)
            
            
# class DownloadCatalog(Command):
# """
# Download the translation catalog from the remote repository
# """
# description = 'download the translation catalog from the remote repository'
# user_options = []

# def initialize_options(self):
# pass

# def finalize_options(self):
# pass

# def run(self):
# raise NotImplementedError


class BdistEgg(bdist_egg):
    """
    Custom ``bdist_egg`` command
    """
    def run(self):
        if not self.dry_run:
            self.run_command('build_node')
            # self.run_command('build_locale')
        bdist_egg.run(self)


class BuildPy(build_py):
    """
    Custom ``build_py`` command
    """
    def run(self):
        if not self.dry_run:
            self.run_command('build_node')
            # self.run_command('build_locale')
        build_py.run(self)

        
setup(
    name='pyload-webui',
    version=read_text('VERSION'),
    description='pyLoad Web User Interface',
    long_description=read_text('README.rst'),
    keywords='pyload download download-manager download-station downloader '
             'jdownloader one-click-hoster upload upload-manager '
             'upload-station uploader',
    url='https://pyload.net',
    download_url='https://github.com/pyload/pyload-webui/releases',
    author='Walter Purcaro',
    author_email='vuolter@gmail.com',
    license='GNU Affero General Public License v3',
    classifiers=[
        "Development Status :: 1 - Planning",
        "Environment :: Web Environment",
        "Intended Audience :: End Users/Desktop",
        "License :: OSI Approved :: GNU Affero General Public License v3",
        "Natural Language :: English",
        # "Operating System :: MacOS :: MacOS X",
        "Operating System :: Microsoft :: Windows",
        "Operating System :: POSIX",
        "Programming Language :: Python :: 2",
        "Programming Language :: Python :: 2.6",
        "Programming Language :: Python :: 2.7",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.3",
        "Programming Language :: Python :: 3.4",
        "Programming Language :: Python :: 3.5",
        "Programming Language :: Python :: 3.6",
        "Programming Language :: Python :: Implementation :: CPython",
        # "Programming Language :: Python :: Implementation :: PyPy",
        "Topic :: Communications",
        "Topic :: Communications :: File Sharing",
        "Topic :: Internet",
        "Topic :: Internet :: File Transfer Protocol (FTP)",
        "Topic :: Internet :: WWW/HTTP",
        'Topic :: Software Development :: Libraries :: Python Modules'],
    platforms=['any'],
    packages=['pyload'],
    include_package_data=True,
    install_requires=get_requires('install'),
    setup_requires=get_requires('setup'),
    extras_require=get_requires('extra'),
    python_requires='>=2.6,!=3.0,!=3.1,!=3.2',
    cmdclass={
        'bdist_egg': BdistEgg,
        # 'build_locale': BuildLocale,
        'build_node': BuildNode,
        'build_py': BuildPy,
        # 'download_catalog': DownloadCatalog
    },
    # message_extractors={
        # 'pyload': [
            # ('**.py', 'python', None), 
            # ('**.js', 'javascript', None)]},
    zip_safe=True)