拉勾教育网络通信Socket模块实现文件传输
实现的效果如下的Gif所示，就是网络通信Socket模块实现文件下载传输。拉勾IT课小编为大家分解
服务端
首先需要获取本机ip，这里服务端采用多线程的方法，就是定义一个函数，然后用threading创建任务。客户端连接成功，接收客户端的请求信息，就是下载的文件名。所以需要判断，有输出文件字节数。然后在问用户是不是要下载，得到信息就使用 while True: 读文件的内容，再一个send。看代码是不是就是这么回事。
import socket
import os
import threading
# 获取本机ip
def get_host_ip():
try:
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
***nnect(('8.8.8.8', 80))
ip = s.getsockname()[0]
finally:
s.close()
return ip
# 处理客户端请求下载文件的操作（从主线程提出来的代码）
def deal_client_request(ip_port, service_client_socket):
# 连接成功后，输出“客户端连接成功”和客户端的ip和端口
print("客户端连接成功", ip_port)
# 接收客户端的请求信息【recv】
file_name = service_client_socket.recv(1024)
# 解码
file_name_data = file_name.decode("utf-8")
# 判断文件是否存在
if os.path.exists(file_name_data):
#输出文件字节数
fsize = os.path.getsize(file_name_data)
#转化为兆单位
fmb = fsize/float(1024*1024)
#要传输的文件信息
senddata = "文件名：%s 文件大小：%.2fMB"%(file_name_data,fmb)
#发送和打印文件信息【send】
service_client_socket.send(senddata.encode("utf-8"))
print("请求文件名：%s 文件大小：%.2f MB"%(file_name_data,fmb))
#接受客户是否需要下载【recv】
options = service_client_socket.recv(1024)
if options.decode("utf-8") == "y":
# 打开文件
with open(file_name_data, "rb") as f:
# 计算总数据包数目
nums = fsize/1024
# 当前传输的数据包数目
cnum = 0
while True:
file_data = f.read(1024)
cnum = cnum + 1
#progress = cnum/nums*100
#print("当前已下载：%.2f%%"%progress,end = "r")
if file_data:
# 只要读取到数据，就向客户端进行发送【send】
service_client_socket.send(file_data)
# 数据读完，退出循环
else:
print("请求的文件数据发送完成")
break
else:
print("下载取消！")
else:
print("下载的文件不存在！")
# 关闭服务当前客户端的套接字【close】
service_client_socket.close()
if __name__ == '__main__':
# 获取本机ip
print("TCP文件传输服务器，本机IP:" + get_host_ip())
# 把工作目录切换到data目录下
os.chdir("./data")
# 创建套接字【socket】
tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 绑定端口号【bind】
tcp_server_socket.bind(("", 3356))
# 设置，将主动套接字变为被动套接字【listen】
tcp_server_socket.listen(128)
# 循环调用【accept】，可以支持多个客户端同时连接，和多个客户端同时下载文件
while True:
service_client_socket, ip_port = tcp_server_socket.accept()
# 连接成功后打印套接字号
#print(id(service_client_socket))
# 创建子线程
sub_thread = threading.Thread(target=deal_client_request, args=(ip_port, service_client_socket))
# 启动子线程
sub_thread.start()
服务端
首先需要获取本机ip，这里服务端采用多线程的方法，就是定义一个函数，然后用threading创建任务。客户端连接成功，接收客户端的请求信息，就是下载的文件名。所以需要判断，有输出文件字节数。然后在问用户是不是要下载，得到信息就使用 while True: 读文件的内容，再一个send。看代码是不是就是这么回事。
import socket
import os
import threading
# 获取本机ip
def get_host_ip():
try:
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
***nnect(('8.8.8.8', 80))
ip = s.getsockname()[0]
finally:
s.close()
return ip
# 处理客户端请求下载文件的操作（从主线程提出来的代码）
def deal_client_request(ip_port, service_client_socket):
# 连接成功后，输出“客户端连接成功”和客户端的ip和端口
print("客户端连接成功", ip_port)
# 接收客户端的请求信息【recv】
file_name = service_client_socket.recv(1024)
# 解码
file_name_data = file_name.decode("utf-8")
# 判断文件是否存在
if os.path.exists(file_name_data):
#输出文件字节数
fsize = os.path.getsize(file_name_data)
#转化为兆单位
fmb = fsize/float(1024*1024)
#要传输的文件信息
senddata = "文件名：%s 文件大小：%.2fMB"%(file_name_data,fmb)
#发送和打印文件信息【send】
service_client_socket.send(senddata.encode("utf-8"))
print("请求文件名：%s 文件大小：%.2f MB"%(file_name_data,fmb))
#接受客户是否需要下载【recv】
options = service_client_socket.recv(1024)
if options.decode("utf-8") == "y":
# 打开文件
with open(file_name_data, "rb") as f:
# 计算总数据包数目
nums = fsize/1024
# 当前传输的数据包数目
cnum = 0
while True:
file_data = f.read(1024)
cnum = cnum + 1
#progress = cnum/nums*100
#print("当前已下载：%.2f%%"%progress,end = "r")
if file_data:
# 只要读取到数据，就向客户端进行发送【send】
service_client_socket.send(file_data)
# 数据读完，退出循环
else:
print("请求的文件数据发送完成")
break
else:
print("下载取消！")
else:
print("下载的文件不存在！")
# 关闭服务当前客户端的套接字【close】
service_client_socket.close()
if __name__ == '__main__':
# 获取本机ip
print("TCP文件传输服务器，本机IP:" + get_host_ip())
# 把工作目录切换到data目录下
os.chdir("./data")
# 创建套接字【socket】
tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 绑定端口号【bind】
tcp_server_socket.bind(("", 3356))
# 设置，将主动套接字变为被动套接字【listen】
tcp_server_socket.listen(128)
# 循环调用【accept】，可以支持多个客户端同时连接，和多个客户端同时下载文件
while True:
service_client_socket, ip_port = tcp_server_socket.accept()
# 连接成功后打印套接字号
#print(id(service_client_socket))
# 创建子线程
sub_thread = threading.Thread(target=deal_client_request, args=(ip_port, service_client_socket))
# 启动子线程
sub_thread.start()
