"""使用模塊線程方式實現網絡資源的下載
# 實現文件下載, 期間顯示文件信息&下載進度
# 控制臺運行以顯示進度
"""
import requests
import os.path as op
import os
from sys import stdout


def downloadfile(url, filename):
  """下載文件并顯示過程
  :param url: 資源地址
  :param filename: 保存的名字, 保存在當前目錄
  """
  # print(url)
  filename = filename + '.' + op.splitext(url)[-1]
  file_to_save = op.join(os.getcwd(), filename)
  # print(file_to_save)

  with open(file_to_save, "wb") as fw:
    with requests.get(url, stream=True) as r:
      # 此時只有響應頭被下載
      # print(r.headers)
      print("下載文件基本信息:")
      print('-' * 30)
      print("文件名稱:", filename)
      print("文件類型:", r.headers["Content-Type"])
      filesize = r.headers["Content-Length"]
      print("文件大小:", filesize, "bytes")
      print("下載地址:", url)
      print("保存路徑:", file_to_save)
      print('-' * 30)
      print("開始下載")

      chunk_size = 128
      times = int(filesize) // chunk_size
      show = 1 / times
      show2 = 1 / times
      start = 1
      for chunk in r.iter_content(chunk_size):
        fw.write(chunk)
        if start <= times:
          stdout.write(f"下載進度: {show:.2%}\r")
          start += 1
          show += show2
        else:
          stdout.write("下載進度: 100%")
      print("\n結束下載")


if __name__ == "__main__":
  downloadfile("https://code.jquery.com/jquery-3.4.1.js", "a")