Initial commit.
This commit is contained in:
parent
a07225a0af
commit
97536a47da
5 changed files with 666 additions and 0 deletions
248
Adafruit_TCS34725/TCS34725.py
Normal file
248
Adafruit_TCS34725/TCS34725.py
Normal file
|
|
@ -0,0 +1,248 @@
|
|||
# The MIT License (MIT)
|
||||
#
|
||||
# Copyright (c) 2016 Adafruit Industries
|
||||
#
|
||||
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
# of this software and associated documentation files (the "Software"), to deal
|
||||
# in the Software without restriction, including without limitation the rights
|
||||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
# copies of the Software, and to permit persons to whom the Software is
|
||||
# furnished to do so, subject to the following conditions:
|
||||
#
|
||||
# The above copyright notice and this permission notice shall be included in all
|
||||
# copies or substantial portions of the Software.
|
||||
#
|
||||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
# SOFTWARE.
|
||||
import time
|
||||
|
||||
|
||||
TCS34725_ADDRESS = 0x29
|
||||
TCS34725_ID = 0x12 # 0x44 = TCS34721/TCS34725, 0x4D = TCS34723/TCS34727
|
||||
|
||||
TCS34725_COMMAND_BIT = 0x80
|
||||
|
||||
TCS34725_ENABLE = 0x00
|
||||
TCS34725_ENABLE_AIEN = 0x10 # RGBC Interrupt Enable
|
||||
TCS34725_ENABLE_WEN = 0x08 # Wait enable - Writing 1 activates the wait timer
|
||||
TCS34725_ENABLE_AEN = 0x02 # RGBC Enable - Writing 1 actives the ADC, 0 disables it
|
||||
TCS34725_ENABLE_PON = 0x01 # Power on - Writing 1 activates the internal oscillator, 0 disables it
|
||||
TCS34725_ATIME = 0x01 # Integration time
|
||||
TCS34725_WTIME = 0x03 # Wait time (if TCS34725_ENABLE_WEN is asserted)
|
||||
TCS34725_WTIME_2_4MS = 0xFF # WLONG0 = 2.4ms WLONG1 = 0.029s
|
||||
TCS34725_WTIME_204MS = 0xAB # WLONG0 = 204ms WLONG1 = 2.45s
|
||||
TCS34725_WTIME_614MS = 0x00 # WLONG0 = 614ms WLONG1 = 7.4s
|
||||
TCS34725_AILTL = 0x04 # Clear channel lower interrupt threshold
|
||||
TCS34725_AILTH = 0x05
|
||||
TCS34725_AIHTL = 0x06 # Clear channel upper interrupt threshold
|
||||
TCS34725_AIHTH = 0x07
|
||||
TCS34725_PERS = 0x0C # Persistence register - basic SW filtering mechanism for interrupts
|
||||
TCS34725_PERS_NONE = 0b0000 # Every RGBC cycle generates an interrupt
|
||||
TCS34725_PERS_1_CYCLE = 0b0001 # 1 clean channel value outside threshold range generates an interrupt
|
||||
TCS34725_PERS_2_CYCLE = 0b0010 # 2 clean channel values outside threshold range generates an interrupt
|
||||
TCS34725_PERS_3_CYCLE = 0b0011 # 3 clean channel values outside threshold range generates an interrupt
|
||||
TCS34725_PERS_5_CYCLE = 0b0100 # 5 clean channel values outside threshold range generates an interrupt
|
||||
TCS34725_PERS_10_CYCLE = 0b0101 # 10 clean channel values outside threshold range generates an interrupt
|
||||
TCS34725_PERS_15_CYCLE = 0b0110 # 15 clean channel values outside threshold range generates an interrupt
|
||||
TCS34725_PERS_20_CYCLE = 0b0111 # 20 clean channel values outside threshold range generates an interrupt
|
||||
TCS34725_PERS_25_CYCLE = 0b1000 # 25 clean channel values outside threshold range generates an interrupt
|
||||
TCS34725_PERS_30_CYCLE = 0b1001 # 30 clean channel values outside threshold range generates an interrupt
|
||||
TCS34725_PERS_35_CYCLE = 0b1010 # 35 clean channel values outside threshold range generates an interrupt
|
||||
TCS34725_PERS_40_CYCLE = 0b1011 # 40 clean channel values outside threshold range generates an interrupt
|
||||
TCS34725_PERS_45_CYCLE = 0b1100 # 45 clean channel values outside threshold range generates an interrupt
|
||||
TCS34725_PERS_50_CYCLE = 0b1101 # 50 clean channel values outside threshold range generates an interrupt
|
||||
TCS34725_PERS_55_CYCLE = 0b1110 # 55 clean channel values outside threshold range generates an interrupt
|
||||
TCS34725_PERS_60_CYCLE = 0b1111 # 60 clean channel values outside threshold range generates an interrupt
|
||||
TCS34725_CONFIG = 0x0D
|
||||
TCS34725_CONFIG_WLONG = 0x02 # Choose between short and long (12x) wait times via TCS34725_WTIME
|
||||
TCS34725_CONTROL = 0x0F # Set the gain level for the sensor
|
||||
TCS34725_ID = 0x12 # 0x44 = TCS34721/TCS34725, 0x4D = TCS34723/TCS34727
|
||||
TCS34725_STATUS = 0x13
|
||||
TCS34725_STATUS_AINT = 0x10 # RGBC Clean channel interrupt
|
||||
TCS34725_STATUS_AVALID = 0x01 # Indicates that the RGBC channels have completed an integration cycle
|
||||
|
||||
TCS34725_CDATAL = 0x14 # Clear channel data
|
||||
TCS34725_CDATAH = 0x15
|
||||
TCS34725_RDATAL = 0x16 # Red channel data
|
||||
TCS34725_RDATAH = 0x17
|
||||
TCS34725_GDATAL = 0x18 # Green channel data
|
||||
TCS34725_GDATAH = 0x19
|
||||
TCS34725_BDATAL = 0x1A # Blue channel data
|
||||
TCS34725_BDATAH = 0x1B
|
||||
|
||||
TCS34725_INTEGRATIONTIME_2_4MS = 0xFF # 2.4ms - 1 cycle - Max Count: 1024
|
||||
TCS34725_INTEGRATIONTIME_24MS = 0xF6 # 24ms - 10 cycles - Max Count: 10240
|
||||
TCS34725_INTEGRATIONTIME_50MS = 0xEB # 50ms - 20 cycles - Max Count: 20480
|
||||
TCS34725_INTEGRATIONTIME_101MS = 0xD5 # 101ms - 42 cycles - Max Count: 43008
|
||||
TCS34725_INTEGRATIONTIME_154MS = 0xC0 # 154ms - 64 cycles - Max Count: 65535
|
||||
TCS34725_INTEGRATIONTIME_700MS = 0x00 # 700ms - 256 cycles - Max Count: 65535
|
||||
|
||||
TCS34725_GAIN_1X = 0x00 # No gain
|
||||
TCS34725_GAIN_4X = 0x01 # 2x gain
|
||||
TCS34725_GAIN_16X = 0x02 # 16x gain
|
||||
TCS34725_GAIN_60X = 0x03 # 60x gain
|
||||
|
||||
# Lookup table for integration time delays.
|
||||
INTEGRATION_TIME_DELAY = {
|
||||
0xFF: 0.0024, # 2.4ms - 1 cycle - Max Count: 1024
|
||||
0xF6: 0.024, # 24ms - 10 cycles - Max Count: 10240
|
||||
0xEB: 0.050, # 50ms - 20 cycles - Max Count: 20480
|
||||
0xD5: 0.101, # 101ms - 42 cycles - Max Count: 43008
|
||||
0xC0: 0.154, # 154ms - 64 cycles - Max Count: 65535
|
||||
0x00: 0.700 # 700ms - 256 cycles - Max Count: 65535
|
||||
}
|
||||
|
||||
|
||||
# Utility methods:
|
||||
def calculate_color_temperature(r, g, b):
|
||||
"""Converts the raw R/G/B values to color temperature in degrees Kelvin."""
|
||||
# 1. Map RGB values to their XYZ counterparts.
|
||||
# Based on 6500K fluorescent, 3000K fluorescent
|
||||
# and 60W incandescent values for a wide range.
|
||||
# Note: Y = Illuminance or lux
|
||||
X = (-0.14282 * r) + (1.54924 * g) + (-0.95641 * b)
|
||||
Y = (-0.32466 * r) + (1.57837 * g) + (-0.73191 * b)
|
||||
Z = (-0.68202 * r) + (0.77073 * g) + ( 0.56332 * b)
|
||||
# Check for divide by 0 (total darkness) and return None.
|
||||
if (X + Y + Z) == 0:
|
||||
return None
|
||||
# 2. Calculate the chromaticity co-ordinates
|
||||
xc = (X) / (X + Y + Z)
|
||||
yc = (Y) / (X + Y + Z)
|
||||
# Check for divide by 0 again and return None.
|
||||
if (0.1858 - yc) == 0:
|
||||
return None
|
||||
# 3. Use McCamy's formula to determine the CCT
|
||||
n = (xc - 0.3320) / (0.1858 - yc)
|
||||
# Calculate the final CCT
|
||||
cct = (449.0 * (n ** 3.0)) + (3525.0 *(n ** 2.0)) + (6823.3 * n) + 5520.33
|
||||
return int(cct)
|
||||
|
||||
def calculate_lux(r, g, b):
|
||||
"""Converts the raw R/G/B values to luminosity in lux."""
|
||||
illuminance = (-0.32466 * r) + (1.57837 * g) + (-0.73191 * b)
|
||||
return int(illuminance)
|
||||
|
||||
|
||||
class TCS34725(object):
|
||||
"""TCS34725 color sensor."""
|
||||
|
||||
def __init__(self, integration_time=TCS34725_INTEGRATIONTIME_2_4MS,
|
||||
gain=TCS34725_GAIN_4X, address=TCS34725_ADDRESS, i2c=None, **kwargs):
|
||||
"""Initialize the TCS34725 sensor."""
|
||||
# Setup I2C interface for the device.
|
||||
if i2c is None:
|
||||
import Adafruit_GPIO.I2C as I2C
|
||||
i2c = I2C
|
||||
self._device = i2c.get_i2c_device(address, **kwargs)
|
||||
# Make sure we're connected to the sensor.
|
||||
chip_id = self._readU8(TCS34725_ID)
|
||||
if chip_id != 0x44:
|
||||
raise RuntimeError('Failed to read TCS34725 chip ID, check your wiring.')
|
||||
# Set default integration time and gain.
|
||||
self.set_integration_time(integration_time)
|
||||
self.set_gain(gain)
|
||||
# Enable the device (by default, the device is in power down mode on bootup).
|
||||
self.enable()
|
||||
|
||||
def _readU8(self, reg):
|
||||
"""Read an unsigned 8-bit register."""
|
||||
return self._device.readU8(TCS34725_COMMAND_BIT | reg)
|
||||
|
||||
def _readU16LE(self, reg):
|
||||
"""Read a 16-bit little endian register."""
|
||||
return self._device.readU16LE(TCS34725_COMMAND_BIT | reg)
|
||||
|
||||
def _write8(self, reg, value):
|
||||
"""Write a 8-bit value to a register."""
|
||||
self._device.write8(TCS34725_COMMAND_BIT | reg, value)
|
||||
|
||||
def enable(self):
|
||||
"""Enable the chip."""
|
||||
# Flip on the power and enable bits.
|
||||
self._write8(TCS34725_ENABLE, TCS34725_ENABLE_PON)
|
||||
time.sleep(0.01)
|
||||
self._write8(TCS34725_ENABLE, (TCS34725_ENABLE_PON | TCS34725_ENABLE_AEN))
|
||||
|
||||
def disable(self):
|
||||
"""Disable the chip (power down)."""
|
||||
# Flip off the power on and enable bits.
|
||||
reg = self._readU8(TCS34725_ENABLE)
|
||||
reg &= ~(TCS34725_ENABLE_PON | TCS34725_ENABLE_AEN)
|
||||
self._write8(TCS34725_ENABLE, reg)
|
||||
|
||||
def set_integration_time(self, integration_time):
|
||||
"""Sets the integration time for the TC34725. Provide one of these
|
||||
constants:
|
||||
- TCS34725_INTEGRATIONTIME_2_4MS = 2.4ms - 1 cycle - Max Count: 1024
|
||||
- TCS34725_INTEGRATIONTIME_24MS = 24ms - 10 cycles - Max Count: 10240
|
||||
- TCS34725_INTEGRATIONTIME_50MS = 50ms - 20 cycles - Max Count: 20480
|
||||
- TCS34725_INTEGRATIONTIME_101MS = 101ms - 42 cycles - Max Count: 43008
|
||||
- TCS34725_INTEGRATIONTIME_154MS = 154ms - 64 cycles - Max Count: 65535
|
||||
- TCS34725_INTEGRATIONTIME_700MS = 700ms - 256 cycles - Max Count: 65535
|
||||
"""
|
||||
self._integration_time = integration_time
|
||||
self._write8(TCS34725_ATIME, integration_time)
|
||||
|
||||
def get_integration_time(self):
|
||||
"""Return the current integration time value. This will be one of the
|
||||
constants specified in the set_integration_time doc string.
|
||||
"""
|
||||
return self._readU8(TCS34725_ATIME)
|
||||
|
||||
def set_gain(self, gain):
|
||||
"""Adjusts the gain on the TCS34725 (adjusts the sensitivity to light).
|
||||
Use one of the following constants:
|
||||
- TCS34725_GAIN_1X = No gain
|
||||
- TCS34725_GAIN_4X = 2x gain
|
||||
- TCS34725_GAIN_16X = 16x gain
|
||||
- TCS34725_GAIN_60X = 60x gain
|
||||
"""
|
||||
self._write8(TCS34725_CONTROL, gain)
|
||||
|
||||
def get_gain(self):
|
||||
"""Return the current gain value. This will be one of the constants
|
||||
specified in the set_gain doc string.
|
||||
"""
|
||||
return self._readU8(TCS34725_CONTROL)
|
||||
|
||||
def get_raw_data(self):
|
||||
"""Reads the raw red, green, blue and clear channel values. Will return
|
||||
a 4-tuple with the red, green, blue, clear color values (unsigned 16-bit
|
||||
numbers).
|
||||
"""
|
||||
# Read each color register.
|
||||
r = self._readU16LE(TCS34725_RDATAL)
|
||||
g = self._readU16LE(TCS34725_GDATAL)
|
||||
b = self._readU16LE(TCS34725_BDATAL)
|
||||
c = self._readU16LE(TCS34725_CDATAL)
|
||||
# Delay for the integration time to allow for next reading immediately.
|
||||
time.sleep(INTEGRATION_TIME_DELAY[self._integration_time])
|
||||
return (r, g, b, c)
|
||||
|
||||
def set_interrupt(self, enabled):
|
||||
"""Enable or disable interrupts by setting enabled to True or False."""
|
||||
enable_reg = self._readU8(TCS34725_ENABLE)
|
||||
if enabled:
|
||||
enable_reg |= TCS34725_ENABLE_AIEN
|
||||
else:
|
||||
enable_reg &= ~TCS34725_ENABLE_AIEN
|
||||
self._write8(TCS34725_ENABLE, enable_reg)
|
||||
time.sleep(1)
|
||||
|
||||
def clear_interrupt(self):
|
||||
"""Clear interrupt."""
|
||||
self._device.write8(0x66 & 0xff)
|
||||
|
||||
def set_interrupt_limits(self, low, high):
|
||||
"""Set the interrupt limits to provied unsigned 16-bit threshold values.
|
||||
"""
|
||||
self._device.write8(0x04, low & 0xFF)
|
||||
self._device.write8(0x05, low >> 8)
|
||||
self._device.write8(0x06, high & 0xFF)
|
||||
self._device.write8(0x07, high >> 8)
|
||||
1
Adafruit_TCS34725/__init__.py
Normal file
1
Adafruit_TCS34725/__init__.py
Normal file
|
|
@ -0,0 +1 @@
|
|||
from .TCS34725 import *
|
||||
61
examples/simpletest.py
Normal file
61
examples/simpletest.py
Normal file
|
|
@ -0,0 +1,61 @@
|
|||
# Simple demo of reading color data with the TCS34725 sensor.
|
||||
# Will read the color from the sensor and print it out along with lux and
|
||||
# color temperature.
|
||||
# Author: Tony DiCola
|
||||
# License: Public Domain
|
||||
import time
|
||||
|
||||
# Import the TCS34725 module.
|
||||
import Adafruit_TCS34725
|
||||
|
||||
|
||||
# Create a TCS34725 instance with default integration time (2.4ms) and gain (4x).
|
||||
tcs = Adafruit_TCS34725.TCS34725()
|
||||
|
||||
# You can also override the I2C device address and/or bus with parameters:
|
||||
#tcs = Adafruit_TCS34725.TCS34725(address=0x30, bus=2)
|
||||
|
||||
# Or you can change the integration time and/or gain:
|
||||
#tcs = Adafruit_TCS34725.TCS34725(integration_time=Adafruit_TCS34725.TCS34725_INTEGRATIONTIME_700MS,
|
||||
# gain=Adafruit_TCS34725.TCS34725_GAIN_60X)
|
||||
# Possible integration time values:
|
||||
# - TCS34725_INTEGRATIONTIME_2_4MS (2.4ms, default)
|
||||
# - TCS34725_INTEGRATIONTIME_24MS
|
||||
# - TCS34725_INTEGRATIONTIME_50MS
|
||||
# - TCS34725_INTEGRATIONTIME_101MS
|
||||
# - TCS34725_INTEGRATIONTIME_154MS
|
||||
# - TCS34725_INTEGRATIONTIME_700MS
|
||||
# Possible gain values:
|
||||
# - TCS34725_GAIN_1X
|
||||
# - TCS34725_GAIN_4X
|
||||
# - TCS34725_GAIN_16X
|
||||
# - TCS34725_GAIN_60X
|
||||
|
||||
# Disable interrupts (can enable them by passing true, see the set_interrupt_limits function too).
|
||||
tcs.set_interrupt(False)
|
||||
|
||||
# Read the R, G, B, C color data.
|
||||
r, g, b, c = tcs.get_raw_data()
|
||||
|
||||
# Calculate color temperature using utility functions. You might also want to
|
||||
# check out the colormath library for much more complete/accurate color functions.
|
||||
color_temp = Adafruit_TCS34725.calculate_color_temperature(r, g, b)
|
||||
|
||||
# Calculate lux with another utility function.
|
||||
lux = Adafruit_TCS34725.calculate_lux(r, g, b)
|
||||
|
||||
# Print out the values.
|
||||
print('Color: red={0} green={1} blue={2} clear={3}'.format(r, g, b, c))
|
||||
|
||||
# Print out color temperature.
|
||||
if color_temp is None:
|
||||
print('Too dark to determine color temperature!')
|
||||
else:
|
||||
print('Color Temperature: {0} K'.format(color_temp))
|
||||
|
||||
# Print out the lux.
|
||||
print('Luminosity: {0} lux'.format(lux))
|
||||
|
||||
# Enable interrupts and put the chip back to low power sleep/disabled.
|
||||
tcs.set_interrupt(True)
|
||||
tcs.disable()
|
||||
332
ez_setup.py
Normal file
332
ez_setup.py
Normal file
|
|
@ -0,0 +1,332 @@
|
|||
#!/usr/bin/env python
|
||||
"""Bootstrap setuptools installation
|
||||
|
||||
To use setuptools in your package's setup.py, include this
|
||||
file in the same directory and add this to the top of your setup.py::
|
||||
|
||||
from ez_setup import use_setuptools
|
||||
use_setuptools()
|
||||
|
||||
To require a specific version of setuptools, set a download
|
||||
mirror, or use an alternate download directory, simply supply
|
||||
the appropriate options to ``use_setuptools()``.
|
||||
|
||||
This file can also be run as a script to install or upgrade setuptools.
|
||||
"""
|
||||
import os
|
||||
import shutil
|
||||
import sys
|
||||
import tempfile
|
||||
import zipfile
|
||||
import optparse
|
||||
import subprocess
|
||||
import platform
|
||||
import textwrap
|
||||
import contextlib
|
||||
|
||||
from distutils import log
|
||||
|
||||
try:
|
||||
from site import USER_SITE
|
||||
except ImportError:
|
||||
USER_SITE = None
|
||||
|
||||
DEFAULT_VERSION = "3.5.1"
|
||||
DEFAULT_URL = "https://pypi.python.org/packages/source/s/setuptools/"
|
||||
|
||||
def _python_cmd(*args):
|
||||
"""
|
||||
Return True if the command succeeded.
|
||||
"""
|
||||
args = (sys.executable,) + args
|
||||
return subprocess.call(args) == 0
|
||||
|
||||
|
||||
def _install(archive_filename, install_args=()):
|
||||
with archive_context(archive_filename):
|
||||
# installing
|
||||
log.warn('Installing Setuptools')
|
||||
if not _python_cmd('setup.py', 'install', *install_args):
|
||||
log.warn('Something went wrong during the installation.')
|
||||
log.warn('See the error message above.')
|
||||
# exitcode will be 2
|
||||
return 2
|
||||
|
||||
|
||||
def _build_egg(egg, archive_filename, to_dir):
|
||||
with archive_context(archive_filename):
|
||||
# building an egg
|
||||
log.warn('Building a Setuptools egg in %s', to_dir)
|
||||
_python_cmd('setup.py', '-q', 'bdist_egg', '--dist-dir', to_dir)
|
||||
# returning the result
|
||||
log.warn(egg)
|
||||
if not os.path.exists(egg):
|
||||
raise IOError('Could not build the egg.')
|
||||
|
||||
|
||||
def get_zip_class():
|
||||
"""
|
||||
Supplement ZipFile class to support context manager for Python 2.6
|
||||
"""
|
||||
class ContextualZipFile(zipfile.ZipFile):
|
||||
def __enter__(self):
|
||||
return self
|
||||
def __exit__(self, type, value, traceback):
|
||||
self.close
|
||||
return zipfile.ZipFile if hasattr(zipfile.ZipFile, '__exit__') else \
|
||||
ContextualZipFile
|
||||
|
||||
|
||||
@contextlib.contextmanager
|
||||
def archive_context(filename):
|
||||
# extracting the archive
|
||||
tmpdir = tempfile.mkdtemp()
|
||||
log.warn('Extracting in %s', tmpdir)
|
||||
old_wd = os.getcwd()
|
||||
try:
|
||||
os.chdir(tmpdir)
|
||||
with get_zip_class()(filename) as archive:
|
||||
archive.extractall()
|
||||
|
||||
# going in the directory
|
||||
subdir = os.path.join(tmpdir, os.listdir(tmpdir)[0])
|
||||
os.chdir(subdir)
|
||||
log.warn('Now working in %s', subdir)
|
||||
yield
|
||||
|
||||
finally:
|
||||
os.chdir(old_wd)
|
||||
shutil.rmtree(tmpdir)
|
||||
|
||||
|
||||
def _do_download(version, download_base, to_dir, download_delay):
|
||||
egg = os.path.join(to_dir, 'setuptools-%s-py%d.%d.egg'
|
||||
% (version, sys.version_info[0], sys.version_info[1]))
|
||||
if not os.path.exists(egg):
|
||||
archive = download_setuptools(version, download_base,
|
||||
to_dir, download_delay)
|
||||
_build_egg(egg, archive, to_dir)
|
||||
sys.path.insert(0, egg)
|
||||
|
||||
# Remove previously-imported pkg_resources if present (see
|
||||
# https://bitbucket.org/pypa/setuptools/pull-request/7/ for details).
|
||||
if 'pkg_resources' in sys.modules:
|
||||
del sys.modules['pkg_resources']
|
||||
|
||||
import setuptools
|
||||
setuptools.bootstrap_install_from = egg
|
||||
|
||||
|
||||
def use_setuptools(version=DEFAULT_VERSION, download_base=DEFAULT_URL,
|
||||
to_dir=os.curdir, download_delay=15):
|
||||
to_dir = os.path.abspath(to_dir)
|
||||
rep_modules = 'pkg_resources', 'setuptools'
|
||||
imported = set(sys.modules).intersection(rep_modules)
|
||||
try:
|
||||
import pkg_resources
|
||||
except ImportError:
|
||||
return _do_download(version, download_base, to_dir, download_delay)
|
||||
try:
|
||||
pkg_resources.require("setuptools>=" + version)
|
||||
return
|
||||
except pkg_resources.DistributionNotFound:
|
||||
return _do_download(version, download_base, to_dir, download_delay)
|
||||
except pkg_resources.VersionConflict as VC_err:
|
||||
if imported:
|
||||
msg = textwrap.dedent("""
|
||||
The required version of setuptools (>={version}) is not available,
|
||||
and can't be installed while this script is running. Please
|
||||
install a more recent version first, using
|
||||
'easy_install -U setuptools'.
|
||||
|
||||
(Currently using {VC_err.args[0]!r})
|
||||
""").format(VC_err=VC_err, version=version)
|
||||
sys.stderr.write(msg)
|
||||
sys.exit(2)
|
||||
|
||||
# otherwise, reload ok
|
||||
del pkg_resources, sys.modules['pkg_resources']
|
||||
return _do_download(version, download_base, to_dir, download_delay)
|
||||
|
||||
def _clean_check(cmd, target):
|
||||
"""
|
||||
Run the command to download target. If the command fails, clean up before
|
||||
re-raising the error.
|
||||
"""
|
||||
try:
|
||||
subprocess.check_call(cmd)
|
||||
except subprocess.CalledProcessError:
|
||||
if os.access(target, os.F_OK):
|
||||
os.unlink(target)
|
||||
raise
|
||||
|
||||
def download_file_powershell(url, target):
|
||||
"""
|
||||
Download the file at url to target using Powershell (which will validate
|
||||
trust). Raise an exception if the command cannot complete.
|
||||
"""
|
||||
target = os.path.abspath(target)
|
||||
cmd = [
|
||||
'powershell',
|
||||
'-Command',
|
||||
"(new-object System.Net.WebClient).DownloadFile(%(url)r, %(target)r)" % vars(),
|
||||
]
|
||||
_clean_check(cmd, target)
|
||||
|
||||
def has_powershell():
|
||||
if platform.system() != 'Windows':
|
||||
return False
|
||||
cmd = ['powershell', '-Command', 'echo test']
|
||||
devnull = open(os.path.devnull, 'wb')
|
||||
try:
|
||||
try:
|
||||
subprocess.check_call(cmd, stdout=devnull, stderr=devnull)
|
||||
except Exception:
|
||||
return False
|
||||
finally:
|
||||
devnull.close()
|
||||
return True
|
||||
|
||||
download_file_powershell.viable = has_powershell
|
||||
|
||||
def download_file_curl(url, target):
|
||||
cmd = ['curl', url, '--silent', '--output', target]
|
||||
_clean_check(cmd, target)
|
||||
|
||||
def has_curl():
|
||||
cmd = ['curl', '--version']
|
||||
devnull = open(os.path.devnull, 'wb')
|
||||
try:
|
||||
try:
|
||||
subprocess.check_call(cmd, stdout=devnull, stderr=devnull)
|
||||
except Exception:
|
||||
return False
|
||||
finally:
|
||||
devnull.close()
|
||||
return True
|
||||
|
||||
download_file_curl.viable = has_curl
|
||||
|
||||
def download_file_wget(url, target):
|
||||
cmd = ['wget', url, '--quiet', '--output-document', target]
|
||||
_clean_check(cmd, target)
|
||||
|
||||
def has_wget():
|
||||
cmd = ['wget', '--version']
|
||||
devnull = open(os.path.devnull, 'wb')
|
||||
try:
|
||||
try:
|
||||
subprocess.check_call(cmd, stdout=devnull, stderr=devnull)
|
||||
except Exception:
|
||||
return False
|
||||
finally:
|
||||
devnull.close()
|
||||
return True
|
||||
|
||||
download_file_wget.viable = has_wget
|
||||
|
||||
def download_file_insecure(url, target):
|
||||
"""
|
||||
Use Python to download the file, even though it cannot authenticate the
|
||||
connection.
|
||||
"""
|
||||
try:
|
||||
from urllib.request import urlopen
|
||||
except ImportError:
|
||||
from urllib2 import urlopen
|
||||
src = dst = None
|
||||
try:
|
||||
src = urlopen(url)
|
||||
# Read/write all in one block, so we don't create a corrupt file
|
||||
# if the download is interrupted.
|
||||
data = src.read()
|
||||
dst = open(target, "wb")
|
||||
dst.write(data)
|
||||
finally:
|
||||
if src:
|
||||
src.close()
|
||||
if dst:
|
||||
dst.close()
|
||||
|
||||
download_file_insecure.viable = lambda: True
|
||||
|
||||
def get_best_downloader():
|
||||
downloaders = [
|
||||
download_file_powershell,
|
||||
download_file_curl,
|
||||
download_file_wget,
|
||||
download_file_insecure,
|
||||
]
|
||||
|
||||
for dl in downloaders:
|
||||
if dl.viable():
|
||||
return dl
|
||||
|
||||
def download_setuptools(version=DEFAULT_VERSION, download_base=DEFAULT_URL,
|
||||
to_dir=os.curdir, delay=15, downloader_factory=get_best_downloader):
|
||||
"""
|
||||
Download setuptools from a specified location and return its filename
|
||||
|
||||
`version` should be a valid setuptools version number that is available
|
||||
as an egg for download under the `download_base` URL (which should end
|
||||
with a '/'). `to_dir` is the directory where the egg will be downloaded.
|
||||
`delay` is the number of seconds to pause before an actual download
|
||||
attempt.
|
||||
|
||||
``downloader_factory`` should be a function taking no arguments and
|
||||
returning a function for downloading a URL to a target.
|
||||
"""
|
||||
# making sure we use the absolute path
|
||||
to_dir = os.path.abspath(to_dir)
|
||||
zip_name = "setuptools-%s.zip" % version
|
||||
url = download_base + zip_name
|
||||
saveto = os.path.join(to_dir, zip_name)
|
||||
if not os.path.exists(saveto): # Avoid repeated downloads
|
||||
log.warn("Downloading %s", url)
|
||||
downloader = downloader_factory()
|
||||
downloader(url, saveto)
|
||||
return os.path.realpath(saveto)
|
||||
|
||||
def _build_install_args(options):
|
||||
"""
|
||||
Build the arguments to 'python setup.py install' on the setuptools package
|
||||
"""
|
||||
return ['--user'] if options.user_install else []
|
||||
|
||||
def _parse_args():
|
||||
"""
|
||||
Parse the command line for options
|
||||
"""
|
||||
parser = optparse.OptionParser()
|
||||
parser.add_option(
|
||||
'--user', dest='user_install', action='store_true', default=False,
|
||||
help='install in user site package (requires Python 2.6 or later)')
|
||||
parser.add_option(
|
||||
'--download-base', dest='download_base', metavar="URL",
|
||||
default=DEFAULT_URL,
|
||||
help='alternative URL from where to download the setuptools package')
|
||||
parser.add_option(
|
||||
'--insecure', dest='downloader_factory', action='store_const',
|
||||
const=lambda: download_file_insecure, default=get_best_downloader,
|
||||
help='Use internal, non-validating downloader'
|
||||
)
|
||||
parser.add_option(
|
||||
'--version', help="Specify which version to download",
|
||||
default=DEFAULT_VERSION,
|
||||
)
|
||||
options, args = parser.parse_args()
|
||||
# positional arguments are ignored
|
||||
return options
|
||||
|
||||
def main():
|
||||
"""Install or upgrade setuptools and EasyInstall"""
|
||||
options = _parse_args()
|
||||
archive = download_setuptools(
|
||||
version=options.version,
|
||||
download_base=options.download_base,
|
||||
downloader_factory=options.downloader_factory,
|
||||
)
|
||||
return _install(archive, _build_install_args(options))
|
||||
|
||||
if __name__ == '__main__':
|
||||
sys.exit(main())
|
||||
24
setup.py
Normal file
24
setup.py
Normal file
|
|
@ -0,0 +1,24 @@
|
|||
from ez_setup import use_setuptools
|
||||
use_setuptools()
|
||||
from setuptools import setup, find_packages
|
||||
|
||||
classifiers = ['Development Status :: 4 - Beta',
|
||||
'Operating System :: POSIX :: Linux',
|
||||
'License :: OSI Approved :: MIT License',
|
||||
'Intended Audience :: Developers',
|
||||
'Programming Language :: Python :: 2.7',
|
||||
'Programming Language :: Python :: 3',
|
||||
'Topic :: Software Development',
|
||||
'Topic :: System :: Hardware']
|
||||
|
||||
setup(name = 'Adafruit_TCS34725',
|
||||
version = '1.0.0',
|
||||
author = 'Tony DiCola',
|
||||
author_email = 'tdicola@adafruit.com',
|
||||
description = 'Python code to use the TCS34725 color sensor with the Raspberry Pi & BeagleBone Black.',
|
||||
license = 'MIT',
|
||||
classifiers = classifiers,
|
||||
url = 'https://github.com/adafruit/Adafruit_Python_TCS34725/',
|
||||
dependency_links = ['https://github.com/adafruit/Adafruit_Python_GPIO/tarball/master#egg=Adafruit-GPIO-0.6.5'],
|
||||
install_requires = ['Adafruit-GPIO>=0.6.5'],
|
||||
packages = find_packages())
|
||||
Loading…
Reference in a new issue