Python3.X Socket 一个编码与解码的坑

在看《Python核心编程》第三版 讲述网络编程Socket的知识,在练习中采用Python 3 的代码中遇到一个与编码解码有关的坑,本文将给予详细的介绍。

软件环境

Python: 3.6.0
库: socket

1. 问题初见

仿照书中的代码(中文版 55-56页) 加上自己的一点改动在我的环境中不能运行,总是报这个错误:TypeError: a bytes-like object is required, not 'str'

这里是我的客户端Socket代码

from socket import *
from time import ctime

HOST = 'localhost'
PORT = 10001
ADDRESS = (HOST, PORT)

clientSocket = socket(AF_INET, SOCK_STREAM)
clientSocket.connect(ADDRESS)

while True:
    data = input('请输入消息:')
    if not data:
        break

    clientSocket.send(data)
    data = clientSocket.recv(1024)
    if not data:
        break

    print("服务器返回的消息是:", data.decode('utf-8'))

clientSocket.close()

我的环境是: Python 3.6.0, 怎么破?

2. 研究错误 TypeError: a bytes-like object is required, not 'str'

错误的位置是在代码clientSocket.send(data)部分,但是翻看python socket .send()源代码_socket.py 方法说明

def send(self, data, flags=None): # real signature unknown; restored from __doc__

    send(data[, flags]) -> count

    Send a data string to the socket.  For the optional flags
    argument, see the Unix manual.  Return the number of bytes
    sent; this may be less than len(data) if the network is busy.

    pass

这个send方法的参数期望的是一个 "a data string" 啊,而我确实给了一个string。

哪里出问题了? 继续查看官方文档Socket,发现原因了。

官方对Socket的说明:

socket.send(bytes[, flags])

可以看到在Python 3中send()方法期望的是一个bytes, 而不是str
看来我我前面看到的是假的源代码参数的说明。哈哈。

3. 用encode() 方法解决客户端Socket 发送错误

解决错误的方法就是在调用send()方法之前对字符串类型数据进行encode,将字符串转化成bytes
代码如下:

clientSocket.send(data.encode())

与此同时,在服务端运行的时候也遇到了类似数据无法接收的问题。
如下代码得到的data,是无法直接打印的。
data = clientSocket.recv(1024)

如果要打印data数据的话,也要调用decode()从而将数据从bytes转化为str。

4. encode() 和 decode()

encode()编码 : str -> bytes
decode()解码 : bytes -> str

默认的encoding是 utf-8

更多内容见官方文档:
str.encode()
bytes.decode()

5. 完整Socket代码

服务端:

from socket import *
from time import ctime

HOST = 'localhost'
PORT = 10001
ADDRESS = (HOST, PORT)

serverSocket = socket(AF_INET, SOCK_STREAM)
serverSocket.bind(ADDRESS)
serverSocket.listen(5)

while True:
    print("等待客户端连接...")
    clientSocket, address = serverSocket.accept()
    print(address, "已经成功连接至本服务器")

    while True:
        data = clientSocket.recv(1024)
        if not data:
            break

        replyMsg = data.decode() + "[" + ctime() + ']'
        clientSocket.send(replyMsg.encode())

    clientSocket.close()
serverSocket.close()

客户端:

from socket import *
from time import ctime

HOST = 'localhost'
PORT = 10001
ADDRESS = (HOST, PORT)

clientSocket = socket(AF_INET, SOCK_STREAM)
clientSocket.connect(ADDRESS)

while True:
    data = input('请输入消息:')
    if not data:
        break

    clientSocket.send(data.encode())
    data = clientSocket.recv(1024)
    if not data:
        break

    print("服务器返回的消息是:", data.decode('utf-8'))

clientSocket.close()









本文转自 yuanzhitang 51CTO博客,原文链接:http://blog.51cto.com/yuanzhitang/2060790,如需转载请自行联系原作者
上一篇:通过改写算法获得数据结构学习的更佳效果


下一篇:go| beego 速览