Viết phần mềm tải file ảnh trong link URL được nhập bằng python

import os
import requests
from bs4 import BeautifulSoup
from urllib.parse import urlparse, urljoin

def download_images(url):
    # Kiểm tra định dạng link nhập vào
    if not url.startswith(('http://', 'https://')):
        print("Định dạng link không hợp lệ. Vui lòng nhập link bắt đầu bằng 'http://' hoặc 'https://'")
        return

    # Tạo thư mục nếu chưa tồn tại
    folder_name = 'taihinh'
    if not os.path.exists(folder_name):
        os.makedirs(folder_name)

    # Lấy nội dung của trang web
    response = requests.get(url)
    soup = BeautifulSoup(response.content, 'html.parser')

    # Tạo biến để thống kê số lượng hình và tổng dung lượng
    count_images = 0
    total_size = 0

    # Tìm và tải hình ảnh từ thẻ img
    img_tags = soup.find_all('img')
    for img_tag in img_tags:
        img_url = img_tag.get('src')
        if img_url:
            img_url = urljoin(url, img_url)
            count_images, total_size = download_image(url, img_url, folder_name, count_images, total_size)

    # Tìm và tải hình ảnh từ thẻ a
    a_tags = soup.find_all('a')
    for a_tag in a_tags:
        img_url = a_tag.get('href')
        if img_url.lower().endswith(('.png', '.jpg', '.jpeg', '.gif')):
            img_url = urljoin(url, img_url)
            count_images, total_size = download_image(url, img_url, folder_name, count_images, total_size)

    # Hiển thị thông tin sau khi đã tải xong
    print(f"-----------------")
    print(f"Đã tải xuống {count_images} hình ảnh.")
    print(f"Tổng dung lượng: {total_size / (1024 * 1024):.2f} MB")
    print(f"Phần mềm tải hình được phát triển bởi GizaData.vn")
    print(f"-----------------")

def download_image(base_url, img_url, folder_name, count_images, total_size):
    img_name = os.path.basename(urlparse(img_url).path)
    img_path = os.path.join(folder_name, img_name)

    # Kiểm tra xem hình đã được tải trước đó chưa
    if not os.path.exists(img_path):
        # Tải hình ảnh xuống thư mục
        with open(img_path, 'wb') as img_file:
            img_response = requests.get(img_url)
            img_file.write(img_response.content)
            print(f"Tải xuống {img_url} thành công!")
            
            # Cập nhật số lượng hình và tổng dung lượng
            count_images += 1
            total_size += len(img_response.content)

    return count_images, total_size

def main():
    while True:
        link = input("Nhập link cần tải hình ảnh : ")
        if link.lower() == 'q':
            break
        download_images(link)

if __name__ == "__main__":
    main()

pip install beautifulsoup4
pip install pyinstaller
pip install requests

pyinstaller --onefile taihinh.py

import os
import requests
from bs4 import BeautifulSoup
from urllib.parse import urlparse, urljoin
import zipfile
import datetime
import random
import string

def download_images(url):
    # Kiểm tra định dạng link nhập vào
    if not url.startswith(('http://', 'https://')):
        print("Định dạng link không hợp lệ. Vui lòng nhập link bắt đầu bằng 'http://' hoặc 'https://'")
        return

    # Tạo thư mục nếu chưa tồn tại
    folder_name = 'taihinh'
    if not os.path.exists(folder_name):
        os.makedirs(folder_name)

    # Lưu trữ các URL đã tải xuống
    downloaded_urls = set()

    # Lấy nội dung của trang web
    response = requests.get(url)
    soup = BeautifulSoup(response.content, 'html.parser')

    # Tạo biến để thống kê số lượng hình và tổng dung lượng
    count_images = 0
    total_size = 0

    # Tìm và tải hình ảnh từ thẻ img và a
    img_tags = soup.find_all(['img', 'a'])
    for tag in img_tags:
        if tag.name == 'img':
            img_url = tag.get('src')
        elif tag.name == 'a':
            img_url = tag.get('href')

        if img_url and (img_url.lower().endswith(('.png', '.jpg', '.jpeg', '.gif'))) and img_url not in downloaded_urls:
            img_url = urljoin(url, img_url)
            img_name = os.path.basename(urlparse(img_url).path)
            img_path = os.path.join(folder_name, img_name)

            # Tải hình ảnh xuống thư mục
            with open(img_path, 'wb') as img_file:
                img_response = requests.get(img_url)
                img_file.write(img_response.content)
                print(f"Tải xuống {img_url} thành công!")
                
                # Cập nhật số lượng hình và tổng dung lượng
                count_images += 1
                total_size += len(img_response.content)

                # Thêm URL đã tải vào danh sách
                downloaded_urls.add(img_url)

    # Nén các hình ảnh đã tải vào tệp tin zip
    zip_file_name = create_zip_file(downloaded_urls)

    # Hiển thị thông tin sau khi đã tải xong và tạo file zip
    print(f"-----------------")
    print(f"Đã tải xuống {count_images} hình ảnh.")
    print(f"Tổng dung lượng: {total_size / (1024 * 1024):.2f} MB")
    print(f"File zip được tạo: {zip_file_name}")
    print(f"Phần mềm tải ảnh được phát triển bởi Gizadata.vn")
    print(f"-----------------")

def create_zip_file(downloaded_urls):
    if downloaded_urls:
        # Tạo tên file zip dựa trên ngày tháng năm
        current_time = datetime.datetime.now().strftime("%d-%m-%Y")
        
        # Tạo chuỗi ngẫu nhiên gồm 5 ký tự
        random_suffix = ''.join(random.choices(string.ascii_lowercase + string.digits, k=5))
        
        # Tạo tên file zip với ngày tháng năm và 5 ký tự ngẫu nhiên
        zip_file_name = f"{current_time}-{random_suffix}.zip"

        # Nén các hình ảnh đã tải vào file zip
        with zipfile.ZipFile(zip_file_name, 'w') as zipf:
            for url in downloaded_urls:
                img_name = os.path.basename(urlparse(url).path)
                img_path = os.path.join('taihinh', img_name)
                zipf.write(img_path, arcname=img_name)

        return zip_file_name

def main():
    while True:
        link = input("Nhập link cần tải hình ảnh : ")
        if link.lower() == 'q':
            break
        download_images(link)

if __name__ == "__main__":
    main()