最近在学习Python的socket编程,简易实现了一个tcp下载器的功能。宇宙惯例,先上代码。
一. 客户端代码(client.py)
import socket def main(): # 创建套接字 tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 链接服务器 tcp_socket.connect(("192.168.21.59", 8090)) # 获取下载的文件名字 file_name = input("请输入你想下载的名字:") # 将文件名字发送到服务器 tcp_socket.send(file_name.encode("utf-8")) # 接收服务器发过来的数据 recv_data = tcp_socket.recv(1024) if recv_data: with open("download_" + file_name, "wb") as f: f.write(recv_data) print("文件下载完成") else: print("服务器找不到该文件(%s)" % file_name) # 关闭套接字 tcp_socket.close() if __name__ == '__main__': main()
二. 服务器端代码(server.py)
import socket def send_file(client_socket, client_addr): """ 服务器端发送文件给客户端函数 :param client_socket: 服务器端与客户端建立链接的socket :param client_addr: 客户端地址 """ # 接收客户端需要下载的文件名 file_name = client_socket.recv(1024).decode("utf-8") print("客户端(%s)需要下载文件是:%s" % (str(client_addr), file_name)) file_content = None # 打开文件,读取数据 try: f = open(file_name, "rb") file_content = f.read() f.close() except Exception as ret: print("服务器找不到该文件(%s)" % file_name) # 发送文件给客户端 if file_content: client_socket.send(file_content) def main(): # 创建socket tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 绑定服务器端本地信息(ip,端口) tcp_socket.bind(("", 8090)) # 设置套接字为被动监听模式 tcp_socket.listen(128) while True: # 等待客户端的链接 new_client_socket, client_addr = tcp_socket.accept() # 调用发送文件函数 send_file(new_client_socket, client_addr) # 关闭当前客户端套接字 new_client_socket.close() # 如无客户端请求,关闭服务器端套接字 tcp_socket.close() if __name__ == '__main__': main()
三. TCP介绍
TCP是一种面向连接的、可靠的、基于字节流的传输层通信协议。这种偏理解概念性的东西,我怕我的只言片语会给读者带来误区,我就不赘述了,但是给大家几个传送门去理解这些协议:
在各编程语言里的socket都聚合实现了tcp,udp等协议。
四.程序说明
首先,运行服务器端程序,再运行客户端程序与服务端建立链接。客户端程序发起一个(下载)请求,服务器端会尝试open这个文件名,如果抛出异常,说明服务器端没有客户端尝试下载的文件。如果服务器端存在该文件,服务器端的程序会读写该文件,并发送给客户端,客户端接收到后,写到一个本地的新文件里。
这就是程序的运行流程说明(注释足够详细,相信你应该能看懂)。