584c4516f0
* Initial attempt at extra checks in CI land * Adding flake8 config * Adding pre-commit configuration and explanation in CONTRIBUTING.md
148 lines
5.0 KiB
Python
148 lines
5.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
DESCRIPTION = """\
|
|
PyQtGraph is a pure-python graphics and GUI library built on PyQt4/PyQt5/PySide/PySide2 and
|
|
numpy.
|
|
|
|
It is intended for use in mathematics / scientific / engineering applications.
|
|
Despite being written entirely in python, the library is very fast due to its
|
|
heavy leverage of numpy for number crunching, Qt's GraphicsView framework for
|
|
2D display, and OpenGL for 3D display.
|
|
"""
|
|
|
|
setupOpts = dict(
|
|
name='pyqtgraph',
|
|
description='Scientific Graphics and GUI Library for Python',
|
|
long_description=DESCRIPTION,
|
|
license = 'MIT',
|
|
url='http://www.pyqtgraph.org',
|
|
author='Luke Campagnola',
|
|
author_email='luke.campagnola@gmail.com',
|
|
classifiers = [
|
|
"Programming Language :: Python",
|
|
"Programming Language :: Python :: 2",
|
|
"Programming Language :: Python :: 2.7",
|
|
"Programming Language :: Python :: 3",
|
|
"Development Status :: 4 - Beta",
|
|
"Environment :: Other Environment",
|
|
"Intended Audience :: Science/Research",
|
|
"License :: OSI Approved :: MIT License",
|
|
"Operating System :: OS Independent",
|
|
"Topic :: Software Development :: Libraries :: Python Modules",
|
|
"Topic :: Scientific/Engineering :: Visualization",
|
|
"Topic :: Software Development :: User Interfaces",
|
|
],
|
|
)
|
|
|
|
|
|
import distutils.dir_util
|
|
from distutils.command import build
|
|
import os, sys, re
|
|
try:
|
|
import setuptools
|
|
from setuptools import setup
|
|
from setuptools.command import install
|
|
except ImportError:
|
|
sys.stderr.write("Warning: could not import setuptools; falling back to distutils.\n")
|
|
from distutils.core import setup
|
|
from distutils.command import install
|
|
|
|
|
|
# Work around mbcs bug in distutils.
|
|
# http://bugs.python.org/issue10945
|
|
import codecs
|
|
try:
|
|
codecs.lookup('mbcs')
|
|
except LookupError:
|
|
ascii = codecs.lookup('ascii')
|
|
func = lambda name, enc=ascii: {True: enc}.get(name=='mbcs')
|
|
codecs.register(func)
|
|
|
|
|
|
path = os.path.split(__file__)[0]
|
|
sys.path.insert(0, os.path.join(path, 'tools'))
|
|
import setupHelpers as helpers
|
|
|
|
## generate list of all sub-packages
|
|
allPackages = (helpers.listAllPackages(pkgroot='pyqtgraph') +
|
|
['pyqtgraph.'+x for x in helpers.listAllPackages(pkgroot='examples')])
|
|
|
|
## Decide what version string to use in the build
|
|
version, forcedVersion, gitVersion, initVersion = helpers.getVersionStrings(pkg='pyqtgraph')
|
|
|
|
|
|
|
|
class Build(build.build):
|
|
"""
|
|
* Clear build path before building
|
|
"""
|
|
def run(self):
|
|
global path
|
|
|
|
## Make sure build directory is clean
|
|
buildPath = os.path.join(path, self.build_lib)
|
|
if os.path.isdir(buildPath):
|
|
distutils.dir_util.remove_tree(buildPath)
|
|
|
|
ret = build.build.run(self)
|
|
|
|
|
|
class Install(install.install):
|
|
"""
|
|
* Check for previously-installed version before installing
|
|
* Set version string in __init__ after building. This helps to ensure that we
|
|
know when an installation came from a non-release code base.
|
|
"""
|
|
def run(self):
|
|
global path, version, initVersion, forcedVersion, installVersion
|
|
|
|
name = self.config_vars['dist_name']
|
|
path = os.path.join(self.install_libbase, 'pyqtgraph')
|
|
if os.path.exists(path):
|
|
raise Exception("It appears another version of %s is already "
|
|
"installed at %s; remove this before installing."
|
|
% (name, path))
|
|
print("Installing to %s" % path)
|
|
rval = install.install.run(self)
|
|
|
|
|
|
# If the version in __init__ is different from the automatically-generated
|
|
# version string, then we will update __init__ in the install directory
|
|
if initVersion == version:
|
|
return rval
|
|
|
|
try:
|
|
initfile = os.path.join(path, '__init__.py')
|
|
data = open(initfile, 'r').read()
|
|
open(initfile, 'w').write(re.sub(r"__version__ = .*", "__version__ = '%s'" % version, data))
|
|
installVersion = version
|
|
except:
|
|
sys.stderr.write("Warning: Error occurred while setting version string in build path. "
|
|
"Installation will use the original version string "
|
|
"%s instead.\n" % (initVersion)
|
|
)
|
|
if forcedVersion:
|
|
raise
|
|
installVersion = initVersion
|
|
sys.excepthook(*sys.exc_info())
|
|
|
|
return rval
|
|
|
|
|
|
setup(
|
|
version=version,
|
|
cmdclass={'build': Build,
|
|
'install': Install,
|
|
'deb': helpers.DebCommand,
|
|
'test': helpers.TestCommand,
|
|
'debug': helpers.DebugCommand,
|
|
'mergetest': helpers.MergeTestCommand,
|
|
'style': helpers.StyleCommand},
|
|
packages=allPackages,
|
|
package_dir={'pyqtgraph.examples': 'examples'}, ## install examples along with the rest of the source
|
|
package_data={'pyqtgraph.examples': ['optics/*.gz', 'relativity/presets/*.cfg']},
|
|
install_requires = [
|
|
'numpy',
|
|
],
|
|
**setupOpts
|
|
)
|