Files
MouseTracker/app.py
T
2024-03-28 09:04:24 +08:00

37 lines
982 B
Python

import os
import sqlite3
from flask import Flask, request, jsonify, render_template
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy import PickleType
basedir = os.path.abspath(os.path.dirname(__file__))
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + os.path.join(basedir, 'database.db') # 使用 SQLite 数据库
db = SQLAlchemy(app)
class Tracker(db.Model):
id = db.Column(db.Integer, primary_key=True)
distance = db.Column(db.Integer, nullable=False)
data = db.Column(PickleType(), nullable=False)
@app.route('/', methods=['GET'])
def index():
db.create_all()
return render_template('index.html')
@app.route("/", methods=["POST"])
def upload_data():
data = request.json
print(data)
record = Tracker(distance=data["distance"], data=data["data"])
print(sum(item[0] for item in data["data"]))
db.session.add(record)
db.session.commit()
return jsonify({"message": "Data recorded"}), 200