netron/setup.py

#!/usr/bin/python

import distutils
import os
import setuptools
import setuptools.command
import setuptools.command.build_py
import json

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

with open(os.path.join(TOP_DIR, 'package.json')) as package_file:
    package_manifest = json.load(package_file)
    package_version = package_manifest['version']

packages = [ 'netron' ]

package_data={
    'netron': [ 
        'netron',
        'netron.py',
        'logo.svg',
        'onnx_ml_pb2.py',
        'onnx.js',
        'onnx-operator.json',
        'view-browser.js',
        'tf.js',
        'tf-operator.pb',
        'tf-model.js',
        'tflite.js',
        'tflite-operator.json',
        'tflite-model.js',
        'hdf5.js',
        'keras-operator.json',
        'keras-model.js',
        'favicon.ico',
        'view-browser.html',
        'onnx-model.js',
        'view-render.css',
        'view-render.js',
        'view-template.js',
        'view.css',
        'view.js',
    ]
}

install_requires = [ 'protobuf' ]

scripts = [ 'src/netron' ]

custom_files = [ 
    ( 'netron', [
        'node_modules/d3/build/d3.min.js',
        'node_modules/dagre/dist/dagre.min.js',
        'node_modules/handlebars/dist/handlebars.min.js',
        'node_modules/marked/marked.min.js',
        'node_modules/protobufjs/dist/protobuf.min.js',
        'node_modules/flatbuffers/js/flatbuffers.js',
        'node_modules/npm-font-open-sans/open-sans.css' ]),
    ( 'netron/fonts/Regular', [
        'node_modules/npm-font-open-sans/fonts/Regular/OpenSans-Regular.eot',
        'node_modules/npm-font-open-sans/fonts/Regular/OpenSans-Regular.svg',
        'node_modules/npm-font-open-sans/fonts/Regular/OpenSans-Regular.ttf',
        'node_modules/npm-font-open-sans/fonts/Regular/OpenSans-Regular.woff',
        'node_modules/npm-font-open-sans/fonts/Regular/OpenSans-Regular.woff2' ]),
    ( 'netron/fonts/Semibold', [
        'node_modules/npm-font-open-sans/fonts/Semibold/OpenSans-Semibold.eot',
        'node_modules/npm-font-open-sans/fonts/Semibold/OpenSans-Semibold.svg',
        'node_modules/npm-font-open-sans/fonts/Semibold/OpenSans-Semibold.ttf',
        'node_modules/npm-font-open-sans/fonts/Semibold/OpenSans-Semibold.woff',
        'node_modules/npm-font-open-sans/fonts/Semibold/OpenSans-Semibold.woff2' ]),
    ( 'netron/fonts/Bold', [
        'node_modules/npm-font-open-sans/fonts/Bold/OpenSans-Bold.eot',
        'node_modules/npm-font-open-sans/fonts/Bold/OpenSans-Bold.svg',
        'node_modules/npm-font-open-sans/fonts/Bold/OpenSans-Bold.ttf',
        'node_modules/npm-font-open-sans/fonts/Bold/OpenSans-Bold.woff',
        'node_modules/npm-font-open-sans/fonts/Bold/OpenSans-Bold.woff2' ])
]

class build_py(setuptools.command.build_py.build_py):
    def run(self):
        result = setuptools.command.build_py.build_py.run(self)
        for target, files in custom_files:
            target = os.path.join(self.build_lib, target)
            if not os.path.exists(target):
                os.makedirs(target)
            for file in files:
                self.copy_file(file, target)
        return result
    def get_outputs(self, include_bytecode=1):
        result = setuptools.command.build_py.build_py.get_outputs(self, include_bytecode)
        print("## get_outputs ##")
        return result

setuptools.setup(
    name="netron",
    version=package_version,
    description="Viewer for neural network models",
    license="MIT",
    cmdclass={ 'build_py': build_py },
    package_dir={ 'netron': 'src' },
    packages=packages,
    package_data=package_data,
    install_requires=install_requires,
    author='Lutz Roeder',
    author_email='[email protected]',
    url='https://github.com/lutzroeder/Netron',
    scripts=scripts,
    classifiers=[
        'Intended Audience :: Developers',
        'Intended Audience :: Science/Research',
        'Topic :: Software Development',
        'Topic :: Scientific/Engineering :: Artificial Intelligence'
    ]
)