from flask import Flask, jsonify, request
from flask_cors import CORS  # 导入 CORS 模块


app = Flask(__name__)
CORS(app)
power=0
current=0

# 激光心跳
@app.route('/heartbeat', methods=['GET'])
def heartbeat():
    return jsonify({'heartbeat': 'ok'}), 200

# 获取电源状态
@app.route('/getpower', methods=['GET'])
def getpower():
    global power
    if power==0:
        power_value = "off"
    elif power==1:
        power_value = "on"

    return jsonify({'power': f'{power_value}'}), 200

# 设置电源状态
@app.route('/setpower', methods=['GET'])
def setpower():
    global power
    power_value = request.args.get('power')
    if power_value is not None:
        if power_value == 'on':
            power=1
        elif power_value == 'off':
            power=0
    return jsonify({'power': f'{power}'}), 200

# 获取当前电流
@app.route('/getcurrent', methods=['GET'])
def getcurrent():
    global current
    current_value=current
    return jsonify({'current': f'{current_value}'}), 200

# 设置当前电流
@app.route('/setcurrent', methods=['GET'])
def setcurrent():
    global current
    current_value = request.args.get('current')
    if current_value is not None:
        current=int(current_value)
    return jsonify({'current': f'{current}'}), 200

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=12224)