1. 程式人生 > 程式設計 >Python socket實現的檔案下載器功能示例

Python socket實現的檔案下載器功能示例

本文例項講述了Python socket實現的檔案下載器功能。分享給大家供大家參考,具體如下:

檔案下載器

先寫客戶端再寫服務端

1.tcp下載器客戶端

import socket
def main():
  # 建立套接字
  tcp_socket = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
  # 獲取伺服器ip和port
  dest_ip = input("請輸入下載伺服器的ip:")
  dest_port = int(input("請輸入下載伺服器的port:"))
  # 連線伺服器
  tcp_socket.connect((dest_ip,dest_port))
  # 獲取下載的檔名稱
  download_file_name = input("請輸入要下載的檔名稱:")
  # 將檔名稱傳送到伺服器
  tcp_socket.send(download_file_name.encode("gbk"))
  # 接收檔案中的資料
  recv_data = tcp_socket.recv(1024) # 1024=1k 1024*1024=1M
  if recv_data:
    # 儲存接收到的資料到新檔案
    with open("[新]"+download_file_name.wb) as f:
      f.write(recv_data)
  # 關閉套接字
  tcp_socket.close()
if __name__ == '__main__':
  main()

2.tcp下載器服務端

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.開啟這個檔案,讀取資料
  try:
    f = open(file_name,"rb")
    file_content = f.read()
    f.close()
  except Exception as ret:
    print("沒有要下載的檔案(%s)" % file_name)
  # 3.傳送檔案資料給客戶端
  if file_content:
    new_client_socket.send(file_content)
def main():
  # 建立套接字
  tcp_server_socket = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
  # 繫結本地資訊
  tcp_server_socket.bind("",7890)
  # 讓預設套接字由主動變為被動listen
  tcp_server_socket.listen(128)
  while True:
    # 等待客戶端的連線 accept
    new_client_socket,client_addr = tcp_server_socket.accept()
    # 呼叫傳送檔案函式,完成客戶端的服務
    send_file_2_client(new_client_socket,client_addr)
    # 關閉套接字
    new_client_socket.close()
  tcp_server_socket.close()
if __name__ == "__main__":
  main()

更多關於Python相關內容可檢視本站專題:《Python Socket程式設計技巧總結》、《Python資料結構與演算法教程》、《Python函式使用技巧總結》、《Python字串操作技巧彙總》、《Python入門與進階經典教程》及《Python檔案與目錄操作技巧彙總》

希望本文所述對大家Python程式設計有所幫助。