mirror of
https://github.com/KevinMidboe/patlite-tower-lights-controller.git
synced 2025-10-29 09:40:29 +00:00
101 lines
2.7 KiB
Python
101 lines
2.7 KiB
Python
import time
|
|
import argparse
|
|
import RPi.GPIO as GPIO
|
|
from light import Tower
|
|
import database
|
|
from flask import Flask, jsonify, request
|
|
|
|
# Pins configuration (order: green, orange, red)
|
|
GPIO_PINS = [{'name': 'green', 'pin': 22},
|
|
{'name': 'orange', 'pin': 17},
|
|
{'name': 'red', 'pin': 27}]
|
|
|
|
# Create a Flask app
|
|
app = Flask(__name__)
|
|
|
|
# Initialize the database
|
|
database.ensure_database()
|
|
|
|
# Initialize the tower (lights)
|
|
tower = Tower(GPIO_PINS)
|
|
|
|
|
|
@app.after_request
|
|
def add_cors_headers(response):
|
|
response.headers['Access-Control-Allow-Origin'] = '*'
|
|
response.headers['Access-Control-Allow-Methods'] = 'GET, POST, OPTIONS'
|
|
response.headers['Access-Control-Allow-Headers'] = 'Content-Type'
|
|
return response
|
|
|
|
|
|
@app.route('/api/state/set', methods=['POST'])
|
|
def set_state():
|
|
data = request.data.decode('utf-8')
|
|
state = [int(n.strip()) for n in data.split(',') if n.strip().isdigit()]
|
|
"""Endpoint to get the current state of all lights."""
|
|
tower.green.on() if state[0] else tower.green.off()
|
|
tower.orange.on() if state[1] else tower.orange.off()
|
|
tower.red.on() if state[2] else tower.red.off()
|
|
|
|
return jsonify(state)
|
|
|
|
|
|
@app.route('/api/state', methods=['GET'])
|
|
def get_state():
|
|
"""Endpoint to get the current state of all lights."""
|
|
state = database.get_all_states()
|
|
return jsonify(state)
|
|
|
|
|
|
@app.route('/api/toggle/<color>', methods=['POST'])
|
|
def toggle_light(color):
|
|
"""Endpoint to toggle the state of a specified light."""
|
|
if color not in ['green', 'orange', 'red']:
|
|
return jsonify({'error': 'Invalid color'}), 404
|
|
|
|
# Get the light object based on the color
|
|
light = getattr(tower, color)
|
|
|
|
# Toggle light state (on/off)
|
|
new_state = GPIO.LOW if light.state == GPIO.HIGH else GPIO.HIGH
|
|
light._set(new_state) # Update the light state in the system
|
|
|
|
return jsonify({color: new_state})
|
|
|
|
|
|
def run(tower):
|
|
"""Run the tower with manual light control."""
|
|
try:
|
|
time.sleep(0.5)
|
|
while True:
|
|
print('on')
|
|
tower.green.on()
|
|
tower.red.off()
|
|
tower.orange.off()
|
|
|
|
time.sleep(10)
|
|
except KeyboardInterrupt:
|
|
print("\nInterrupted by user. Cleaning up...")
|
|
finally:
|
|
GPIO.cleanup()
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(
|
|
description="Control lights or run server.")
|
|
parser.add_argument('--server', action='store_true',
|
|
help='Run the Flask server to control lights via API')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.server:
|
|
print("Starting Flask server...")
|
|
app.run(host='0.0.0.0', port=5000)
|
|
else:
|
|
print("Running tower control manually...")
|
|
run(tower)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|