一个可以接受POST请求的python脚本

# docker build -t flow_test:0.1.0 .
FROM ubuntu:18.04

WORKDIR /home/linghu8812
ADD . /home/linghu8812

RUN mv sources.list /etc/apt/ && mv .pip/ /root/.pip/ && apt-get update && apt-get install -y python3-pip curl \
    && pip3 install --upgrade pip && pip3 install ai-hub flask

"""
依赖：pip install ai-hub flask
测试用例：
model为y=2*x
请求数据为json:{"img":3}
-----------
post请求：
curl localhost:8080/tccapi -X POST -d '{"img":3}'
返回结果 6
"""

from ai_hub import inferServer
import json


class MyInfer(inferServer):
    def __init__(self, model):
        super().__init__(model)
        print("init_myInfer")

    # 数据前处理
    def pre_process(self, data):
        print("my_pre_process")
        # json process
        json_data = json.loads(data.get_data().decode('utf-8'))
        img = json_data.get("img")
        print("processed data: ", img)
        return img

    # 数据后处理
    def post_process(self, data):
        print("post_process")
        processed_data = data
        print(processed_data)
        return processed_data

    # 模型预测：默认执行self.model(preprocess_data)，一般不用重写
    # 如需自定义，可覆盖重写
    # def predict(self, data)：
    #    ret = self.model(data)
    #    return ret


if __name__ == "__main__":
    my_model = lambda x: x * 2
    my_infer = MyInfer(my_model)
    my_infer.run("0.0.0.0", 8080, debuge=True)  # 默认为("127.0.0.1", 80)，可自定义端口，如用于天池大赛请默认即可，指定debuge=True可获得更多报错信息