Enum

.travis.yml

@@ -1,6 +1,8 @@
 language: python
 python: 2.7
-install: pip install -qq flake8 pytest pytest-cov tox
+install:
+  - pip install -qq flake8 pytest pytest-cov tox
+  - python setup.py install
 script:
   - flake8 --version
   - flake8 --ignore=E501 pingo

README.rst

@@ -46,11 +46,12 @@ Once you have a board instance, it's possible to access its pins through the ``b
 .. code-block:: python
 
     import pingo
+    from pingo import State, Mode
     from time import sleep
 
     board = pingo.detect.get_board()
     led_pin = board.pins[13]
-    led_pin.mode = pingo.OUT
+    led_pin.mode = Mode.OUT
 
     while True:
         led_pin.hi()

pingo/__init__.py

@@ -1,32 +1,32 @@
+from __future__ import absolute_import
+
 # api
-from board import ANALOG  # noqa
-from board import IN  # noqa
-from board import OUT  # noqa
-from board import PWM  # noqa
-from board import HIGH  # noqa
-from board import LOW  # noqa
-from board import ModeNotSuported  # noqa
-from board import WrongPinMode  # noqa
-from board import PwmOutputCapable  # noqa
-from board import AnalogInputCapable  # noqa
-from board import Board  # noqa
-from board import PwmPin  # noqa
-from board import AnalogPin  # noqa
-from board import DigitalPin  # noqa
-from board import GroundPin  # noqa
-from board import Pin  # noqa
-from board import VccPin  # noqa
-import parts  # noqa
+from .board import State  # noqa
+from .board import Mode  # noqa
+from .board import ModeNotSuported  # noqa
+from .board import WrongPinMode  # noqa
+from .board import PwmOutputCapable  # noqa
+from .board import AnalogInputCapable  # noqa
+from .board import Board  # noqa
+from .board import PwmPin  # noqa
+from .board import AnalogPin  # noqa
+from .board import DigitalPin  # noqa
+from .board import GroundPin  # noqa
+from .board import Pin  # noqa
+from .board import VccPin  # noqa
+import pingo.parts  # import * # noqa
+
 
 # boards
-import rpi  # noqa
-import ghost  # noqa
-import intel  # noqa
-import udoo  # noqa
-import pcduino  # noqa
-import arduino  # noqa
-import bbb  # noqa
+import pingo.rpi  # noqa
+import pingo.ghost  # noqa
+import pingo.intel  # noqa
+import pingo.udoo  # noqa
+import pingo.pcduino  # noqa
+import pingo.arduino  # noqa
+import pingo.bbb  # noqa
+
 
 # resources
-import detect  # noqa
-import test  # noqa
+import pingo.detect  # noqa
+import pingo.test  # noqa

pingo/arduino/__init__.py

@@ -1,4 +1,4 @@
-from firmata import ArduinoFirmata  # noqa
-from firmata import get_arduino  # noqa
-from pyun import YunBridge  # noqa
-from pyun import ArduinoYun  # noqa
+from .firmata import ArduinoFirmata  # noqa
+from .firmata import get_arduino  # noqa
+from .pyun import YunBridge  # noqa
+from .pyun import ArduinoYun  # noqa

pingo/arduino/firmata.py

@@ -9,9 +9,10 @@
 
 import pingo
 from pingo.board import Board, DigitalPin, AnalogPin, PwmPin
+from pingo.board import State, Mode
 from pingo.board import AnalogInputCapable, PwmOutputCapable
 from pingo.detect import detect
-from util_firmata import pin_list_to_board_dict
+from .util_firmata import pin_list_to_board_dict
 
 PyMata = None
 
@@ -20,14 +21,14 @@
     True: 1,
     0: 0,
     1: 1,
-    pingo.LOW: 0,
-    pingo.HIGH: 1,
+    State.LOW: 0,
+    State.HIGH: 1,
 }
 
 # TODO: PyMata suports Input, Output, PWM, Servo, Encoder and Tone
 PIN_MODES = {
-    pingo.IN: 0,
-    pingo.OUT: 1,
+    Mode.IN: 0,
+    Mode.OUT: 1,
 }
 
 VERBOSE = False

pingo/arduino/pyun.py

@@ -47,7 +47,7 @@ def makeURL(self, command, pin, *args):
         else:
             url = self.base_url + '%s/%d' % (command, pin)
         if self.verbose:
-            print '[YunBridge] url: ', url
+            print('[YunBridge] url: %s' % url)
         return url
 
     def get(self, command, pin, *args):

pingo/arduino/test_util_firmata.py

@@ -1,6 +1,6 @@
 import unittest
 
-from util_firmata import pin_list_to_board_dict
+from .util_firmata import pin_list_to_board_dict
 
 
 class FirmataCapabilityDetect(unittest.TestCase):
@@ -43,12 +43,8 @@ def test_capability_response(self):
             0x7F,  # END_SYSEX (Pin delimiter)
         ]
 
-        data_arduino = list(
-            # [0x6C]  # CAPABILITY_RESPONSE
-            unavailible_pin
-            + digital_pin
-            + analog_pin
-        )
+        data_arduino = list(unavailible_pin + digital_pin + analog_pin)
+        # [0x6C]  # CAPABILITY_RESPONSE
 
         pinmap = pin_list_to_board_dict(data_arduino)
         for key in test_layout.keys():

pingo/bbb/__init__.py

@@ -1 +1 @@
-from bbb import BeagleBoneBlack  # noqa
+from .bbb import BeagleBoneBlack  # noqa

pingo/board.py

@@ -5,15 +5,11 @@
 from abc import ABCMeta, abstractmethod
 
 from .util import StrKeyDict
+from enum import Enum
 
-HIGH = 'HIGH'
-LOW = 'LOW'
+State = Enum('State', 'LOW HIGH')
 
-# TODO: 4 states implementation: IN, OUT, ANALOG, PWM
-IN = 'IN'
-OUT = 'OUT'
-ANALOG = 'ANALOG'
-PWM = 'PWM'
+Mode = Enum('Mode', 'IN OUT ANALOG PWM')
 
 
 class WrongPinMode(Exception):
@@ -249,24 +245,28 @@ def __repr__(self):
 
     @property
     def mode(self):
-        """[property] Get/set pin mode to ``pingo.IN``, ``pingo.OUT``
-         ``pingo.ANALOG`` or ``pingo.PWM``"""
+        """[property] Get/set pin mode to ``Mode.IN``, ``Mode.OUT``
+         ``Mode.ANALOG`` or ``Mode.PWM``"""
         return self._mode
 
     @mode.setter
     def mode(self, value):
         if value not in self.suported_modes:
             raise ModeNotSuported()
 
-        if value in [IN, OUT]:
+        if value in [Mode.IN, Mode.OUT]:
             self.board._set_digital_mode(self, value)
-        elif value == ANALOG:
+        elif value == Mode.ANALOG:
             self.board._set_analog_mode(self, value)
-        elif value == PWM:
+        elif value == Mode.PWM:
             self.board._set_pwm_mode(self, value)
 
         self._mode = value
 
+    @property
+    def is_analog(self):
+        return issubclass(type(self), AnalogPin)
+
 
 class DigitalPin(Pin):
     """Defines common interface for all digital pins.
@@ -279,60 +279,60 @@ class DigitalPin(Pin):
     because pins delegate all board-dependent behavior to the board.
     """
 
-    suported_modes = [IN, OUT]
+    suported_modes = [Mode.IN, Mode.OUT]
 
     def __init__(self, board, location, gpio_id=None):
         Pin.__init__(self, board, location, gpio_id)
         self._state = None
 
     @property
     def state(self):
-        """[property] Get/set pin state to ``pingo.HIGH`` or ``pingo.LOW``"""
-        if self.mode not in [IN, OUT]:
+        """[property] Get/set pin state to ``State.HIGH`` or ``State.LOW``"""
+        if self.mode not in [Mode.IN, Mode.OUT]:
             raise WrongPinMode()
 
-        if self.mode == IN:
+        if self.mode == Mode.IN:
             self._state = self.board._get_pin_state(self)
 
         return self._state
 
     @state.setter
     def state(self, value):
-        if self.mode != OUT:
+        if self.mode != Mode.OUT:
             raise WrongPinMode()
 
         self.board._set_pin_state(self, value)
         self._state = value
 
     def low(self):
-        """Set voltage of pin to ``pingo.LOW`` (GND)."""
-        self.state = LOW
+        """Set voltage of pin to ``State.LOW`` (GND)."""
+        self.state = State.LOW
 
     lo = low  # shortcut for interactive use
 
     def high(self):
         """Set state of the pin to ``pingo.HIGH`` (Vcc)."""
-        self.state = HIGH
+        self.state = State.HIGH
 
     hi = high  # shortcut for interactive use
 
     def toggle(self):
         """Change state of the pin."""
-        self.state = HIGH if self.state == LOW else LOW
+        self.state = State.HIGH if self.state == State.LOW else State.LOW
 
     def pulse(self):
         """Generate a pulse in state of the pin."""
-        if self.state == LOW:
-            self.state = HIGH
-            self.state = LOW
+        if self.state == State.LOW:
+            self.state = State.HIGH
+            self.state = State.LOW
         else:
-            self.state = LOW
-            self.state = HIGH
+            self.state = State.LOW
+            self.state = State.HIGH
 
 
 class PwmPin(DigitalPin):
 
-    suported_modes = [IN, OUT, PWM]
+    suported_modes = [Mode.IN, Mode.OUT, Mode.PWM]
 
     def __init__(self, board, location, gpio_id=None, frequency=None):
         DigitalPin.__init__(self, board, location, gpio_id)
@@ -344,13 +344,13 @@ def __init__(self, board, location, gpio_id=None, frequency=None):
 
     @property
     def value(self):
-        if self.mode != PWM:
+        if self.mode != Mode.PWM:
             raise WrongPinMode()
         return self.board._get_pwm_duty_cycle(self)
 
     @value.setter
     def value(self, value):
-        if self.mode != PWM:
+        if self.mode != Mode.PWM:
             raise WrongPinMode()
         if not 0.0 <= value <= 100.0:
             raise ArgumentOutOfRange()
@@ -359,13 +359,13 @@ def value(self, value):
 
     @property
     def frequency(self):
-        if self.mode != PWM:
+        if self.mode != Mode.PWM:
             raise WrongPinMode()
         return self.board._get_pwm_frequency(self)
 
     @frequency.setter
     def frequency(self, new_frequency):
-        if self.mode != PWM:
+        if self.mode != Mode.PWM:
             raise WrongPinMode()
         if new_frequency <= 0.0:
             raise ArgumentOutOfRange()
@@ -382,7 +382,7 @@ class AnalogPin(Pin):
     This pin type supports read operations only.
     """
 
-    suported_modes = [IN, ANALOG]
+    suported_modes = [Mode.IN, Mode.ANALOG]
 
     def __init__(self, board, location, resolution, gpio_id=None):
         """

pingo/compat.py

@@ -0,0 +1,28 @@
+import sys
+
+PY2 = sys.version_info[0] == 2
+PY3 = (sys.version_info[0] >= 3)
+
+
+def iteritems(obj, **kwargs):
+    """replacement for six's iteritems for Python2/3 compat
+       uses 'iteritems' if available and otherwise uses 'items'.
+       Passes kwargs to method.
+    """
+    func = getattr(obj, "iteritems", None)
+    if not func:
+        func = obj.items
+    return func(**kwargs)
+
+
+def iterkeys(obj, **kwargs):
+    func = getattr(obj, "iterkeys", None)
+    if not func:
+        func = obj.keys
+    return func(**kwargs)
+
+
+if PY2:
+    from UserDict import UserDict  # noqa
+else:
+    from collections import UserDict  # noqa

pingo/detect/__init__.py

@@ -1,4 +1,4 @@
-from detect import get_board
+from .detect import get_board
 
 
 def has_module(module_name):

pingo/detect/detect.py

@@ -36,8 +36,7 @@ def _find_arduino_dev(system):
             return os.path.join(os.path.sep, 'dev', devices[0])
 
     elif system == 'Darwin':
-        devices = (glob.glob('/dev/tty.usbmodem*')
-                   + glob.glob('/dev/tty.usbserial*'))
+        devices = (glob.glob('/dev/tty.usbmodem*') + glob.glob('/dev/tty.usbserial*'))
         if len(devices) == 1:
             return os.path.join(os.path.sep, 'dev', devices[0])
     return False

pingo/detect/test_detect.py

@@ -2,7 +2,6 @@
 import platform
 
 import pingo
-import detect
 
 
 class DetectBasics(unittest.TestCase):
@@ -13,7 +12,7 @@ def test_board(self):
 
     @unittest.skipIf(not platform.system() == 'Linux', 'Not Linux')
     def test_read_cpu_info(self):
-        info = detect._read_cpu_info()
+        info = pingo.detect.detect._read_cpu_info()
         assert isinstance(info, dict)
 
 if __name__ == '__main__':