kbengine-httpserver/httpserver/app.py at master · kenny67/kbengine-httpserver · GitHub

History

129 lines (96 loc) · 3.42 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

# -*- coding: utf-8 -*-

"""

FileName: app

Author: Tao Hao

@contact: taohaohust@outlook.com

Created time: 2019/5/29

Description:

Changelog:

"""

from httpserver.exceptions import MethodNotAllowed

import json

from httpserver.response import HTTPResponse

from httpserver.router import Router

from httpserver.server import Server

try:

from ujson import dumps as json_dumps

except ImportError:

json_dumps = json.dumps

class Application(object):

def __init__(self, url_handler_list):

self.router = None

self.server = None

self.create_router(url_handler_list)

def create_router(self, url_handler_list):

if self.router:

raise Exception("route has exist")

self.router = Router()

for uri, handler in url_handler_list:

if not issubclass(handler, RequestHandler):

raise Exception("handler type is error")

self.router.add(uri, handler)

def run(self, host, port, backlog=128):

self.server = Server(self)

self.server.run(host, port, backlog)

def stop(self):

self.server.stop()

self.router = None

class RequestHandler(object):

def __init__(self, request):

self.request = request

def execute_handler(self, *args, **kwargs):

method = self.request.method

if method == "HEAD":

self.head(*args, **kwargs)

elif method == "GET":

self.get(*args, **kwargs)

elif method == "POST":

self.post(*args, **kwargs)

elif method == "DELETE":

self.delete(*args, **kwargs)

elif method == "PATCH":

self.patch(*args, **kwargs)

elif method == "PUT":

self.put(*args, **kwargs)

elif method == "OPTIONS":

self.options(*args, **kwargs)

def head(self, *args, **kwargs):

raise MethodNotAllowed("HEAD")

def get(self, *args, **kwargs):

raise MethodNotAllowed("GET")

def post(self, *args, **kwargs):

raise MethodNotAllowed("POST")

def delete(self, *args, **kwargs):

raise MethodNotAllowed("DELETE")

def patch(self, *args, **kwargs):

raise MethodNotAllowed("PATCH")

def put(self, *args, **kwargs):

raise MethodNotAllowed("PUT")

def options(self, *args, **kwargs):

raise MethodNotAllowed("OPTIONS")

def write_text(self, text, status=200, headers=None,

content_type="text/plain; charset=utf-8"):

http_response = HTTPResponse(

body=text, status=status, headers=headers,

content_type=content_type

)

self.request.connection.write_response(http_response)

self._cleanup()

def write_json(self, body, status=200, headers=None,

content_type="application/json", dumps=json_dumps):

http_response = HTTPResponse(

body=dumps(body), status=status, headers=headers,

content_type=content_type

)

self.request.connection.write_response(http_response)

self._cleanup()

def write_raw(self, body, status=200, headers=None,

content_type="application/octet-stream"):

http_response = HTTPResponse(

body_bytes=body, status=status, headers=headers,

content_type=content_type

)

self.request.connection.write_response(http_response)

self._cleanup()

def _cleanup(self):

self.request = None

kbengine-httpserver/httpserver/app.py at master · kenny67/kbengine-httpserver · GitHub

Expand file tree

Breadcrumbs

app.py

Latest commit

History

Breadcrumbs

app.py

File metadata and controls