init commit

server/main.py

@@ -0,0 +1,100 @@
+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()