bugbug/setup.py

80 строки
2.3 KiB
Python

# -*- coding: utf-8 -*-
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this file,
# You can obtain one at http://mozilla.org/MPL/2.0/.
import os
from setuptools import find_packages, setup
here = os.path.dirname(__file__)
def read_requirements(file_):
requires = []
links = []
with open(os.path.join(here, file_)) as f:
for line in f.readlines():
line = line.strip()
if line.startswith("https://"):
links.append(line + "-1.0.0")
extras = ""
if "[" in line:
extras = "[" + line.split("[")[1].split("]")[0] + "]"
line = line.split("#")[1].split("egg=")[1] + extras
elif line == "" or line.startswith("#") or line.startswith("-"):
continue
line = line.split("#")[0].strip()
requires.append(line)
return sorted(list(set(requires))), links
install_requires, dependency_links = read_requirements("requirements.txt")
with open(os.path.join(here, "VERSION")) as f:
version = f.read().strip()
# Read the extra requirements
extras = ["nlp", "nn"]
extras_require = {}
for extra in extras:
extras_install, extra_links = read_requirements("extra-%s-requirements.txt" % extra)
# Merge the dependency links
dependency_links.extend(extra_links)
extras_require[extra] = extras_install
setup(
name="bugbug",
version=version,
description="ML tools for Mozilla projects",
author="Marco Castelluccio",
author_email="mcastelluccio@mozilla.com",
install_requires=install_requires,
extras_require=extras_require,
dependency_links=dependency_links,
packages=find_packages(exclude=["contrib", "docs", "tests"]),
include_package_data=True,
license="MPL2",
entry_points={
"console_scripts": [
"bugbug-data-commits = scripts.commit_retriever:main",
"bugbug-data-bugzilla = scripts.bug_retriever:main",
"bugbug-train = scripts.trainer:main",
"bugbug-check = scripts.check:main",
]
},
classifiers=[
"Programming Language :: Python :: 3.6",
"Programming Language :: Python :: 3 :: Only",
"License :: OSI Approved :: Mozilla Public License 2.0 (MPL 2.0)",
],
)