Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2025-12-15 14:22:55 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

✨ Add optional proxy support to image hash download function

Browse Source
This commit is contained in:
meng-luo 2024-10-21 23:44:03 +08:00
parent 4a84666075
commit 275e7af5dc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
zhenxun/utils/image_utils.py
View File

@ -390,7 +390,7 @@ def get_img_hash(image_file: str | Path) -> str:
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:
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")
return str(img_hash)