【从零学习python 】90. 使用if管理请求路径

2024-02-29 20:35:21 浏览数 (2)

文件结构:

代码语言:javascript复制
├── server.py
├── utils.py
├── pages
    └── index.html
└── templates
    └── info.html

utils.py文件:

代码语言:javascript复制
PAGE_ROOT = './pages'
TEMPLATE_ROOT = './templates'

def load_html(file_name, start_response, root=PAGE_ROOT):
    """
    加载HTML文件时调用的方法
    :param file_name: 需要加载的HTML文件
    :param start_response: 函数,用来设置响应头。如果找到文件,请求头设置为200,否则设置为410
    :param root: HTML文件所在的目录。默认PAGE_ROOT表示静态HTML文件,TEMPLATE_ROOT表示的是模板文件
    :return: 读取HTML文件成功的话,返回HTML文件内容;读取失败提示资源被删除
    """

    file_name = root   file_name

    try:
        file = open(file_name, 'rb')
    except IOError:
        start_response('410 GONE', [('Content-Type', "text/html;charset=utf-8")])
        return ['资源被删除了'.encode('utf-8')]
    else:
        start_response('200 OK', [('Content-Type', "text/html;charset=utf-8")])
        content = file.read()
        return [content]

def load_template(file_name, start_respone, **kwargs):
    """
    加载模板文件
    :param file_name: 需要加载的模板文件名
    :param start_respone: 函数,用来设置响应头。如果找到文件,请求头设置为200,否则设置为410
    :param kwargs: 用来设置模板里的变量
    :return: 读取HTML文件成功的话,返回HTML文件内容;读取失败提示资源被删除
    """

    content = load_html(file_name, start_respone, root=TEMPLATE_ROOT)
    html = content[0].decode('utf-8')

    if html.startswith('<!DOCTYPE html>'):
        return [html.format(**kwargs).encode('utf-8')]
    else:
        return content

service.py文件:

代码语言:javascript复制
from wsgiref.simple_server import make_server
from utils import load_html, load_template

def show_home(start_response):
    return load_html('/index.html', start_response)

def show_test(start_response):
    start_response('200 OK', [('Content-Type', "text/html;charset=utf-8")])
    return ['我是一段普通的文字'.encode('utf-8')]

def show_info(start_response):
    return load_template('/info.html', start_response, name='张三', age=18)

def application(environ, start_response):
    path = environ.get('PATH_INFO')

    # 处理首页请求(加载一个HTML文件)
    if path == '/' or path == '/index.html':
        result = show_home(start_response)
        return result
    # 处理test.html请求(返回一个普通的字符串)
    elif path == '/test.html':
        return show_test(start_response)
    # 处理info.html请求(加载一个模板并且返回)
    elif path == '/info.html':
        return show_info(start_response)
    # 其它请求暂时无法处理,返回404
    else:
        start_response('400 NOT FOUND', [('Content-Type', "text/html;charset=utf-8")])
        return ['页面未找到'.encode('utf-8')]

httpd = make_server('', 8000, application)
print("Serving HTTP on port 8000...")
httpd.serve_forever()

0 人点赞