域名屏蔽代码

import os

def block_domain(domain):
    # 假设我们使用的是BIND DNS服务器
    dns_config_file = "/etc/bind/named.conf.local"
    
    # 检查文件是否存在
    if not os.path.exists(dns_config_file):
        print(f"配置文件 {dns_config_file} 不存在")
        return
    
    # 读取配置文件内容
    with open(dns_config_file, 'r') as file:
        config_content = file.read()
    
    # 检查是否已经存在屏蔽规则
    if f"zone \"{domain}\" {{ type master; file \"/etc/bind/db.empty\"; }};" in config_content:
        print(f"{domain} 已经被屏蔽")
        return
    
    # 添加屏蔽规则
    new_rule = f"""
zone \"{domain}\" {{
    type master;
    file \"/etc/bind/db.empty\";
}};
"""
    config_content += new_rule
    
    # 写回配置文件
    with open(dns_config_file, 'w') as file:
        file.write(config_content)
    
    print(f"{domain} 已经被成功屏蔽")

# 示例调用
block_domain("example.com")