最近在学习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这个文件名,如果抛出异常,说明服务器端没有客户端尝试下载的文件。如果服务器端存在该文件,服务器端的程序会读写该文件,并发送给客户端,客户端接收到后,写到一个本地的新文件里。
这就是程序的运行流程说明(注释足够详细,相信你应该能看懂)
。