Add optional proxy support to image hash download function

This commit is contained in:
meng-luo 2024-10-21 23:44:03 +08:00
parent 4a84666075
commit 275e7af5dc

View File

@ -390,7 +390,7 @@ def get_img_hash(image_file: str | Path) -> str:
return str(hash_value) return str(hash_value)
async def get_download_image_hash(url: str, mark: str) -> str: async def get_download_image_hash(url: str, mark: str, use_proxy: bool = False) -> str:
"""下载图片获取哈希值 """下载图片获取哈希值
参数: 参数:
@ -402,7 +402,7 @@ async def get_download_image_hash(url: str, mark: str) -> str:
""" """
try: try:
if await AsyncHttpx.download_file( if await AsyncHttpx.download_file(
url, TEMP_PATH / f"compare_download_{mark}_img.jpg", use_proxy=False url, TEMP_PATH / f"compare_download_{mark}_img.jpg", use_proxy=use_proxy
): ):
img_hash = get_img_hash(TEMP_PATH / f"compare_download_{mark}_img.jpg") img_hash = get_img_hash(TEMP_PATH / f"compare_download_{mark}_img.jpg")
return str(img_hash) return str(img_hash)