mirror of
https://github.com/wangdage12/Snap.Hutao.git
synced 2026-02-17 15:06:39 +08:00
69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
import hashlib
|
||
import os
|
||
import shutil
|
||
import re
|
||
|
||
def sha1_name(filename):
|
||
"""对输入文件名进行 SHA1 哈希，返回大写 40 位字符串"""
|
||
sha = hashlib.sha1(filename.encode('utf-8')).hexdigest().upper()
|
||
return sha
|
||
|
||
def search_and_copy(original_name, search_dir, output_dir):
|
||
sha_name = sha1_name(original_name)
|
||
print(f"SHA1 计算结果: {sha_name}")
|
||
|
||
# 查找匹配文件
|
||
matched_path = None
|
||
for root, dirs, files in os.walk(search_dir):
|
||
for file in files:
|
||
if file.upper() == sha_name:
|
||
matched_path = os.path.join(root, file)
|
||
break
|
||
|
||
if not matched_path:
|
||
print("❌ 未找到匹配文件！")
|
||
return
|
||
|
||
# 创建输出目录
|
||
os.makedirs(output_dir, exist_ok=True)
|
||
|
||
# 复制并重命名
|
||
# 文件名是url，取最后一部分作为原文件名
|
||
original_filename = os.path.basename(original_name)
|
||
# 创建url中最后一个文件夹作为输出目录
|
||
last_folder = original_name.split('/')[-2]
|
||
output_subdir = os.path.join(output_dir, last_folder)
|
||
os.makedirs(output_subdir, exist_ok=True)
|
||
new_path = os.path.join(output_subdir, original_filename)
|
||
shutil.copy(matched_path, new_path)
|
||
print(f"✔ 已复制并重命名文件: {new_path}")
|
||
|
||
def extract_urls_from_log(log_file):
|
||
"""
|
||
从日志中提取原始文件 URL，并打印到终端
|
||
"""
|
||
base_url = "https://api.snapgenshin.com/static/raw"
|
||
url_pattern = re.compile(r'GET\s+/static/raw/([^/]+)/([^ ]+)\s+HTTP')
|
||
|
||
urls = []
|
||
|
||
with open(log_file, 'r', encoding='utf-8') as f:
|
||
for line in f:
|
||
match = url_pattern.search(line)
|
||
if match:
|
||
category, filename = match.groups()
|
||
full_url = f"{base_url}/{category}/{filename}"
|
||
urls.append(full_url)
|
||
|
||
return urls
|
||
|
||
|
||
if __name__ == "__main__":
|
||
logfile="1.txt" # 日志文件路径
|
||
original_file = extract_urls_from_log(logfile)
|
||
search_directory = "C:\\Users\\username\\AppData\\Local\\Packages\\60568DGPStudio.SnapHutao_wbnnev551gwxy1\\LocalCache\\ImageCache" # 搜索目录
|
||
output_directory = "C:\\Users\\username\\AppData\\Local\\Packages\\60568DGPStudio.SnapHutao_wbnnev551gwxy1\\LocalCache\\ImageCache\\output" # 输出目录
|
||
|
||
for url in original_file:
|
||
search_and_copy(url, search_directory, output_directory)
|