jmeter简单实践（九）

简单版本的httpserver

json模块可能需要下载，详细方法请百度，增加个人的能力

主要是json的，接收到非json的http请求，返回"415, "Only json data is supported."

coding: utf-8

from http.server import BaseHTTPRequestHandler,HTTPServer
import cgi
import json



def Cjson(datas):
    try:
        messages = json.loads(datas)
    except ValueError:
        return False
    return True
class TodoHandler(BaseHTTPRequestHandler):
    """A simple TODO server

    which can display and manage todos for you.
    """

    # Global instance to store todos. You should use a database in reality.
    TODOS = []
    def do_GET(self):
        # return all todos

        if self.path != '/':
            self.send_error(404, "File not found.")
            return

        # Just dump data to json, and return it
        message = json.dumps(self.TODOS)

        self.send_response(200)
        self.send_header('Content-type', 'application/json')
        self.end_headers()
        self.wfile.write(bytes(message,'utf-8'))

    def do_POST(self):
        """Add a new todo

        Only json data is supported, otherwise send a 415 response back.
        Append new todo to class variable, and it will be displayed
        in following get request
        """
        ctype, pdict = cgi.parse_header(self.headers['content-type'])
        if ctype == 'application/json':
            length = int(self.headers['content-length'])
            dd = self.rfile.read(length)
            post_values =Cjson(str(dd,'UTF-8'))
            if post_values == False:
                print(dd)
                self.TODOS.append(str(dd, 'UTF-8'))
                self.send_response(200)
                self.send_header(b'Content-type', b'application/json')
                self.end_headers()
                jss = {
                    "code": 200,
                    "msg": "OK",
                }
                r = json.dumps(jss)
                self.wfile.write(bytes(r, 'utf-8'))
            if post_values == True:
                self.TODOS.append(json.loads(str(dd,'utf-8')))
                self.send_response(200)
                self.send_header(b'Content-type', b'application/json')
                self.end_headers()
                jss = {
                    "code": 200,
                    "msg": "OK",
                }
                r = json.dumps(jss)
                self.wfile.write(bytes(r, 'utf-8'))
        else:
            length = int(self.headers['content-length'])
            post_values = self.rfile.read(length)
            self.send_error(415, "Only json data is supported.")
            return




if __name__ == '__main__':
    # Start a simple server, and loop forever

    server = HTTPServer(('localhost', 8888), TodoHandler)
    print("Starting server, use <Ctrl-C> to stop")
    server.serve_forever()