PYTHON DOWNLOAD PICTURE

  • 207
  • 0

 PYTHON DOWNLOAD PICTURE

import requests
import os
def download_image(url, save_directory):
   try:
       response = requests.get(url, stream=True)
       if response.status_code == 200:
           file_name = url.split("/")[-1]
           save_path = os.path.join(save_directory, file_name)
           with open(save_path, "wb") as file:
               total_length = response.headers.get('content-length')
               if total_length is None:
                   file.write(response.content)
               else:
                   total_length = int(total_length)
                   downloaded_length = 0
                   for chunk in response.iter_content(chunk_size=1024):
                       downloaded_length += len(chunk)
                       file.write(chunk)
                       progress = downloaded_length / total_length * 100
                       print(f"Downloading: {progress:.2f}%\r", end='')
                   print("\nImage downloaded successfully")
       else:
           print("Failed to download the image")
   except Exception as e:
       print(f"An error occurred: {e}")
def main():
   print("Welcome to the Image Downloader!")
   url = input("Please enter the image URL: ")
   save_directory = input("Please enter the save directory: ")
   
   try:
       download_image(url, save_directory)
   except Exception as e:
       print(f"An error occurred: {e}")
main()