setup.py

"""
Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
SPDX-License-Identifier: Apache-2.0
"""

from os.path import join as pjoin
from setuptools import setup, find_packages
from setupbase import (
    create_cmdclass, install_npm, ensure_targets,
    combine_commands, HERE, widgets_root
)

nb_path = pjoin(widgets_root, 'nbextension')
lab_path = pjoin(widgets_root, 'labextension')

js_targets = [
    pjoin(nb_path, 'index.js'),
    pjoin(lab_path, 'package.json')
]

package_data_spec = {
    'graph_notebook_widgets': [
        'nbextension/**js*',
        'labextension/**'
    ]
}

data_files_spec = [
    ('share/jupyter/nbextensions/graph_notebook_widgets', nb_path, '**'),
    ('share/jupyter/labextensions/graph_notebook_widgets', lab_path, '**'),
    ('share/jupyter/labextensions/graph_notebook_widgets', HERE, 'install.json'),
    ('etc/jupyter/nbconfig/notebook.d', HERE, 'graph_notebook_widgets.json')
]

cmd_class = create_cmdclass('jsdeps', package_data_spec=package_data_spec, data_files_spec=data_files_spec)
cmd_class['jsdeps'] = combine_commands(
    install_npm(widgets_root, build_cmd='build:prod'),
    ensure_targets(js_targets),
)


def get_version():
    with open('src/graph_notebook/__init__.py') as f:
        for line in f:
            if line.startswith('__version__'):
                _, _, version = line.replace("'", '').split()
                break
    if version == '':
        raise ValueError('no version found')
    return version


with open("README.md", "r") as fh:
    long_description = fh.read()

setup(
    name='graph-notebook',
    author='amazon-neptune',
    author_email='amazon-neptune-pypi@amazon.com',
    description='jupyter notebook extension to connect to graph databases',
    long_description=long_description,
    long_description_content_type='text/markdown',
    url='https://github.com/aws/graph-notebook',
    version=get_version(),
    packages=find_packages(where='src', exclude=('test',)),
    package_dir={'': 'src'},
    include_package_data=True,
    install_requires=[
        'gremlinpython>=3.5.1,<=3.7.2',
        'SPARQLWrapper==2.0.0',
        'requests>=2.32.0,<=2.32.2',
        'ipywidgets==7.7.2',
        'jupyterlab_widgets>=1.0.0,<=1.1.10',
        'networkx==2.4',
        'Jinja2>=3.0.3,<=3.1.4',
        'notebook>=6.1.5,<7.0.0',
        'nbclient<=0.7.3',
        'jupyter-contrib-nbextensions<=0.7.0',
        'widgetsnbextension<=3.6.1',
        'jupyter==1.0.0',
        'botocore>=1.34.74',
        'boto3>=1.34.74',
        'ipython>=7.16.1,<=8.10.0',
        'neo4j>=5.0.0,<=5.23.1',
        'rdflib==7.0.0',
        'ipykernel==5.3.4',
        'ipyfilechooser==0.6.0',
        'nbconvert>=6.3.0,<=7.2.8',
        'jedi>=0.18.1,<=0.18.2',
        'itables>=2.0.0,<=2.1.0',
        'pandas>=2.1.0,<=2.2.2',
        'numpy<1.24.0',
        'nest_asyncio>=1.5.5,<=1.6.0',
        'async-timeout>=4.0,<5.0',
        'json-repair==0.29.2'
    ],
    package_data={
        'graph_notebook': ['graph_notebook/widgets/nbextensions/**',
                           'graph_notebook/widgets/labextension/**'],
        '': ['*.ipynb', '*.html', '*.css', '*.js', '*.txt', '*.json', '*.ts', '*.css', '*.yaml', '*.md', '*.tgz']
    },
    cmdclass=cmd_class,
    classifiers=[
        'Development Status :: 5 - Production/Stable',
        'Programming Language :: Python :: 3.10',
        'License :: OSI Approved :: Apache Software License'
    ],
    keywords='jupyter neptune gremlin sparql',
    extras_require={
        'test': [
            'pytest==6.2.5'
            ],
    }
)