Files
E-Manager/server/main.py

100 lines
3.3 KiB
Python

from flask import Flask, request, jsonify
import influxdb_client
import os
import json
from datetime import datetime
from pytz import timezone
app = Flask(__name__)
# InfluxDB-Verbindungsinformationen
INFLUXDB_HOST = 'http://192.168.20.21:8086'
INFLUXDB_PORT = 8086
INFLUXDB_TOKEN = 'yjgsLHr3Fxj6LwCRvJlPSMTkTg7CwEn2ajsAeaxLHIZ_17ZHyUUn7P9nCpkdioOexDlWq73zTatmZ_ajvTPozA=='
INFLUXDB_ORG = 'Belevo AG'
INFLUXDB_BUCKET = 'E-Manger_Demo'
# InfluxDB-Client initialisieren
influx_client = influxdb_client.InfluxDBClient(url=INFLUXDB_HOST, token=INFLUXDB_TOKEN, org=INFLUXDB_ORG)
# Route to get configuration
# Route to get configuration
@app.route('/getconfig', methods=['POST'])
def get_config():
#try:
token_id = request.headers.get('ID')
if not token_id:
return jsonify({'error': 'Token ID is missing'}), 400
# Query to get the latest timestamp from InfluxDB
query = f'from(bucket: "{INFLUXDB_BUCKET}") |> range(start: -1d) |> filter(fn: (r) => r._measurement == "{token_id}") |> last()'
result = influx_client.query_api().query(query)
records = list(result)
if records:
last_timestamp = records[0].records[0].get_time()
last_timestamp = last_timestamp.strftime('%Y-%m-%d %H:%M')
else:
# Set last timestamp to epoch time if no records found
last_timestamp = datetime(1970, 1, 1).strftime('%Y-%m-%d %H:%M')
# Load config from JSON file
config_file = os.path.join('config', f'{token_id}.json')
if os.path.exists(config_file):
with open(config_file, 'r') as file:
config = json.load(file)
else:
config = {}
# Add last timestamp to config
config['last_timestamp'] = last_timestamp
return jsonify(config), 200
#except Exception as e:
# return jsonify({'error': str(e)}), 500
# Route to set new data
@app.route('/setdata', methods=['POST'])
def set_data():
# Token-ID aus den Request-Headern extrahieren
token_id = request.headers.get('ID')
# Neue Daten aus den Request-Headern laden
new_data = request.headers.get('JSON')
try:
# Versuche, das JSON zu laden
new_data = json.loads(new_data)
except json.JSONDecodeError:
# Wenn das JSON nicht korrekt formatiert ist, schreibe das neue JSON in die Datenbank und beende die Funktion
write_to_influx(token_id, new_data)
return 'Invalid JSON format. New JSON data has been written to InfluxDB and overwritten existing data.'
# Daten in die InfluxDB schreiben
write_to_influx(token_id, new_data)
return 'Data successfully set!'
def write_to_influx(token_id, json_data):
"""
Funktion zum Schreiben von Daten in InfluxDB.
"""
json_body = [
{
"measurement": token_id, # Verwende die Token-ID als Messungsnamen
"fields": {
"data": json.dumps(json_data) # Das gesamte JSON als Text speichern
}
}
]
# Datenpunkt in die InfluxDB einfügen
influx_client.write_api().write(bucket=INFLUXDB_BUCKET, org=INFLUXDB_ORG, record=json_body)
if __name__ == '__main__':
app.run(debug=False, host="127.0.0.1")#, port=5000)
#app.run(host="0.0.0.0", port=5000)