setup.py

#!/usr/bin/env python
# Copyright (c) 2012-2024, Cenobit Technologies, Inc. http://cenobit.es/
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice,
#    this list of conditions and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright notice,
#    this list of conditions and the following disclaimer in the documentation
#    and/or other materials provided with the distribution.
# * Neither the name of the Cenobit Technologies nor the names of
#    its contributors may be used to endorse or promote products derived from
#    this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
from __future__ import annotations

import os
import sys
import typing as t
import pathlib

import setuptools

USE_MYPYC = os.getenv('MYPYC_ENABLE', 'False').lower() in ('true', 't', '1')
if len(sys.argv) > 1 and sys.argv[1] == '--use-mypyc':
    USE_MYPYC = True

setup_attrs = {'name': 'Flask-JSONRPC', 'packages': setuptools.find_packages()}


def find_python_files(path: pathlib.Path) -> t.Generator[pathlib.Path, None, None]:
    return path.rglob('*.py')


if USE_MYPYC:
    from mypyc.build import mypycify

    project_dir = pathlib.Path(__file__).resolve().parent

    blocklist = []

    discovered = []
    discovered.extend(find_python_files(project_dir / 'src' / 'flask_jsonrpc'))

    ext_modules = [
        '--config-file',
        str(project_dir / 'pyproject.toml'),
        '--install-types',
        '--non-interactive',
        '--strict',
        '--check-untyped-defs',
        '--ignore-missing-imports',
        '--disallow-untyped-defs',
        '--disable-error-code',
        'unused-ignore',
        '--disable-error-code',
        'no-any-return',
        '--disable-error-code',
        'misc',
    ]
    ext_modules.extend([str(p) for p in discovered if p.relative_to(project_dir).as_posix() not in blocklist])

    opt_level = os.getenv('MYPYC_OPT_LEVEL', '3')
    setup_attrs['requires'] = ['mypy', 'pydantic']
    setup_attrs['ext_modules'] = mypycify(ext_modules, opt_level=opt_level, verbose=True)
    setup_attrs['package_data'] = {'flask_jsonrpc.contrib.browse': ['static*', 'templates*']}

setuptools.setup(**setup_attrs)