116 lines
4.4 KiB
Python
116 lines
4.4 KiB
Python
from flask import Flask, request, jsonify
|
|
from datetime import datetime
|
|
import sqlite3
|
|
import requests
|
|
import os
|
|
from instances.db import init_db, DB_PATH
|
|
from mission_utils import delayed_post_mission, post_mission, get_robots, get_status
|
|
|
|
app = Flask(__name__)
|
|
|
|
# Variable to store the status of inputs and outputs
|
|
status = {
|
|
"input1": "off",
|
|
"input2": "off",
|
|
"input3": "off",
|
|
"input4": "off",
|
|
"relay1": "off",
|
|
"relay2": "off",
|
|
"relay3": "off",
|
|
"relay4": "off"
|
|
}
|
|
|
|
@app.route('/log', methods=['POST'])
|
|
def log():
|
|
data = request.json
|
|
if not data:
|
|
return jsonify({"error": "No data provided"}), 400
|
|
|
|
# Extract data from the JSON payload
|
|
hostname = data.get('hostname', 'unknown')
|
|
ip_address = data.get('ip_address', 'unknown')
|
|
message = data.get('message', '')
|
|
|
|
print(f"Board hostname: {hostname}")
|
|
print(f"Board IP address: {ip_address}")
|
|
print(f"Message: {message}")
|
|
|
|
# Record the log in the database
|
|
conn = sqlite3.connect(DB_PATH)
|
|
c = conn.cursor()
|
|
c.execute("INSERT INTO logs (timestamp, hostname, ip_address, message) VALUES (?, ?, ?, ?)",
|
|
(datetime.now().isoformat(), hostname, ip_address, message))
|
|
conn.commit()
|
|
print(f"Log entry added: {hostname}, {ip_address}, {message}")
|
|
|
|
# Check if the board exists in the database
|
|
c.execute("SELECT * FROM boards WHERE ip_address = ?", (ip_address,))
|
|
board = c.fetchone()
|
|
|
|
if not board:
|
|
# Create a new board entry if it doesn't exist
|
|
c.execute('''INSERT INTO boards (hostname, ip_address, input1, input2, input3, input4, relay1, relay2, relay3, relay4)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)''',
|
|
(hostname, ip_address, 'off', 'off', 'off', 'off', 'off', 'off', 'off', 'off'))
|
|
conn.commit()
|
|
print(f"New board entry created: {hostname}, {ip_address}")
|
|
|
|
# Update the status of inputs and relays based on the message
|
|
for i in range(1, 5):
|
|
if f"Input {i} pressed" in message:
|
|
status[f"input{i}"] = 'on'
|
|
print(f"Input {i} is {status[f'input{i}']}")
|
|
if i == 1:
|
|
print("Input 1 is on, triggering delayed_post_mission")
|
|
delayed_post_mission({"mission_id": "a1343edc-975c-11ef-87e6-0242ac120002"}) # Example mission ID
|
|
c.execute(f'''UPDATE boards SET input{i} = ? WHERE ip_address = ?''',
|
|
(status[f"input{i}"], ip_address))
|
|
conn.commit()
|
|
print(f"Board status updated: {hostname}, {ip_address}, input{i}={status[f'input{i}']}")
|
|
|
|
if f"Input {i} released" in message:
|
|
status[f"input{i}"] = 'off'
|
|
print(f"Input {i} is {status[f'input{i}']}")
|
|
c.execute(f'''UPDATE boards SET input{i} = ? WHERE ip_address = ?''',
|
|
(status[f"input{i}"], ip_address))
|
|
conn.commit()
|
|
print(f"Board status updated: {hostname}, {ip_address}, input{i}={status[f'input{i}']}")
|
|
|
|
for i in range(1, 5):
|
|
if f"Relay {i} turned ON" in message:
|
|
status[f"relay{i}"] = 'on'
|
|
print(f"Relay {i} status is {status[f'relay{i}']}")
|
|
c.execute(f'''UPDATE boards SET relay{i} = ? WHERE ip_address = ?''',
|
|
(status[f"relay{i}"], ip_address))
|
|
conn.commit()
|
|
print(f"Board status updated: {hostname}, {ip_address}, relay{i}={status[f'relay{i}']}")
|
|
|
|
if f"Relay {i} turned OFF" in message:
|
|
status[f"relay{i}"] = 'off'
|
|
print(f"Relay {i} status is {status[f'relay{i}']}")
|
|
c.execute(f'''UPDATE boards SET relay{i} = ? WHERE ip_address = ?''',
|
|
(status[f"relay{i}"], ip_address))
|
|
conn.commit()
|
|
print(f"Board status updated: {hostname}, {ip_address}, relay{i}={status[f'relay{i}']}")
|
|
|
|
conn.close()
|
|
return jsonify({"message": "Log received"}), 200
|
|
|
|
@app.route('/control', methods=['POST'])
|
|
def control():
|
|
data = request.json
|
|
if not data:
|
|
return jsonify({"error": "No data provided"}), 400
|
|
|
|
# Control the relays based on the received data
|
|
for i in range(1, 5):
|
|
if f'relay{i}' in data:
|
|
status[f'relay{i}'] = data[f'relay{i}']
|
|
print(f"Setting Relay {i} to {data[f'relay{i}']}")
|
|
|
|
return jsonify({"message": "Control command received"}), 200
|
|
|
|
if __name__ == '__main__':
|
|
init_db()
|
|
get_robots()
|
|
app.run(host='0.0.0.0', port=80) |