lint

Author: BlitzCityDIY

Adabot_RP2040_Prop-Maker_Feather/code.py

@@ -3,7 +3,6 @@
 # SPDX-License-Identifier: MIT
 
 import os
-import time
 import random
 import board
 import audiocore
@@ -12,7 +11,6 @@
 import pwmio
 import neopixel
 import adafruit_lis3dh
-from random import randint
 from adafruit_ticks import ticks_ms, ticks_add, ticks_diff
 from digitalio import DigitalInOut, Direction, Pull
 from adafruit_motor import servo
@@ -32,7 +30,6 @@
 lis3dh = adafruit_lis3dh.LIS3DH_I2C(i2c, int1=int1)
 lis3dh.range = adafruit_lis3dh.RANGE_2_G
 
-# Setup button switch
 switch = DigitalInOut(board.EXTERNAL_BUTTON)
 switch.direction = Direction.INPUT
 switch.pull = Pull.UP
@@ -56,45 +53,43 @@
 mixer.voice[0].play(wave)
 
 def open_audio(num):
-    wav_filename = wavs[num]
-    wav_file = open(wav_filename, "rb")
-    wave = audiocore.WaveFile(wav_file)
-    return wave
+    n = wavs[num]
+    f = open(n, "rb")
+    w = audiocore.WaveFile(f)
+    return w
 
 PIXEL_PIN = board.EXTERNAL_NEOPIXELS
 SERVO_PIN = board.EXTERNAL_SERVO
 NUM_PIXELS = 8
 ORDER = neopixel.GRB
 BRIGHTNESS = 0.3
 
-# Initialize servo
 PWM = pwmio.PWMOut(SERVO_PIN, duty_cycle=2 ** 15, frequency=50)
 SERVO = servo.Servo(PWM)
 
-# Initialize NeoPixels and animations
 pixel = neopixel.NeoPixel(board.NEOPIXEL, 1)
 pixel.brightness = 1
 
 PIXELS = neopixel.NeoPixel(PIXEL_PIN, NUM_PIXELS, auto_write=False,
                            pixel_order=ORDER)
-LARSON = Comet(PIXELS, bounce=True, speed=0.07,
+LARSON = Comet(PIXELS, bounce=True, speed=0.7,
                tail_length=NUM_PIXELS//2,
-               color=(BLUE[0] * BRIGHTNESS,  # This is a little faster than
-                      BLUE[1] * BRIGHTNESS,  # using the NeoPixel brightness
-                      BLUE[2] * BRIGHTNESS)) # setting.
+               color=(BLUE[0] * BRIGHTNESS,
+                      BLUE[1] * BRIGHTNESS,
+                      BLUE[2] * BRIGHTNESS))
 pulse = Pulse(PIXELS, speed=0.05,
-                color=(BLUE[0] * BRIGHTNESS,  # This is a little faster than
-                      BLUE[1] * BRIGHTNESS,  # using the NeoPixel brightness
-                      BLUE[2] * BRIGHTNESS), period=3)
+                color=(BLUE[0] * BRIGHTNESS,
+                       BLUE[1] * BRIGHTNESS,
+                       BLUE[2] * BRIGHTNESS), period=3)
 sparkle = Sparkle(PIXELS, speed=0.2,
-                color=(RED[0] * BRIGHTNESS,  # This is a little faster than
-                      RED[1] * BRIGHTNESS,  # using the NeoPixel brightness
-                      RED[2] * BRIGHTNESS), num_sparkles=10)
+                color=(RED[0] * BRIGHTNESS,
+                       RED[1] * BRIGHTNESS,
+                       RED[2] * BRIGHTNESS), num_sparkles=10)
 
 SERVO.angle = POSITION = NEXT_POSITION = 90
-MOVING = False                # Initial state = paused
-START_TIME = ticks_ms() # Initial time
-DURATION = 1000                # Hold initial position for 1 sec
+MOVING = False
+START_TIME = ticks_ms()
+DURATION = 1000
 
 adabot_talk = False
 
@@ -106,43 +101,37 @@ def open_audio(num):
 while mixer.playing:
     LARSON.animate()
 
-while True: # Loop forever...
+while True:
     if ticks_diff(ticks_ms(), clock) >= prop_time:
         x, y, z = [
         value / adafruit_lis3dh.STANDARD_GRAVITY for value in lis3dh.acceleration
         ]
         if z > 0.9:
             adabot_nap = True
             SERVO.angle = POSITION = NEXT_POSITION = 90
-            pulse.color=(BLUE[0] * BRIGHTNESS,  # This is a little faster than
-                          BLUE[1] * BRIGHTNESS,  # using the NeoPixel brightness
-                          BLUE[2] * BRIGHTNESS)
         else:
             adabot_nap = False
-            LARSON.color=(BLUE[0] * BRIGHTNESS,  # This is a little faster than
-                          BLUE[1] * BRIGHTNESS,  # using the NeoPixel brightness
-                          BLUE[2] * BRIGHTNESS)
         if not adabot_nap:
             MOVING = not MOVING
-            if MOVING:             # Switching from paused to moving
+            if MOVING:
                 POSITION = NEXT_POSITION
-                while abs(POSITION - NEXT_POSITION) < 10:  # Min +/- 10 degrees
-                    NEXT_POSITION = random.uniform(0, 180) # Try, try again
+                while abs(POSITION - NEXT_POSITION) < 10:
+                    NEXT_POSITION = random.uniform(0, 180)
                 DURATION = 0.2 + 0.6 * abs(POSITION - NEXT_POSITION) / 180
-            else:                  # Switching from moving to paused
-                SERVO.angle = NEXT_POSITION         # Move to end of sweep
-                DURATION = random.uniform(0.5, 2.5) # Pause time
+            else:
+                SERVO.angle = NEXT_POSITION
+                DURATION = random.uniform(0.5, 2.5)
         clock = ticks_add(clock, prop_time)
     if MOVING:
-        FRACTION = 0.0 / DURATION                        # Linear 0 to 1
-        FRACTION = (3 * FRACTION ** 2) - (2 * FRACTION ** 3) # Ease in/out
+        FRACTION = 0.0 / DURATION
+        FRACTION = (3 * FRACTION ** 2) - (2 * FRACTION ** 3)
         SERVO.angle = POSITION + (NEXT_POSITION - POSITION) * FRACTION
     if adabot_talk:
-        wave = open_audio(randint(1, 17))
+        wave = open_audio(random.randint(1, 7))
         mixer.voice[0].play(wave)
         while mixer.playing:
             sparkle.animate()
-        else:
+        if not mixer.playing:
             adabot_talk = False
             PIXELS.fill(BLACK)
             PIXELS.show()