mirror of
https://github.com/DJ2LS/FreeDATA
synced 2024-05-14 08:04:33 +00:00
77 lines
2 KiB
Python
77 lines
2 KiB
Python
|
from flask import Flask, request, jsonify
|
||
|
|
from flask_sock import Sock
|
||
|
|
import argparse, json
|
||
|
|
from config import CONFIG
|
||
|
|
|
||
|
|
app = Flask(__name__)
|
||
|
|
sock = Sock(app)
|
||
|
|
|
||
|
|
# CLI arguments
|
||
|
|
def get_config_filename_from_args():
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument("-c", "--config", help = "Specifiy config file", default = 'config.ini')
|
||
|
|
args, unknown = parser.parse_known_args()
|
||
|
|
return args.config
|
||
|
|
|
||
|
|
config = CONFIG(get_config_filename_from_args())
|
||
|
|
|
||
|
|
def api_response(data, status = 'ok'):
|
||
|
|
response = {
|
||
|
|
'status': status,
|
||
|
|
'data': data
|
||
|
|
}
|
||
|
|
return jsonify(response)
|
||
|
|
|
||
|
|
## REST API
|
||
|
|
@app.route('/', methods=['GET'])
|
||
|
|
def index():
|
||
|
|
return jsonify({'name': 'FreeDATA API',
|
||
|
|
'description': '',
|
||
|
|
'api_version': 1,
|
||
|
|
'license': 'GPL3.0',
|
||
|
|
'documentation': 'https://wiki.freedata.app',
|
||
|
|
})
|
||
|
|
|
||
|
|
# get and set config
|
||
|
|
@app.route('/config', methods=['GET', 'POST'])
|
||
|
|
def config():
|
||
|
|
if request.method == 'POST':
|
||
|
|
new_config = json.loads(request.form['config'])
|
||
|
|
set_config = config.write(new_config)
|
||
|
|
if not set_config:
|
||
|
|
response = api_response(None, 'error writing config')
|
||
|
|
else:
|
||
|
|
response = api_response(set_config)
|
||
|
|
return response
|
||
|
|
elif request.method == 'GET':
|
||
|
|
return api_response(config.read())
|
||
|
|
|
||
|
|
# get activity
|
||
|
|
@app.route('/activity', methods=['GET'])
|
||
|
|
def activity():
|
||
|
|
return "Not implemented yet."
|
||
|
|
|
||
|
|
# get received messages
|
||
|
|
@app.route('/messages', methods=['GET'])
|
||
|
|
def messages():
|
||
|
|
return "Not implemented yet."
|
||
|
|
|
||
|
|
# new message / delete message
|
||
|
|
""" @app.route('/message', methods=['POST', 'DELETE'])
|
||
|
|
def message():
|
||
|
|
if request.method == 'POST':
|
||
|
|
message = new Message(request.form['message'])
|
||
|
|
status = modem.send_message(message)
|
||
|
|
elif request.method == 'DELETE':
|
||
|
|
status = messageDb.deleteMessage(request.form['id'])
|
||
|
|
return status
|
||
|
|
"""
|
||
|
|
# Event websocket
|
||
|
|
@sock.route('/events')
|
||
|
|
def echo(sock):
|
||
|
|
while True:
|
||
|
|
data = sock.receive()
|
||
|
|
sock.send(data)
|
||
|
|
|
||
|
|
|