# _*_ coding: utf-8 _*_ """ Socket编程 """ import sys import socket def server_func(port): """ 服务端 """ # 1. 创建socket对象 server = socket.socket() # 2. 绑定ip和端口 server.bind(("127.0.0.1", port)) # 3. 监听是否有客户端连接 server.listen(10) print("服务端已经启动%s端口......" % port) # 4. 接收客户端连接 sock_obj, address = server.accept() sock_obj.settimeout(3) print("客户端：%s，超时时间：%s" % (address, sock_obj.gettimeout())) while True: try: # 5. 接收客户端发送的消息 recv_data = sock_obj.recv(1024).decode("utf-8") print("客户端端 -> 服务端: %s" % recv_data) if recv_data == "quit": break # 6. 给客户端回复消息 send_data = "received[%s]" % recv_data sock_obj.send(send_data.encode("utf-8")) print("服务端 -> 客户端: %s" % send_data) except Exception as excep: print("error: ", excep) # 7. 关闭socket对象 sock_obj.close() server.close() def client_func(port): """ 客户端 """ # 1. 创建客户端的socket对象 client = socket.socket() # 2. 连接服务端， 需要指定端口和IP client.connect(("127.0.0.1", port)) while True: # 3. 给服务端发送数据 send_data = input("客户端>").strip() client.send(send_data.encode("utf-8")) if send_data == "quit": break # 4. 获取服务端返回的消息 recv_data = client.recv(1024).decode("utf-8") print("服务端 -> 客户端: %s" % recv_data) # 5. 关闭socket连接 client.close() if __name__ == '__main__': flag = sys.argv[1] if flag == "server": server_func(9901) else: client_func(9901)