py_web.py

from flask import Flask, jsonify, request
import os
import random

app = Flask(__name__)

@app.route("/hello", methods=["GET"])
def hello():
    return "hello"

@app.route("/polling", methods=["GET"])
def getMidState():
    id = request.args["id"]
    res = {
        "id": id,
        "name": "test_train_task",
        "loss": random.randint(1, 999),
        "epoch": random.randint(1, 999),
        "acc": random.random(),
    }
    return jsonify(res)

@app.route("/upload", methods=["POST"])
def upload():
    file = request.files.get('file')
    if file is None:
        return "训练服务器未收到文件"
    storePath = "/root/py_web/" + file.filename
    # basePath = os.path.dirname(__file__)
    # storePath = basePath + os.altsep + "py_web" + os.altsep + file.filename
    print(storePath)
    file.save(storePath)
    print('user: ' + request.form.get('user'))
    print('task: ' + request.form.get('task'))
    print('label: ' + request.form.get('label'))
    print('isSupervised: ' + request.form.get('isSupervised'))
    return "训练服务器已收到数据"

@app.route("/train", methods=["POST"])
def train():
    print('id: ' + request.form.get('id'))
    print('name: ' + request.form.get('name'))
    print('method: ' + request.form.get('method'))
    print('epoch: ' + request.form.get('epoch'))
    print('learningRate: ' + request.form.get('learningRate'))
    print('batchSize: ' + request.form.get('batchSize'))
    print('second: ' + request.form.get('second'))
    print('isSupervised: ' + request.form.get('isSupervised'))
    print('uid: ' + request.form.get('uid'))
    print('task: ' + request.form.get('task'))
    print('sigFileNameList: ' + request.form.get('sigFileNameList'))
    return "训练服务器已收到训练请求"

if __name__ == "__main__":
    app.run()