Spaces:

ernoban
/

cs

Paused

cs / app.py

Create app.py

5044684 verified about 1 year ago

4.24 kB

import gradio as gr
import socket
import time

def check_dns_resolution(domain):
    """
    尝试将域名解析为IP地址。
    这是测试DNS问题的最直接方法。
    返回一个包含结果的元组 (domain, status, detail)。
    """
    try:
        # socket.gethostbyname 是对系统DNS解析器的直接调用。
        # 如果此调用失败，则说明从Space内部无法解析该域名。
        ip_address = socket.gethostbyname(domain)
        return (domain, "✅ 可访问", f"成功解析到IP: {ip_address}")
    except socket.gaierror as e:
        # socket.gaierror 是专门处理地址相关错误的异常（例如，主机名未知）。
        return (domain, "❌ 无法访问", f"DNS解析失败: {e}")
    except Exception as e:
        return (domain, "❌ 出错", f"发生未知错误: {e}")

def batch_test_domains(domain_list_str, progress=gr.Progress(track_tqdm=True)):
    """
    对给定的域名列表进行批量测试，并生成Markdown格式的报告。
    """
    if not domain_list_str or not domain_list_str.strip():
        return "⚠️ 请在文本框中输入至少一个域名。"

    # 将输入字符串分割成域名列表，并清除空白行和多余的空格
    domains_to_test = [d.strip() for d in domain_list_str.split('\n') if d.strip()]
    if not domains_to_test:
        return "⚠️ 输入无效，请输入有效域名。"

    total_domains = len(domains_to_test)
    results = []

    # 使用串行循环代替并发，并加入延迟，以避免触发速率限制
    for i, domain in enumerate(domains_to_test):
        # 更新进度条，这能保持与前端的连接，防止超时
        progress(i / total_domains, desc=f"正在测试 ({i+1}/{total_domains}): {domain}")
        
        # 执行DNS解析
        result = check_dns_resolution(domain)
        results.append(result)
        
        # !!! 重要: 加入短暂延迟，防止请求过于密集而被服务器限制 !!!
        time.sleep(0.25)

    # --- 开始生成报告 ---
    accessible_domains = []
    inaccessible_domains = []

    for domain, status, detail in results:
        # 使用 Markdown 的代码格式 `domain` 来突出显示域名
        formatted_line = f"`{domain}` - {detail}"
        if "可访问" in status:
            accessible_domains.append(formatted_line)
        else:
            inaccessible_domains.append(formatted_line)
    
    report = f"## 批量测试报告 (共测试 {total_domains} 个域名)\n\n"
    
    report += f"### ✅ 可访问的域名 ({len(accessible_domains)} 个)\n"
    if accessible_domains:
        report += "\n".join([f"- {d}" for d in accessible_domains])
    else:
        report += "无"
        
    report += f"\n\n### ❌ 无法访问或出错的域名 ({len(inaccessible_domains)} 个)\n"
    if inaccessible_domains:
        report += "\n".join([f"- {d}" for d in inaccessible_domains])
    else:
        report += "无"
        
    return report

# --- Gradio 应用界面 ---
with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="blue")) as demo:
    gr.Markdown(
        """
        # Hugging Face Space 出站网络可访问性测试 (优化版)
        在下面的文本框中输入或粘贴您需要测试的域名列表（每行一个）。
        工具将以安全、稳定的方式逐个测试，并生成清晰的报告，以避免IP被临时限制。
        """
    )

    with gr.Row():
        domain_input = gr.Textbox(
            lines=15, 
            label="域名列表 (每行一个)", 
            placeholder="www.google.com\nwww.facebook.com\nhuggingface.co\napi.openai.com\n..."
        )
    
    submit_btn = gr.Button("🚀 开始批量测试", variant="primary")
    
    output_report = gr.Markdown(label="测试报告")
    
    # 将按钮点击事件连接到主函数
    submit_btn.click(
        fn=batch_test_domains,
        inputs=domain_input,
        outputs=output_report
    )
    
    gr.Examples(
        [
            "www.google.com\ngithub.com\nhuggingface.co\nwww.wikipedia.org",
            "www.facebook.com\nwww.instagram.com\nwww.youtube.com\nweb.whatsapp.com"
        ],
        inputs=domain_input
    )

demo.launch()

	import gradio as gr
	import socket
	import time

	def check_dns_resolution(domain):
	"""
	尝试将域名解析为IP地址。
	这是测试DNS问题的最直接方法。
	返回一个包含结果的元组 (domain, status, detail)。
	"""
	try:
	# socket.gethostbyname 是对系统DNS解析器的直接调用。
	# 如果此调用失败，则说明从Space内部无法解析该域名。
	ip_address = socket.gethostbyname(domain)
	return (domain, "✅ 可访问", f"成功解析到IP: {ip_address}")
	except socket.gaierror as e:
	# socket.gaierror 是专门处理地址相关错误的异常（例如，主机名未知）。
	return (domain, "❌ 无法访问", f"DNS解析失败: {e}")
	except Exception as e:
	return (domain, "❌ 出错", f"发生未知错误: {e}")

	def batch_test_domains(domain_list_str, progress=gr.Progress(track_tqdm=True)):
	"""
	对给定的域名列表进行批量测试，并生成Markdown格式的报告。
	"""
	if not domain_list_str or not domain_list_str.strip():
	return "⚠️ 请在文本框中输入至少一个域名。"

	# 将输入字符串分割成域名列表，并清除空白行和多余的空格
	domains_to_test = [d.strip() for d in domain_list_str.split('\n') if d.strip()]
	if not domains_to_test:
	return "⚠️ 输入无效，请输入有效域名。"

	total_domains = len(domains_to_test)
	results = []

	# 使用串行循环代替并发，并加入延迟，以避免触发速率限制
	for i, domain in enumerate(domains_to_test):
	# 更新进度条，这能保持与前端的连接，防止超时
	progress(i / total_domains, desc=f"正在测试 ({i+1}/{total_domains}): {domain}")

	# 执行DNS解析
	result = check_dns_resolution(domain)
	results.append(result)

	# !!! 重要: 加入短暂延迟，防止请求过于密集而被服务器限制 !!!
	time.sleep(0.25)

	# --- 开始生成报告 ---
	accessible_domains = []
	inaccessible_domains = []

	for domain, status, detail in results:
	# 使用 Markdown 的代码格式 `domain` 来突出显示域名
	formatted_line = f"`{domain}` - {detail}"
	if "可访问" in status:
	accessible_domains.append(formatted_line)
	else:
	inaccessible_domains.append(formatted_line)

	report = f"## 批量测试报告 (共测试 {total_domains} 个域名)\n\n"

	report += f"### ✅ 可访问的域名 ({len(accessible_domains)} 个)\n"
	if accessible_domains:
	report += "\n".join([f"- {d}" for d in accessible_domains])
	else:
	report += "无"

	report += f"\n\n### ❌ 无法访问或出错的域名 ({len(inaccessible_domains)} 个)\n"
	if inaccessible_domains:
	report += "\n".join([f"- {d}" for d in inaccessible_domains])
	else:
	report += "无"

	return report

	# --- Gradio 应用界面 ---
	with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="blue")) as demo:
	gr.Markdown(
	"""
	# Hugging Face Space 出站网络可访问性测试 (优化版)
	在下面的文本框中输入或粘贴您需要测试的域名列表（每行一个）。
	工具将以安全、稳定的方式逐个测试，并生成清晰的报告，以避免IP被临时限制。
	"""
	)

	with gr.Row():
	domain_input = gr.Textbox(
	lines=15,
	label="域名列表 (每行一个)",
	placeholder="www.google.com\nwww.facebook.com\nhuggingface.co\napi.openai.com\n..."
	)

	submit_btn = gr.Button("🚀 开始批量测试", variant="primary")

	output_report = gr.Markdown(label="测试报告")

	# 将按钮点击事件连接到主函数
	submit_btn.click(
	fn=batch_test_domains,
	inputs=domain_input,
	outputs=output_report
	)

	gr.Examples(
	[
	"www.google.com\ngithub.com\nhuggingface.co\nwww.wikipedia.org",
	"www.facebook.com\nwww.instagram.com\nwww.youtube.com\nweb.whatsapp.com"
	],
	inputs=domain_input
	)

	demo.launch()