描述:
客户端:
思路:
- 向服务端请求某个文件(eg:1.txt)(将文件名发过去)
- 接收数据
- 如果有数据,就新建一个文件(eg:new-1.txt),将接收到的数据写进文件
- 关闭套接字
代码:
import socket
def main():
# 1.创建套接字
tcp_socket_client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 2.获取服务器的ip port
dest_ip = input("请输入下载服务器的ip:")
dest_port = input("请输入下载服务器的port:")
# 3.连接服务器
tcp_socket_client.connect((dest_ip, dest_port))
# 4.获取下载的文件名字
filename = input("请输入要下载的文件名字:")
# 5.将文件名发送到服务器
tcp_socket_client.send(filename.encode('utf-8'))
# 6.接收文件中的数据
recv_data = tcp_socket_client.recv(1024*1024) # 1M
# 7.保存接收到的数据到一个文件中
if recv_data:
with open("new-"+filename, "wb") as ref: # wb:2进制模式写
ref.write(recv_data)
# 8.关闭套接字
tcp_socket_client.close()
服务端:
思路:
- 接收到一个客户端的连接请求,收到需要下载的文件名(eg:1.txt)
- 找到这个文件(eg:1.txt)
- 读取文件内容,并将文件内容发送回客户端
- 关闭套接字
import socket
def send_file_to_client(new_socket):
# 1.接收所需文件名
filename = new_socket.recv(1024).decode('utf-8')
print("需要下载的文件:", filename)
# 2.模拟打开的文件内容
# # 此处模拟成功。
# new_socket.send("This is content".encode('utf-8'))
# # with的前提是这个文件能打开,所以此处用with不合适
# with open(filename, 'rb') as ref:
# msg = ref.read
# print(msg)
filecontent = None
try:
f = open(filename, 'rb')
filecontent = f.read()
f.close()
except:
print("没有要下载的文件", filename)
# 3.发送数据到客户端
if filecontent:
new_socket.send(filecontent)
def main():
# 1.创建监听套接字
tcp_socket_server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 2.绑定信息
server_ip = "127.0.0.1"
server_port = 7890
tcp_socket_server.bind((server_ip, server_port))
# 3.设为监听模式
tcp_socket_server.listen(128)
while True:
# 4.accept
print("等待连接:")
new_socket, client_addr = tcp_socket_server.accept()
print("连接客户端-", client_addr)
# 5.调用发送文件函数,完成客户端服务
send_file_to_client(new_socket)
# 6.关闭套接字
new_socket.close()
tcp_socket_server.close()
if __name__ == "__main__":
main()