python如何實現圖片壓縮
阿新 • • 發佈:2020-09-14
本工具是通過將圖片上傳到第三方網站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 圖片壓縮的資料請關注我們其它相關文章!