1. 程式人生 > 程式設計 >python如何實現圖片壓縮

python如何實現圖片壓縮

本工具是通過將圖片上傳到第三方網站tinypng,進行壓縮後下載,覆蓋本地圖片,tinypng是一個強大的圖片處理網站,目前最可靠的無失真壓縮網站。

程式碼如下:

import requests
from idna import unicode
from selenium import webdriver
import time
import os

browser = webdriver.Firefox(executable_path='/Users/lyf/Library/Google/geckodriver')



def tiny_png(url):
  # browser.get('https://tinypng.com/')
  upload_file = browser.find_element_by_tag_name("input")
  try:
    upload_file.send_keys(url)
    browser.implicitly_wait(20)
    a = browser.find_element_by_link_text('download')
    img_url = a.get_attribute('href')
    print(img_url)
    r = requests.get(img_url)
    with open(url,'wb') as f:
      f.write(r.content)
    browser.refresh()
    time.sleep(2)
  except Exception as e:
    print(e)


def is_need_compress(img_path):
  """
  判斷是否需要壓縮處理 >10k 進行壓縮處理
  :param img_path:
  :return:
  """
  if img_path.endswith('.jpg') or img_path.endswith('.png'):
    size = os.path.getsize(img_path) / 1024
    if size > 10.0:
      print('檔案大小:%sk' % size)
      return True
  return False


def file_loop(file_path):
  """
  遍歷資料夾
  :param file_path:
  :return:
  """
  files = os.listdir(file_path)
  for fi in files:
    fi_d = os.path.join(file_path,fi)
    if os.path.isdir(fi_d):
      file_loop(fi_d)
    else:
      child_path = os.path.join(file_path,fi_d)
      print(child_path)
      if is_need_compress(child_path):
        tiny_png(child_path)


if __name__ == "__main__":
  file_path = "/Users/lyf/AndroidStudioProjects/fubei/new-fubei-android-2.5-up/app/src/main/assets/www/assets"
  browser.get('https://tinypng.com/')
  file_loop(file_path)

改進版

優化點:

1.遍歷完成本地資料夾再去上傳網站

2.所有圖片壓縮完成再去下載

3.啟動多執行緒下載

4.設定時間為載入完網路就去上傳檔案(非常非常重要,提速N倍)

import requests
from selenium import webdriver
import time
import os
import _thread
import threading
from selenium.webdriver.support import expected_conditions as EC

from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.common.by import By

# browser = webdriver.Firefox(executable_path='/Users/lyf/Library/Google/geckodriver')

browser = None

image_map = {}
compress_list = []

def tiny_png(url):
  """
  開啟網站進行圖片上傳下載
  :param url:
  :return:
  """
  try:
    upload_file = WebDriverWait(browser,10).until(
      EC.presence_of_element_located((By.TAG_NAME,"input"))
    )
    upload_file.send_keys(url)
    a = WebDriverWait(browser,20).until(
      EC.presence_of_element_located((By.LINK_TEXT,"download"))
    )
    img_url = a.get_attribute('href')
    compress_list.remove(url)
    print(img_url)
    image_map[url] = img_url
    _thread.start_new_thread(sleep,(4,))
    print('重新整理網頁')
    browser.refresh()
    time.sleep(2)
  except Exception as e:
    print(e.__str__())
    browser.execute_script('window.stop()')


def sleep(delay):
  """
  一定的時間後 未載入完網頁 只要控制元件加載出來就可以停止網頁載入
  :param delay:
  :return:
  """
  browser.set_page_load_timeout(delay)
  browser.set_script_timeout(delay)


def down_img(file_path,down_url):
  """
  下載圖片覆蓋原地址
  :param file_path:
  :param down_url:
  :return:
  """
  r = requests.get(down_url)
  with open(file_path,'wb') as f:
    f.write(r.content)
  print('下載完成:%s' % down_url)


def is_need_compress(img_path):
  """
  判斷是否需要壓縮處理 >10k 進行壓縮處理
  :param img_path:
  :return:
  """
  if img_path.endswith('.jpg') or img_path.endswith('.png'):
    size = os.path.getsize(img_path) / 1024
    print(img_path)
    print('檔案大小:%sk' % size)
    if size > 5000.0:
      print('*****' * 30)
      print('這麼大的圖片搞笑嗎')
      print(img_path)
      print('*****' * 30)
    if size > 0.0 and size < 10.0:
      return True
  return False


def file_loop(file_path,compress_list):
  """
  遍歷資料夾
  :param file_path:
  :return:
  """
  files = os.listdir(file_path)
  for fi in files:
    fi_d = os.path.join(file_path,fi)
    if os.path.isdir(fi_d):
      file_loop(fi_d,compress_list)
    else:
      child_path = os.path.join(file_path,fi_d)
      if is_need_compress(child_path):
        compress_list.append(child_path)


def down_all():
  """
  下載所有的圖片
  :return:
  """
  thread_list = []
  for k,v in image_map.items():
    print('key:%s value:%s' % (k,v))
    th = threading.Thread(target=down_img,args=(k,v))
    th.start()
    thread_list.append(th)
  for r in thread_list:
    r.join()


def loop_press():
  """
  輪詢獲取下載地址
  :return:
  """
  for url in compress_list:
    tiny_png(url)


def start_browser():
  """
  啟動瀏覽器
  :return:
  """
  global browser
  browser = webdriver.Firefox(executable_path='/Users/lyf/Library/Google/geckodriver')
  _thread.start_new_thread(sleep,(10,))
  print('載入網頁')
  try:
    browser.get('https://tinypng.com/')
  except:
    browser.execute_script('window.stop()')


if __name__ == "__main__":
  start_time = time.time()
  file_path = "/Users/lyf/Desktop/www/assets"
  # 獲取本地所有需要壓縮的圖片
  file_loop(file_path,compress_list)
  print('符合條件的圖片有%s張' % len(compress_list))
  start_browser()
  loop_press()
  while len(compress_list) > 0:
    browser.quit()
    start_browser()
    loop_press()

  # 多執行緒下載拿到所有返回下載的地址
  down_all()

  end = time.time()
  time_m = end - start_time
  print("time: " + str(time_m))
  browser.quit()

以上就是python如何實現圖片壓縮的詳細內容,更多關於python 圖片壓縮的資料請關注我們其它相關文章!