如何为Python循环输出添加两类标题并格式化URL检测结果

嘿，我来帮你把脚本的输出调整得更规整，让安全和恶意URL分别归类到对应标题下展示！

思路很简单：我们先提前创建两个列表分别存储安全和恶意的URL，在循环检测每个URL时，把结果添加到对应的列表里，等所有URL检测完成后，再统一用Markdown格式输出标题和对应的URL列表。

这里是修改后的完整代码示例（假设你已经正确配置了Google Safe Browsing的客户端和API调用逻辑）：

import google.cloud.safebrowsing_v4 as safebrowsing

# 初始化Safe Browsing客户端（替换成你的实际API密钥和配置）
client = safebrowsing.SafeBrowsingClient(api_key="YOUR_API_KEY")

# 初始化两个列表，分别存储安全和恶意URL
safe_urls = []
malicious_urls = []

# 读取文本文件中的URL（假设每行一个URL）
with open("urls.txt", "r") as f:
    urls = [line.strip() for line in f if line.strip()]

# 循环检测每个URL
for url in urls:
    # 构建请求体（根据你的需求调整威胁类型、平台类型等参数）
    body = {
        "client": {"clientId": "your-custom-client-id", "clientVersion": "1.0"},
        "threatInfo": {
            "threatTypes": ["MALWARE", "SOCIAL_ENGINEERING", "UNWANTED_SOFTWARE"],
            "platformTypes": ["WINDOWS", "LINUX", "ANDROID"],
            "threatEntryTypes": ["URL"],
            "threatEntries": [{"url": url}]
        }
    }
    # 调用威胁检测接口
    response = client.threat_matches_find(body=body)
    
    # 根据检测结果分类存储URL
    if not response.matches:
        safe_urls.append(url)
    else:
        malicious_urls.append(url)

# 格式化输出分类结果
print("# Safe URLs")
for url in safe_urls:
    print(f"- {url}")

print("\n# Malicious URLs")
for url in malicious_urls:
    print(f"- {url}")

关键调整说明： #

1. 结果预存储：新增safe_urls和malicious_urls两个列表，替代原来的实时打印，这样能统一收集所有检测结果后再整理输出。
2. Markdown格式化输出：最后用#标记一级标题，用-生成列表项，让安全/恶意URL各自归类，输出结构一目了然。
3. 逻辑复用性：核心的URL检测逻辑完全保留，只是调整了结果的处理方式，后续要扩展功能（比如导出到文件）也更方便。

这样运行脚本后，你就能得到整洁的分类输出啦！

内容的提问来源于stack exchange，提问作者D.Philip