0x01接收路径

函数式的路径判断

截取请求路径的扩展功能

0x02 代码

"""

使用socket 搭建一个简单的web服务端

函数进阶版

"""

import time

import socket

sk = socket.socket()

sk.bind(('127.0.0.1', 8080))

sk.listen()



def yingying(url):

    data = 'ni fang wen de shi: {}'.format(url)

    return bytes(data, encoding='utf8')



def hpg(url):

    data = 'a huang de computer is bei tou le !'

    return bytes(data, encoding='utf8')



def login(url):

    # 返回一个字节类型HTML文件的内容

    with open('login.html', 'r', encoding='utf8') as f:

        data = f.read()

    now = str(time.time())

    data = data.replace('@@xx@@', now)

    return bytes(data, encoding='utf8')



# url和函数的对应关系

url_func = [

    ('/yingying/', yingying),

    ('/hpg/', hpg),

    ('/login/', login),

]



while 1:

    # ------------------- a。建立连接 接收消息部分 --------------------

    conn, addr = sk.accept()

    data = conn.recv(9000)



    # -------------------- b. 对用户发来的消息做处理的部分 -------------

    # 把收到的字节类型的数据转换成字符串

    data_str = str(data, encoding='utf8')

    # 按照课前练习切分字符串，得到请求行

    first_line = data_str.split('\r\n')[0]

    # 对请求行按照空格切割

    url = first_line.split(' ')[1]



    # --------------- c. 业务逻辑处理部分 ---------------------

    # 使用func变量保存将要执行的函数

    for i in url_func:

        if i[0] == url:

            func = i[1]

            break

    else:

        func = None

    # 执行函数

    if func:

        msg = func(url)

    else:

        msg = b'404 not found!'



    # ----------------- d. 回复响应消息 部分 ----------------------

    conn.send(b'HTTP/1.1 200 OK\r\n\r\n')

    conn.send(msg)

    conn.close()

0x03 备注说明

功能单一,可以继续扩展,参数判断, 案例可以以登录为例, 再加上数据库连接和查询的扩展