Python網絡編程之使用TCP方式傳輸文件操作示例

# 導入模塊
import socket
# 創(chuàng)建套接字
tcp_client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 綁定端口
tcp_client_socket.bind(("", 8080))
# 連接IP地址和端口
tcp_client_socket.connect(("192.168.43.216", 8080))
file_name = input("請輸入要下載的文件：\n")
# 文件名編碼
tcp_client_socket.send(file_name.encode())
try:
  # 文件傳輸
  with open("C:/Users/Administrator/Desktop/" + file_name, "wb") as file:
    while True:
      # 接收數(shù)據(jù)
      file_data = tcp_client_socket.recv(1024)
      # 數(shù)據(jù)長度不為0寫入文件
      if file_data:
        file.write(file_data)
      # 數(shù)據(jù)長度為0表示下載完成
      else:
        break
# 下載出現(xiàn)異常時捕獲異常
except Exception as e:
  print("下載異常", e)
# 無異常則下載成功
else:
  print(file_name, "下載成功")
# 關閉客戶端
tcp_client_socket.close()

# 導入模塊
import socket
# 創(chuàng)建tcp服務端socket
tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 綁定端口
tcp_server_socket.bind(("", 8080))
# 設置監(jiān)聽，把服務端socket由主動套接字改成被動套接字，只能接收客戶端的連接請求
tcp_server_socket.listen(128)
while True:
  # 接收客戶端信息
  client_socket, client_ip = tcp_server_socket.accept()
  print("客戶端：", client_ip, "連接")
  # 接收下載信息
  file_name_data = client_socket.recv(1024)
  # 解碼下載信息
  file_name = file_name_data.decode()
  try:
    # 數(shù)據(jù)傳輸
    with open("/home/python/Desktop/" + file_name, "rb") as file:
      while True:
        # 讀取文件數(shù)據(jù)
        file_data = file.read(1024)
        # 數(shù)據(jù)長度不為0表示還有數(shù)據(jù)沒有寫入
        if file_data:
          client_socket.send(file_data)
        # 數(shù)據(jù)為0表示傳輸完成
        else:
          print(file_name, "傳輸成功")
          break
  except Exception as e:
    print("傳輸異常：", e)
  # 關閉客戶端連接
  client_socket.close()