diff --git a/main.py b/main.py
index 5d3c51c..2ac7c99 100644
--- a/main.py
+++ b/main.py
@@ -4,19 +4,12 @@ from datetime import datetime
try:
from rpi_ws281x import *
import RPi.GPIO as GPIO
- import socket
+ from flask import Flask
+ import json
except ImportError:
print('Some modules are missing. Try to install them with "pip3 install -r requirements.txt"')
exit()
-led_power = True
-idle_mode = 1
-global power_button_pressed
-power_button_pressed = False
-
-SOCKET_BIND_IP = '0.0.0.0'
-SOCKET_BIND_PORT = 5760
-
# LED strip configuration:
LED_COUNT = 60 # Number of LED pixels.
LED_PIN = 18 # GPIO pin connected to the pixels (18 uses PWM!).
@@ -27,6 +20,9 @@ LED_BRIGHTNESS = 255 # Set to 0 for darkest and 255 for brightest
LED_INVERT = False # True to invert the signal (when using NPN transistor level shift)
LED_CHANNEL = 0 # set to '1' for GPIOs 13, 19, 41, 45 or 53
+mode = 0
+power = True
+
def debug_print(message: str):
now = datetime.now()
current_time = now.strftime("%H:%M:%S")
@@ -44,9 +40,7 @@ def wheel(pos):
def rainbowCycle(strip, wait_ms=5, iterations=10):
for j in range(256*iterations):
- if power_button_pressed == True:
- break
- elif idle_mode > 1:
+ if mode > 0:
break
for i in range(strip.numPixels()):
strip.setPixelColor(i, wheel((int(i * 256 / strip.numPixels()) + j) & 255))
@@ -59,67 +53,98 @@ def setColor(strip, color, wait_ms=10):
strip.show()
time.sleep(wait_ms/1000.0)
-def blink(strip, color1, color2):
- for i in range(strip.numPixels()):
- strip.setPixelColor(i, color1)
- strip.show()
- time.sleep(1)
- for i in range(strip.numPixels()):
- strip.setPixelColor(i, color2)
- strip.show()
- time.sleep(1)
-
-def change_idle_mode(channel):
- global idle_mode
- idle_mode += 1
- if idle_mode == 8:
- idle_mode = 1
- debug_print('Change Idlemode executed')
-
-def power_toggle(channel):
- if led_power == True:
- led_power = False
- debug_print('LED Power OFF')
- else:
- led_power = True
- debug_print('LED Power ON')
-
if __name__ == '__main__':
+ app = Flask(__name__)
+ @app.route('/')
+ def index():
+ return '''
+
+
+
+
+
+
+ Server Rack LED stripe
+
+
+
+ My Server Rack LED stripe control API
+
+ Please use the following links to access the API:
+
+
+ How to use the API:
+
+ /api/send: Send a SMSlist of parameters:
+
+ number: The phone number of the recipientmessage: The message to send
+ Example:
+
+ /api/send?number=01234567890&message="Hello World"
+
+
+
+ '''
+
+ @app.route('/mode', methods=['GET'])
+ def mode():
+ args = request.args
+ mode = args.get('mode')
+ color = args.get('color')
+ if mode is None or color is None:
+ return 'No mode specified'
+ elif mode == "" or color == "":
+ return 'No mode specified'
+ elif mode == "0":
+ mode = 0
+ return 'Rainbow Cycle'
+ elif mode == "1":
+ mode = 1
+ return color
+ return 'Set Color'
+
+ @app.route('/power', methods=['GET'])
+ def power():
+ if power == True:
+ power = False
+ return "Power off"
+ else:
+ power = True
+ return "Power on"
+
strip = Adafruit_NeoPixel(LED_COUNT, LED_PIN, LED_FREQ_HZ, LED_DMA, LED_INVERT, LED_BRIGHTNESS, LED_CHANNEL)
strip.begin()
- GPIO.setmode(GPIO.BOARD)
- GPIO.setup(13, GPIO.IN, pull_up_down=GPIO.PUD_UP)
- GPIO.setup(19, GPIO.IN, pull_up_down=GPIO.PUD_UP)
- GPIO.add_event_detect(13, GPIO.RISING, callback=power_toggle, bouncetime=300)
- GPIO.add_event_detect(19, GPIO.RISING, callback=change_idle_mode, bouncetime=300)
-
- # with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
- # s.bind((SOCKET_BIND_IP, SOCKET_BIND_PORT))
- # s.listen()
- # conn, addr = s.accept()
-
try:
while True:
- # socket_data = conn.recv(1024)
- # print(socket_data)
- # print(type(socket_data))
- if led_power == True:
- if idle_mode == 1:
- rainbowCycle(strip)
- elif idle_mode == 2:
- setColor(strip, Color(255,0,0))
- elif idle_mode == 3:
- setColor(strip, Color(0,255,0))
- elif idle_mode == 4:
- setColor(strip, Color(0,0,255))
- elif idle_mode == 5:
- setColor(strip, Color(125,245,255))
- elif idle_mode == 6:
- setColor(strip, Color(255,0,100))
- elif idle_mode == 7:
- setColor(strip, Color(0,255,255))
- else:
- blink(strip, Color(255,0,0), Color(0,0,0))
+ if power == True:
+
time.sleep(1)
else:
setColor(strip, Color(0,0,0))