Python使用socket_TCP实现小文件下载功能

import socket
 
# 1.创建套接字
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 2.绑定端口
server.bind((\'127.0.0.1\', 7890))
# 3.变为被动监听模式
server.listen(4)
while True:
  # 4.等待客户端链接
  s_new, addr = server.accept()
  print(\'用户【{}】已经成功连接！！\'.format(addr[0]))
  while True:
    # 5.接受消息
    try:
      # 若客户端强行断开连接服务器不会崩溃
      recv_data = s_new.recv(1024).decode(\'utf-8\')
      if recv_data:
        print(\'来自【{}】的用户表示要下载： 【{}】\'.format(addr[0], recv_data))
        # 6.发送消息
        try:
          f = open(recv_data, \'rb\')
          content = f.read()
          send_data = s_new.send(content)
          f.close()
          print(\'下载成功！\')
        except Exception as result:
          send_data = s_new.send(\'Non\'.encode(\'utf-8\'))
          print(\'未找到要下载的文件\')
      else:
        print(\'用户已经正常退出！\')
        break
    except Exception as res:
      print(\'用户已经退出连接！\')
      break
  s_new.close()
# 7，关闭套接字
server.close()

import socket
# 创建套接字
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 绑定IP地址
s.connect((\'127.0.0.1\', 7890))
while True:
 
  # 发送数据
  send_data = str(input(\'请输入想要下载的文件名：\'))
  if send_data == \'exit\':
    print(\'已退出下载！\')
    break
  s.send(send_data.encode(\'utf-8\'))
 
  # 接收数据
  recv_data = s.recv(2048)
  if recv_data == \'Non\'.encode(\'utf-8\'):
    print(\'未找到要下载的文件！\')
  else:
    with open(\'【new】\' + send_data, \'wb\') as f:
      f.write(recv_data)
      print(\'[{}]下载成功！\'.format(send_data))
      f.close()
 
# 关闭套接字
s.close()