Massive pylinting session and added Github Actions

[Adafruit_Blinka-hackapet.git] / src / adafruit_blinka / microcontroller / bcm283x / neopixel.py

diff --git a/src/adafruit_blinka/microcontroller/bcm283x/neopixel.py b/src/adafruit_blinka/microcontroller/bcm283x/neopixel.py
index d744b255f25bee7e3657354c75a9050609a1dd83..813a51a092706d6d54b4f86833dce9c03c4171b6 100644 (file)
--- a/src/adafruit_blinka/microcontroller/bcm283x/neopixel.py
+++ b/src/adafruit_blinka/microcontroller/bcm283x/neopixel.py
@@ -1,22 +1,26 @@
+"""BCM283x NeoPixel Driver Class"""
 import time
-import math
-import _rpi_ws281x as ws
 import atexit
+import _rpi_ws281x as ws
 
 # LED configuration.
-LED_CHANNEL    = 0
-LED_FREQ_HZ    = 800000     # Frequency of the LED signal.  We only support 800KHz
-LED_DMA_NUM    = 10         # DMA channel to use, can be 0-14.
-LED_BRIGHTNESS = 255        # We manage the brightness in the neopixel library
-LED_INVERT     = 0          # We don't support inverted logic
-LED_STRIP      = None # We manage the color order within the neopixel library
+# pylint: disable=redefined-outer-name,too-many-branches,too-many-statements
+# pylint: disable=global-statement,protected-access
+LED_CHANNEL = 0
+LED_FREQ_HZ = 800000  # Frequency of the LED signal.  We only support 800KHz
+LED_DMA_NUM = 10  # DMA channel to use, can be 0-14.
+LED_BRIGHTNESS = 255  # We manage the brightness in the neopixel library
+LED_INVERT = 0  # We don't support inverted logic
+LED_STRIP = None  # We manage the color order within the neopixel library
 
 # a 'static' object that we will use to manage our PWM DMA channel
 # we only support one LED strip per raspi
 _led_strip = None
 
+
 def neopixel_write(gpio, buf):
-    global _led_strip # we'll have one strip we init if its not at first
+    """NeoPixel Writing Function"""
+    global _led_strip  # we'll have one strip we init if its not at first
 
     if _led_strip is None:
         # Create a ws2811_t structure from the LED configuration.
@@ -40,14 +44,16 @@ def neopixel_write(gpio, buf):
         if len(buf) % 3 == 0:
             # most common, divisible by 3 is likely RGB
             LED_STRIP = ws.WS2811_STRIP_RGB
-            count = len(buf)//3
+            count = len(buf) // 3
         elif len(buf) % 4 == 0:
             LED_STRIP = ws.SK6812_STRIP_RGBW
-            count = len(buf)//4
+            count = len(buf) // 4
         else:
             raise RuntimeError("We only support 3 or 4 bytes-per-pixel")
 
-        ws.ws2811_channel_t_count_set(channel, count) # we manage 4 vs 3 bytes in the library
+        ws.ws2811_channel_t_count_set(
+            channel, count
+        )  # we manage 4 vs 3 bytes in the library
         ws.ws2811_channel_t_gpionum_set(channel, gpio._pin.id)
         ws.ws2811_channel_t_invert_set(channel, LED_INVERT)
         ws.ws2811_channel_t_brightness_set(channel, LED_BRIGHTNESS)
@@ -60,9 +66,13 @@ def neopixel_write(gpio, buf):
         resp = ws.ws2811_init(_led_strip)
         if resp != ws.WS2811_SUCCESS:
             if resp == -5:
-                raise RuntimeError("NeoPixel support requires running with sudo, please try again!")
+                raise RuntimeError(
+                    "NeoPixel support requires running with sudo, please try again!"
+                )
             message = ws.ws2811_get_return_t_str(resp)
-            raise RuntimeError('ws2811_init failed with code {0} ({1})'.format(resp, message))
+            raise RuntimeError(
+                "ws2811_init failed with code {0} ({1})".format(resp, message)
+            )
         atexit.register(neopixel_cleanup)
 
     channel = ws.ws2811_channel_get(_led_strip, LED_CHANNEL)
@@ -75,24 +85,27 @@ def neopixel_write(gpio, buf):
         bpp = 4
     # assign all colors!
     for i in range(len(buf) // bpp):
-        r = buf[bpp*i]
-        g = buf[bpp*i+1]
-        b = buf[bpp*i+2]
+        r = buf[bpp * i]
+        g = buf[bpp * i + 1]
+        b = buf[bpp * i + 2]
         if bpp == 3:
             pixel = (r << 16) | (g << 8) | b
         else:
-            w = buf[bpp*i+3]
+            w = buf[bpp * i + 3]
             pixel = (w << 24) | (r << 16) | (g << 8) | b
         ws.ws2811_led_set(channel, i, pixel)
 
     resp = ws.ws2811_render(_led_strip)
     if resp != ws.WS2811_SUCCESS:
         message = ws.ws2811_get_return_t_str(resp)
-        raise RuntimeError('ws2811_render failed with code {0} ({1})'.format(resp, message))
-    time.sleep(0.001 * ((len(buf)//100)+1))  # about 1ms per 100 bytes
+        raise RuntimeError(
+            "ws2811_render failed with code {0} ({1})".format(resp, message)
+        )
+    time.sleep(0.001 * ((len(buf) // 100) + 1))  # about 1ms per 100 bytes
 
 
 def neopixel_cleanup():
+    """Cleanup when we're done"""
     global _led_strip
 
     if _led_strip is not None: