实现简单的REST接口

最新推荐文章于 2024-07-01 16:15:00 发布

Allen_by

最新推荐文章于 2024-07-01 16:15:00 发布

阅读量1.8k

点赞数

分类专栏： PYTHON网络编程文章标签： REST Python

本文链接：https://blog.csdn.net/boyun58/article/details/85317832

版权

PYTHON网络编程专栏收录该内容

26 篇文章 2 订阅

订阅专栏

有时候想使用一个简单的REST接口通过网络远程控制或访问你的应用程序，但是又不想自己去安装一个完整的web框架。

#resty.py
import cgi


def notfound_404(environ, start_response):
    start_response('404 Not Found', [('Content-type', 'text/plain')])
    return [b'Not Found']


class PathDispatcher:
    def __init__(self):
        self.pathmap = {}

    def __call__(self, environ, start_response):
        path = environ['PATH_INFO']
        params = cgi.FieldStorage(environ['wsgi.input'],
                                  environ=environ)
        method = environ['REQUEST_METHOD'].lower()
        environ['params'] = {key: params.getvalue(key) for key in params}
        handler = self.pathmap.get((method, path), notfound_404)
        return handler(environ, start_response)

    def register(self, method, path, function):
        self.pathmap[method.lower(), path] = function
        return function

#processor.py
import time

_hello_resp = '''\
<html>
  <head>
     <title>Hello {name}</title>
   </head>
   <body>
     <h1>Hello {name}!</h1>
   </body>
</html>'''


def hello_world(environ, start_response):
    start_response('200 OK', [('Content-type', 'text/html')])
    params = environ['params']
    resp = _hello_resp.format(name=params.get('name'))
    yield resp.encode('utf-8')


_localtime_resp = '''\
<?xml version="1.0"?>
<time>
  <year>{t.tm_year}</year>
  <month>{t.tm_mon}</month>
  <day>{t.tm_mday}</day>
  <hour>{t.tm_hour}</hour>
  <minute>{t.tm_min}</minute>
  <second>{t.tm_sec}</second>
</time>'''


def localtime(environ, start_response):
    start_response('200 OK', [('Content-type', 'application/xml')])
    resp = _localtime_resp.format(t=time.localtime())
    yield resp.encode('utf-8')


def welcome(environ, start_response):
    start_response('200 OK', [('Content-type', 'text/html')])
    resp = "Welcome"
    yield resp.encode("utf-8")


if __name__ == '__main__':
    from resty import PathDispatcher
    from wsgiref.simple_server import make_server

    # Create the dispatcher and register functions
    dispatcher = PathDispatcher()
    dispatcher.register('GET', '/hello', hello_world)
    dispatcher.register('GET', '/localtime', localtime)
    dispatcher.register('GET', '/welcome', welcome)


    # Launch a basic server
    httpd = make_server('127.0.0.1', 8080, dispatcher)
    print('Serving on port 8080...')
    httpd.serve_forever()

测试可通过浏览器访问响应的地址以查看效果