import socket
import os
import threading
# 處理客戶端請(qǐng)求下載文件的操作（從主線程提出來(lái)的代碼）
def deal_client_request(ip_port, service_client_socket):
  # 連接成功后，輸出“客戶端連接成功”和客戶端的ip和端口
  print("客戶端連接成功", ip_port)
  # 接收客戶端的請(qǐng)求信息
  file_name = service_client_socket.recv(1024)
  # 解碼
  file_name_data = file_name.decode("utf-8")
  # 判斷文件是否存在
  if os.path.exists(file_name_data):
    #輸出文件字節(jié)數(shù)
    fsize = os.path.getsize(file_name_data)
    #轉(zhuǎn)化為兆單位
    fmb = fsize/float(1024*1024)
    #要傳輸?shù)奈募畔?
    senddata = "文件名：%s 文件大?。?.2fMB"%(file_name_data,fmb)
    #發(fā)送和打印文件信息
    service_client_socket.send(senddata.encode("utf-8"))
    print("請(qǐng)求文件名：%s 文件大小：%.2f MB"%(file_name_data,fmb))
    #接受客戶是否需要下載
    options = service_client_socket.recv(1024)
    if options.decode("utf-8") == "y":
      # 打開(kāi)文件
      with open(file_name_data, "rb") as f:
        #　計(jì)算總數(shù)據(jù)包數(shù)目
        nums = fsize/1024 
        #　當(dāng)前傳輸?shù)臄?shù)據(jù)包數(shù)目
        cnum = 0
        while True:
          file_data = f.read(1024)
          cnum = cnum + 1
          jindu = cnum/nums*100
          print("當(dāng)前已下載：%.2f%%"%jindu,end = "\r")
          if file_data:
            # 只要讀取到數(shù)據(jù)，就向客戶端進(jìn)行發(fā)送
            service_client_socket.send(file_data)
          # 數(shù)據(jù)讀完，退出循環(huán)
          else:
            print("請(qǐng)求的文件數(shù)據(jù)發(fā)送完成")
            break
    else:
      print("下載取消！")
  else:
    print("下載的文件不存在！")
  # 關(guān)閉服務(wù)當(dāng)前客戶端的套接字
  service_client_socket.close()
if __name__ == '__main__':
  # 把工作目錄切換到data目錄下
  os.chdir("./data")
  # 創(chuàng)建套接字
  tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  # 綁定端口號(hào)
  tcp_server_socket.bind(("", 3356))
  # 設(shè)置監(jiān)聽(tīng)，將主動(dòng)套接字變?yōu)楸粍?dòng)套接字
  tcp_server_socket.listen(128)
  # 循環(huán)調(diào)用accept，可以支持多個(gè)客戶端同時(shí)連接，和多個(gè)客戶端同時(shí)下載文件
  while True:
    service_client_socket, ip_port = tcp_server_socket.accept()
    # 連接成功后打印套接字號(hào)
    #print(id(service_client_socket))
    # 創(chuàng)建子線程
    sub_thread = threading.Thread(target=deal_client_request, args=(ip_port, service_client_socket))
    # 啟動(dòng)子線程
    sub_thread.start()

# 多任務(wù)文件下載器客戶端
import socket
if __name__ == '__main__':
  # 創(chuàng)建套接字
  tcp_client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  # 和服務(wù)端連接
  server_ip = input("輸入服務(wù)器IP：")
  tcp_client_socket.connect((server_ip, 3356))
  # 發(fā)送下載文件的請(qǐng)求
  file_name = input("請(qǐng)輸入要下載的文件名：")
  # 編碼
  file_name_data = file_name.encode("utf-8")
  # 發(fā)送文件下載請(qǐng)求數(shù)據(jù)
  tcp_client_socket.send(file_name_data)
  #　接收要下載的文件信息
  file_info = tcp_client_socket.recv(1024)
  # 文件信息解碼
  info_decode = file_info.decode("utf-8")
  print(info_decode)
  #獲取文件大小
  fileszie = float(info_decode.split('：')[2].split('MB')[0])
  fileszie2 = fileszie*1024
  # 是否下載？輸入ｙ　確認(rèn)　輸入ｑ 取消
  opts = input("是否下載？(y 確認(rèn)　q 取消)")
  if opts == 'q':
    print("下載取消！程序退出")
  else:
    print("正在下載　》》》")
    #向服務(wù)器確認(rèn)正在下載
    tcp_client_socket.send(b'y')
    # 把數(shù)據(jù)寫(xiě)入到文件里
    with open("./" + file_name, "wb") as file:
      #目前接收到的數(shù)據(jù)包數(shù)目
      cnum = 0
      while True:
        # 循環(huán)接收文件數(shù)據(jù)
        file_data = tcp_client_socket.recv(1024)
        # 接收到數(shù)據(jù)
        if file_data:
          # 寫(xiě)入數(shù)據(jù)
          file.write(file_data)
          cnum = cnum+1
          jindu =cnum/fileszie2*100
          print("當(dāng)前已下載：%.2f%%"%jindu,end = "\r")
        # 接收完成
        else:
          print("下載結(jié)束！")
          break
  # 關(guān)閉套接字
  tcp_client_socket.close()