43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
import gzip
|
|
import ipaddress
|
|
import re
|
|
|
|
import requests
|
|
|
|
|
|
def process_blacklist(url, output_file_name):
|
|
response = requests.get(url)
|
|
lines = response.text.splitlines()
|
|
|
|
processed_lines = []
|
|
|
|
for line in lines:
|
|
line = line.strip()
|
|
if not line or line.startswith("#"):
|
|
continue
|
|
# 检查是否是合法的IP地址或CIDR
|
|
if re.match(r"(\d{1,3}\.){3}\d{1,3}(\/\d{1,2})?$", line) or re.match(
|
|
r"([a-fA-F0-9:]+)(\/\d{1,3})?$", line
|
|
):
|
|
if "/" in line:
|
|
network = ipaddress.ip_network(line, strict=False)
|
|
ip_range = f"{network.network_address}-{network.broadcast_address}"
|
|
else:
|
|
ip_range = f"{line}-{line}"
|
|
processed_lines.append(f"btn:{ip_range}")
|
|
|
|
with open(f"{output_file_name}.txt", "w", encoding="utf-8") as file:
|
|
for line in processed_lines:
|
|
file.write(line + "\n")
|
|
|
|
with gzip.open(f"{output_file_name}.gz", "wt", encoding="utf-8") as file:
|
|
for line in processed_lines:
|
|
file.write(line + "\n")
|
|
|
|
|
|
# 调用函数处理远程文件并压缩为.gz格式
|
|
process_blacklist(
|
|
"https://raw.githubusercontent.com/PBH-BTN/BTN-Collected-Rules/main/combine/all.txt",
|
|
"transmission_blacklist",
|
|
)
|