GMTROM/REST/old/app.py

99 lines
3.8 KiB
Python
Raw Normal View History

import subprocess
import json
import time
from flask import Flask, request, jsonify
from I2C_Function import analyze_data, read_data
from flask_cors import CORS
from threading import Lock
app = Flask(__name__)
CORS(app)
initialized = False
lock = Lock() # Sperre (Lock) für die analyze_data-Methode
def create_response(status_code, description):
return jsonify({
'status': status_code,
'description': description
})
@app.route('/trng/randomNum/init', methods=['GET'])
def initialize_generator():
global initialized
with lock: # Sperre (Lock) verwenden, um sicherzustellen, dass nur ein Thread die Methode ausführt
result = analyze_data(int(1), int(1), startup=True)
if result is True:
initialized = True
return create_response(200, 'Successful operation; random number generator is ready and random numbers can be requested')
if result == 400:
return create_response(543, 'Tests failed, try again')
if result is False:
return create_response(500, 'Unable to generate random numbers. Restart/Reset System')
else:
return create_response(555, 'Unable to initialize the random number generator within a timeout of 60 seconds')
@app.route('/trng/randomNum/getRandom', methods=['GET'])
def get_random_numbers():
global initialized
quantity = request.args.get('quantity', default=1)
bits = request.args.get('numBits', default=1)
if not quantity.isdigit() or not bits.isdigit():
return create_response(400, 'Invalid input. Quantity and bits must be numeric')
if int(quantity) <= 0 or int(bits) <= 0:
return create_response(400, 'Invalid input. Quantity and bits must be positive integers')
if not initialized:
return create_response(432, 'System not ready; try init')
with lock: # Sperre (Lock) verwenden, um sicherzustellen, dass nur ein Thread die Methode ausführt
random_numbers = analyze_data(int(quantity), int(bits), startup=False)
if random_numbers == 400:
return create_response(543, 'Tests failed, try again')
if random_numbers is False:
return create_response(500, 'Unable to generate random numbers. Restart/Reset System')
if random_numbers:
data = {
'status': 200,
'description': 'Successful operation; HEX-encoded bit arrays (with leading zeros if required)',
'randomNumbers': random_numbers
}
return jsonify(data)
else:
return create_response(500, 'Unable to generate random numbers.')
@app.route('/trng/randomNum/shutdown', methods=['GET'])
def shutdown_generator():
global initialized
initialized = False
return create_response(200, "Successful operation; random number generator has been set to 'standby mode'")
# Fehlerbehandlungsroutine für nicht übereinstimmende Routen
@app.route('/', defaults={'path': ''})
@app.route('/<path:path>')
def handle_invalid_routes(path):
return create_response(404, 'Route not found')
if __name__ == '__main__':
# Pfade zu den SSL-Zertifikat- und Schlüsseldateien
cert_path = '/etc/nginx/ssl/cert-gmtrom.pem'
key_path = '/etc/nginx/ssl/cert-gmtrom-key.pem'
#alte Zertifikate self signed
#cert_path = '/etc/nginx/ssl/server.crt'
#key_path = '/etc/nginx/ssl/server.key'
# Starte die Flask-Anwendung mit SSL-Konfiguration
app.run(host='0.0.0.0', ssl_context=(cert_path, key_path))
#app.run(host='172.16.78.57', port=5000, ssl_context=(cert_path, key_path))
#app.run(host='0.0.0.0', ssl_context=adhoc)
#app.run(ssl_context=(cert_path, key_path))