Python socket實現(xiàn)的文件下載器功能示例

import socket
def main():
  # 創(chuàng)建套接字
  tcp_socket = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
  # 獲取服務(wù)器ip和port
  dest_ip = input("請輸入下載服務(wù)器的ip：")
  dest_port = int(input("請輸入下載服務(wù)器的port："))
  # 連接服務(wù)器
  tcp_socket.connect((dest_ip,dest_port))
  # 獲取下載的文件名稱
  download_file_name = input("請輸入要下載的文件名稱：")
  # 將文件名稱發(fā)送到服務(wù)器
  tcp_socket.send(download_file_name.encode("gbk"))
  # 接收文件中的數(shù)據(jù)
  recv_data = tcp_socket.recv(1024) # 1024=1k 1024*1024=1M
  if recv_data:
    # 保存接收到的數(shù)據(jù)到新文件
    with open("[新]"+download_file_name.wb) as f:
      f.write(recv_data)
  # 關(guān)閉套接字
  tcp_socket.close()
if __name__ == '__main__':
  main()

import socket
def send_file_2_client(new_client_socket,client_addr):
  # 1.接收客戶端需要下載的文件名
  file_name = new_client_socket.recv(1024).decode("gbk")
  print("客戶端(%s)需要下載的文件是：%s" %(str(client_addr),file_name))
  file_content = None
  # 2.打開這個文件，讀取數(shù)據(jù)
  try:
    f = open(file_name,"rb")
    file_content = f.read()
    f.close()
  except Exception as ret:
    print("沒有要下載的文件（%s）" % file_name)
  # 3.發(fā)送文件數(shù)據(jù)給客戶端
  if file_content:
    new_client_socket.send(file_content)
def main():
  # 創(chuàng)建套接字
  tcp_server_socket = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
  # 綁定本地信息
  tcp_server_socket.bind("",7890)
  # 讓默認套接字由主動變?yōu)楸粍觢isten
  tcp_server_socket.listen(128)
  while True:
    # 等待客戶端的連接 accept
    new_client_socket ,client_addr = tcp_server_socket.accept()
    # 調(diào)用發(fā)送文件函數(shù)，完成客戶端的服務(wù)
    send_file_2_client(new_client_socket,client_addr)
    # 關(guān)閉套接字
    new_client_socket.close()
  tcp_server_socket.close()
if __name__ == "__main__":
  main()