快捷導(dǎo)航

python基于TCP實(shí)現(xiàn)的文件下載器功能案例

更新時(shí)間：2019年12月10日 08:45:19 作者：小飛俠v科比

這篇文章主要介紹了python基于TCP實(shí)現(xiàn)的文件下載器功能,結(jié)合具體實(shí)例形式分析了Python使用socket模塊實(shí)現(xiàn)的tcp協(xié)議下載功能客戶端與服務(wù)器端相關(guān)操作技巧,需要的朋友可以參考下

本文實(shí)例講述了python基于TCP實(shí)現(xiàn)的文件下載器功能。分享給大家供大家參考，具體如下：

服務(wù)器參考代碼如下:

from socket import *
import sys
def get_file_content(file_name):
  """獲取文件的內(nèi)容"""
  try:
    with open(file_name, "rb") as f:
      content = f.read()
    return content
  except:
    print("沒(méi)有下載的文件:%s" % file_name)
def main():
  if len(sys.argv) != 2:
    print("請(qǐng)按照如下方式運(yùn)行：python3 xxx.py 7890")
    return
  else:
    # 運(yùn)行方式為python3 xxx.py 7890
    port = int(sys.argv[1])
  # 創(chuàng)建socket
  tcp_server_socket = socket(AF_INET, SOCK_STREAM)
  # 本地信息
  address = ('', port)
  # 綁定本地信息
  tcp_server_socket.bind(address)
  # 將主動(dòng)套接字變?yōu)楸粍?dòng)套接字
  tcp_server_socket.listen(128)
  while True:
    # 等待客戶端的鏈接，即為這個(gè)客戶端發(fā)送文件
    client_socket, clientAddr = tcp_server_socket.accept()
    # 接收對(duì)方發(fā)送過(guò)來(lái)的數(shù)據(jù)
    recv_data = client_socket.recv(1024) # 接收1024個(gè)字節(jié)
    file_name = recv_data.decode("utf-8")
    print("對(duì)方請(qǐng)求下載的文件名為:%s" % file_name)
    file_content = get_file_content(file_name)
    # 發(fā)送文件的數(shù)據(jù)給客戶端
    # 因?yàn)楂@取打開(kāi)文件時(shí)是以rb方式打開(kāi)，所以file_content中的數(shù)據(jù)已經(jīng)是二進(jìn)制的格式，因此不需要encode編碼
    if file_content:
      client_socket.send(file_content)
    # 關(guān)閉這個(gè)套接字
    client_socket.close()
  # 關(guān)閉監(jiān)聽(tīng)套接字
  tcp_server_socket.close()
if __name__ == "__main__":
  main()

客戶端參考代碼如下:

from socket import *
def main():
  # 創(chuàng)建socket
  tcp_client_socket = socket(AF_INET, SOCK_STREAM)
  # 目的信息
  server_ip = input("請(qǐng)輸入服務(wù)器ip:")
  server_port = int(input("請(qǐng)輸入服務(wù)器port:"))
  # 鏈接服務(wù)器
  tcp_client_socket.connect((server_ip, server_port))
  # 輸入需要下載的文件名
  file_name = input("請(qǐng)輸入要下載的文件名：")
  # 發(fā)送文件下載請(qǐng)求
  tcp_client_socket.send(file_name.encode("utf-8"))
  # 接收對(duì)方發(fā)送過(guò)來(lái)的數(shù)據(jù)，最大接收1024個(gè)字節(jié)（1K）
  recv_data = tcp_client_socket.recv(1024)
  # print('接收到的數(shù)據(jù)為:', recv_data.decode('utf-8'))
  # 如果接收到數(shù)據(jù)再創(chuàng)建文件，否則不創(chuàng)建
  if recv_data:
    with open("[接收]"+file_name, "wb") as f:
      f.write(recv_data)
  # 關(guān)閉套接字
  tcp_client_socket.close()
if __name__ == "__main__":
  main()

更多關(guān)于Python相關(guān)內(nèi)容可查看本站專題：《Python Socket編程技巧總結(jié)》、《Python數(shù)據(jù)結(jié)構(gòu)與算法教程》、《Python函數(shù)使用技巧總結(jié)》、《Python字符串操作技巧匯總》、《Python入門與進(jìn)階經(jīng)典教程》及《Python文件與目錄操作技巧匯總》

希望本文所述對(duì)大家Python程序設(shè)計(jì)有所幫助。

您可能感興趣的文章: