爱站程序员基地前言
套接字(Sockets)是双向通信信道的端点。 套接字可以在一个进程内，在同一机器上的进程之间，或者在不同主机的进程之间进行通信，主机可以是任何一台有连接互联网的机器。
套接字可以通过多种不同的通道类型实现：Unix域套接字，TCP，UDP等。 套接字库提供了处理公共传输的特定类，以及一个用于处理其余部分的通用接口。
socket模块：
要创建套接字，必须使用套接字模块中的socket.socket()函数，该函数具有一般语法
s = socket.socket (socket_family, socket_type, protocol = 0)
| 参数 | 描述 |
|---|---|
| socket_family | 它的值可以是：AF_UNIX或AF_INET，如前所述。 |
| socket_type | 它的值可以是：SOCK_STREAM或SOCK_DGRAM。 |
| protocol | 这通常被省略，默认为0。 |
常用方法:
| 序号 | 方法 | 描述 |
|---|---|---|
| 1 | s.bind() | 此方法将地址(主机名，端口号对)绑定到套接字。 |
| 2 | s.recvfrom() | 此方法接收UDP消息，返回值是一对（字节， 地址） ，其中字节是代表接收到的数据的字节对象，而地址是发送数据的套接字的地址 |
| 3 | s.sendto() | 此方法发送UDP消息，将数据发送到套接字。该套接字不应连接到远程套接字，因为目标套接字是由address指定的 |
| 4 | s.close() | 此方法关闭套接字，套接字对象上所有以后的操作都将失败。远端将不再接收任何数据（在清除排队的数据之后）。套接字在被垃圾回收时会自动关闭 |
| 5 | s.gethostname() | 返回主机名，返回一个字符串，其中包含当前正在执行Python解释器的计算机的主机名。 |
示例1
服务器端
#sever.pyimport sockets = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)host = socket.gethostname()port = 8088s.bind((host,port))try:while True:receive_data,addr = s.recvfrom(1024)print(\"来自服务器\" + str(addr) + \"的消息:\")print(receive_data.decode(\'utf-8\'))msg = input(\'please input send to msg:\')s.sendto(msg.encode(\'utf-8\'),addr)except:s.close()
客户端
#client.pyimport sockets = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)try:while True:host = socket.gethostname()port = 8088send_data = input(\'please input msg:\')s.sendto(send_data.encode(\'utf-8\'),(host,port))msg,addr = s.recvfrom(1024)print(\"来自服务器\" + str(addr) + \"的消息:\")print(msg.decode(\'utf-8\'))except:s.close()
服务端示例
客户端示例
简易的UDP聊天实现了,下面我们来优化一下示例。
示例2
服务端：
#server.pyimport socketimport loggingdef main():s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) # 创建socket对象addr = (\'127.0.0.1\', 9999)s.bind(addr) # 绑定地址和端口logging.info(\'UDP Server on %s:%s...\', addr[0], addr[1])user = {} # 存放字典{addr:name}while True:try:data, addr = s.recvfrom(1024) # 等待接收客户端消息存放在2个变量da1cbfta和addr里if not addr in user: # 如果addr不在user字典里则执行以下代码for address in user: # 从user遍历数据出来addresss.sendto(data + \' 进入聊天室...\'.encode(\'utf-8\'), address) # 发送user字典的data和address到客户端user[addr] = data.decode(\'utf-8\') # 接收的消息解码成utf-8并存在字典user里,键名定义为addrcontinue # 如果addr在user字典里，跳过本次循环if \'EXIT\'.lower() in data.decode(\'utf-8\'):#如果EXIT在发送的data里name = user[addr] #user字典addr键对应的值赋值给变量nameuser.pop(addr) #删除user里的addrfor address in user: #从user取出addresss.sendto((name + \' 离开了聊天室...\').encode(), address) #发送name和address到客户端else:print(\'\"%s\" from %s:%s\' %(data.decode(\'utf-8\'), addr[0], addr[1]))for address in user: #从user遍历出addressif address != addr: #address不等于addr时间执行下面的代码s.sendto(data, address) #发送data和address到客户端except ConnectionResetError:logging.warning(\'Someone left unexcept.\')if __name__ == \'__main__\':main()
客户端：
#clinet.pyimport socketimport threadingdef recv(sock, addr):\'\'\'一个UDP连接在接收消息前必须要让系统知道所占端口也就是需要send一次，否则win下会报错\'\'\'sock.sendto(name.encode(\'utf-8\'), addr)while True:data = sock.recv(1024)print(data.decode(\'utf-8\'))def send(sock, addr):\'\'\'发送数据的方法参数：sock：定义一个实例化socket对象server：传递的服务器IP和端口\'\'\'while True:string = input(\'\')message = name + \' : \' + stringdata = message.encode(\'utf-8\')sock.sendto(data, addr)if string.lower() == \'EXIT\'.lower():breakdef main():\'\'\'主函数执行方法，通过多线程来实现多个客户端之间的通信\'\'\'s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)server = (\'127.0.0.1\', 9999)tr = threading.Thread(target=recv, args=(s, server), daemon=True)ts = threading.Thread(target=send, args=(s, server))tr.start()ts.start()ts.join()s.close()if __name__ == \'__main__\':print(\"-----欢迎来到聊天室,退出聊天室请输入\'EXIT(不分大小写)\'-----\")name = input(\'请输入你的名称:\')print(\'-----------------%s------------------\' % name)main()
支持多人的简易聊天室示例，多个客户端通过一个服务器进行之间通信
