app.py

from urllib.request import urlopen, HTTPError, URLError
import xml.etree.ElementTree as ET
from flask import Flask, render_template, jsonify, request, redirect, url_for
from apscheduler.schedulers.background import BackgroundScheduler
import json

app = Flask(__name__)


receivers = {
    "RX1": {"url": "http://10.0.101.109/data.xml", "data": {}, "graph_settings": {"thresholds": [{"value": 10, "color": "#FF0000"}, {"value": 20, "color": "#FFFF00"}]}},
    "RX2": {"url": "http://10.0.101.110/data.xml", "data": {}, "graph_settings": {"thresholds": [{"value": 10, "color": "#FF0000"}, {"value": 20, "color": "#FFFF00"}]}},
    "RX3": {"url": "http://10.0.101.111/data.xml", "data": {}, "graph_settings": {"thresholds": [{"value": 10, "color": "#FF0000"}, {"value": 20, "color": "#FFFF00"}]}},
    "RX4": {"url": "http://10.0.101.112/data.xml", "data": {}, "graph_settings": {"thresholds": [{"value": 10, "color": "#FF0000"}, {"value": 20, "color": "#FFFF00"}]}},
}


def save_to_file():
    with open('receiver_settings.json', 'w') as f:
        json.dump(receivers, f)

def load_from_file():
    try:
        with open('receiver_settings.json', 'r') as f:
            data = json.load(f)
            for key in receivers:
                receivers[key]['url'] = data.get(key, {}).get('url', receivers[key]['url'])
    except FileNotFoundError:
        pass  # If the file doesn't exist yet, just use the default URLs
    except json.JSONDecodeError:
        print("Error: receiver_settings.json contains invalid JSON. Please check or delete the file.")

load_from_file()

def retrieve_data_for_all_receivers():
    for key in receivers.keys():
        retrieve_data(key)

def frequency(freq):
    for page in freq.findall('parameter'):
        if page.find('name').text == 'DB_L2174_FREQ1':
            return page.find('value').text

def MER_1(MER1):
    for page in MER1.findall('parameter'):
        if page.find('name').text == 'DB_DEMOD_MER_1':
            return round(float(page.find('value').text))

def MER_2(MER2):
    for page in MER2.findall('parameter'):
        if page.find('name').text == 'DB_DEMOD_MER_2':
            return round(float(page.find('value').text))

def MER_3(MER3):
    for page in MER3.findall('parameter'):
        if page.find('name').text == 'DB_DEMOD_MER_3':
            return round(float(page.find('value').text))

def MER_4(MER4):
    for page in MER4.findall('parameter'):
        if page.find('name').text == 'DB_DEMOD_MER_4':
            return round(float(page.find('value').text))

"""       
def TRIAX_MODE(TRIAXMODE):
    for page in TRIAXMODE.findall('parameter'):
        if page.find('name').text == '':
            return round(float(page.find('value').text))

"""
def retrieve_data(receiver_key):
    try:
        with urlopen(receivers[receiver_key]["url"], timeout=0.5) as d:
            tree = ET.parse(d)
            root = tree.getroot()

        data = {
            "frequency": frequency(root),
            "mer_1": MER_1(root),
            "mer_2": MER_2(root),
            "mer_3": MER_3(root),
            "mer_4": MER_4(root)
        }

        receivers[receiver_key]["data"] = data

        for key, value in data.items():
            if key == "frequency":
                print(f"{receiver_key} {key.capitalize()}: {value}")
            else:
                print(f"{key.capitalize()}: {value}")

    except (HTTPError, URLError) as e:
        print(f"Error accessing URL for {receiver_key}. URL: {receivers[receiver_key]['url']}. Error: {e}")
    except ET.ParseError as e:
        print(f"XML ParseError for {receiver_key}. URL: {receivers[receiver_key]['url']}. Error: {e}")

@app.route('/')
def index():
    return render_template('index.html', receivers=receivers)

@app.route('/data')
def get_data():
    all_data = {}
    for key in receivers.keys():
        retrieve_data(key)
        all_data[key] = receivers[key]["data"]
    return jsonify(all_data)

@app.route('/settings', methods=['GET', 'POST'])
def settings():
    if request.method == 'POST':
        print("Received POST data:")
        print(request.form)  # Print the entire form data

        for key in receivers:
            # Check if the specific receiver should be updated
            if request.form.get(f'{key}_update') == 'yes':
                # Update IP address
                full_ip = request.form.get(f'{key}_ip')
                if full_ip:  # Ensure there is an IP address provided
                    receivers[key]['url'] = f'http://{full_ip}/data.xml'

                # Update graph settings
                threshold_settings = []
                for i in range(1, 6):  # Assuming a maximum of 5 thresholds
                    threshold = request.form.get(f'{key}_threshold_{i}')
                    color = request.form.get(f'{key}_color_{i}')
                    if threshold and color:
                        threshold_settings.append({
                            "value": int(threshold),
                            "color": color
                        })

                # Only update thresholds if there are any
                if threshold_settings:
                    receivers[key]['graph_settings']['thresholds'] = threshold_settings

        save_to_file()
        return redirect(url_for('settings'))
    return render_template('settings.html', receivers=receivers)



if __name__ == '__main__':
    scheduler = BackgroundScheduler()
    scheduler.add_job(retrieve_data_for_all_receivers, 'interval', seconds=5)
    scheduler.start()
    app.run(host='0.0.0.0', port=5000, debug=False)