You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
120 lines
3.5 KiB
120 lines
3.5 KiB
from flask import Flask, jsonify, request
|
|
from flask_socketio import SocketIO, emit
|
|
from flask_cors import CORS
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
from datetime import datetime, timedelta
|
|
from threading import Thread
|
|
import time
|
|
import random
|
|
|
|
app = Flask(__name__)
|
|
CORS(app)
|
|
socketio = SocketIO(app, cors_allowed_origins="*")
|
|
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///pairing_codes.db'
|
|
db = SQLAlchemy(app)
|
|
|
|
|
|
class PairingCode(db.Model):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
code = db.Column(db.String(80), unique=True, nullable=False)
|
|
session_id = db.Column(db.String(120), unique=True, nullable=True)
|
|
timestamp = db.Column(db.DateTime, default=db.func.current_timestamp())
|
|
last_heartbeat = db.Column(db.DateTime, default=db.func.current_timestamp())
|
|
|
|
|
|
with app.app_context():
|
|
db.create_all()
|
|
|
|
|
|
def cleanup_expired_codes():
|
|
while True:
|
|
with app.app_context():
|
|
expired_time = datetime.utcnow() - timedelta(minutes=15)
|
|
expired_codes = PairingCode.query.filter(PairingCode.last_heartbeat < expired_time).all()
|
|
|
|
for code in expired_codes:
|
|
db.session.delete(code)
|
|
db.session.commit()
|
|
|
|
time.sleep(600)
|
|
|
|
|
|
cleanup_thread = Thread(target=cleanup_expired_codes)
|
|
cleanup_thread.start()
|
|
|
|
|
|
@app.route('/generate_code', methods=['GET'])
|
|
def generate_code():
|
|
while True:
|
|
code = str(random.randint(1000, 9999))
|
|
existing_code = PairingCode.query.filter_by(code=code).first()
|
|
if not existing_code:
|
|
break
|
|
|
|
new_code = PairingCode(code=code)
|
|
db.session.add(new_code)
|
|
db.session.commit()
|
|
|
|
return jsonify({"code": code})
|
|
|
|
|
|
@socketio.on('pair')
|
|
def handle_pairing(json):
|
|
code = json.get('code')
|
|
pairing_code = PairingCode.query.filter_by(code=code).first()
|
|
|
|
if pairing_code:
|
|
pairing_code.session_id = request.sid
|
|
|
|
PairingCode.query.filter(PairingCode.session_id == request.sid, PairingCode.id != pairing_code.id).delete()
|
|
|
|
db.session.commit()
|
|
emit('paired', {'status': 'success'}, room=request.sid)
|
|
else:
|
|
emit('error', {'status': 'invalid_code'}, room=request.sid)
|
|
|
|
|
|
@socketio.on('heartbeat')
|
|
def handle_heartbeat(json):
|
|
code = json.get('code')
|
|
pairing_code = PairingCode.query.filter_by(code=code).first()
|
|
|
|
if pairing_code:
|
|
now_without_ms = datetime.utcnow().replace(microsecond=0)
|
|
pairing_code.last_heartbeat = now_without_ms
|
|
db.session.commit()
|
|
|
|
|
|
@app.route('/validate_code', methods=['POST'])
|
|
def validate_code():
|
|
data = request.json
|
|
code = data.get('code')
|
|
|
|
pairing_code = PairingCode.query.filter_by(code=code).first()
|
|
|
|
if pairing_code and pairing_code.last_heartbeat:
|
|
time_since_last_heartbeat = datetime.utcnow() - pairing_code.last_heartbeat
|
|
if time_since_last_heartbeat < timedelta(minutes=15):
|
|
return jsonify({'status': 'valid'})
|
|
else:
|
|
return jsonify({'status': 'expired'}), 400
|
|
else:
|
|
return jsonify({'status': 'invalid'}), 400
|
|
|
|
@app.route('/send_gesture', methods=['POST'])
|
|
def handle_gesture():
|
|
data = request.json
|
|
code = data.get('code')
|
|
gesture_value = data.get('gesture')
|
|
|
|
pairing_code = PairingCode.query.filter_by(code=code).first()
|
|
|
|
if pairing_code and pairing_code.session_id:
|
|
socketio.emit('gesture_event', {'gesture': gesture_value}, room=pairing_code.session_id)
|
|
return jsonify({'status': 'success'})
|
|
else:
|
|
return jsonify({'status': 'code_not_paired'}), 400
|
|
|
|
|
|
if __name__ == '__main__':
|
|
socketio.run(app, debug=True)
|