mirror of
https://github.com/KaringX/karing-ruleset.git
synced 2025-06-09 04:34:12 +08:00
99 lines
2.9 KiB
Python
99 lines
2.9 KiB
Python
import os, sys
|
||
import requests
|
||
import re
|
||
|
||
|
||
def debug_log(msg: str):
|
||
if msg.startswith('ERR'):
|
||
msg = msg.replace('ERR', '\x1b[31mERR\x1b[0m')
|
||
print(msg)
|
||
# pass
|
||
|
||
|
||
def get_out_path(out_path: str = None):
|
||
return (
|
||
os.getcwd()
|
||
if out_path is None or len(out_path) == 0
|
||
else os.path.abspath(out_path)
|
||
)
|
||
|
||
|
||
def writeto_rulefile(out_file: str, content: dict | str):
|
||
file_string = json.dumps(content) if isinstance(content, dict) else content
|
||
|
||
with open(out_file, "w") as json_file:
|
||
json_file.write(file_string)
|
||
debug_log(f"wirte to {out_file}")
|
||
|
||
|
||
def get_url_content(url: str):
|
||
try:
|
||
response = requests.get(url)
|
||
response.raise_for_status() # Raises an HTTPError for bad responses (4xx and 5xx)
|
||
return response.text if response.status_code == 200 else None
|
||
|
||
except requests.exceptions.HTTPError as http_err:
|
||
debug_log(f"HTTP error occurred: {http_err} \tURL: {url}")
|
||
except requests.exceptions.ConnectionError as conn_err:
|
||
debug_log(f"Connection error occurred: {conn_err} \tURL: {url}")
|
||
except requests.exceptions.Timeout as timeout_err:
|
||
debug_log(f"Timeout error occurred: {timeout_err} \tURL: {url}")
|
||
except requests.exceptions.RequestException as req_err:
|
||
debug_log(f"An error occurred: {req_err} \tURL: {url}")
|
||
return None
|
||
|
||
|
||
def correct_name(text: str) -> str:
|
||
# Define the replacements as a dictionary
|
||
replacements = {
|
||
" ": "-",
|
||
"–": "-",
|
||
",": "-",
|
||
"/": "_",
|
||
":": "",
|
||
"'": "",
|
||
"(": "_",
|
||
")": "",
|
||
}
|
||
|
||
# Apply the replacements
|
||
replaced_text = ''.join(replacements.get(c, c) for c in text)
|
||
|
||
# Collapse consecutive "-" or "_" into a single "_"
|
||
collapsed_text = re.sub(r'-+', '-', replaced_text)
|
||
collapsed_text = re.sub(r'_+', '_', collapsed_text)
|
||
|
||
# Remove trailing "-" or "_"
|
||
final_text = re.sub(r'[-_]+$', '', collapsed_text)
|
||
|
||
# Replace "-_" and "_-" with "_"
|
||
final_text = re.sub(r'-_|_-', '_', final_text)
|
||
|
||
return final_text
|
||
|
||
|
||
def remove_ansi_escape_codes(text: str) -> str:
|
||
# Regular expression pattern for ANSI escape codes
|
||
ansi_escape = re.compile(r'\x1b\[[0-9;]*m')
|
||
return ansi_escape.sub('', text)
|
||
|
||
|
||
def is_url(string):
|
||
pattern = re.compile(
|
||
r'^(https?://)?' # http:// 或 https://（可选）
|
||
r'([a-zA-Z0-9_-]+\.)+[a-zA-Z]{2,6}' # 域名部分
|
||
r'(:[0-9]{1,5})?' # 端口号（可选）
|
||
r'(/.*)?$' # 路径部分（可选）
|
||
)
|
||
return bool(pattern.match(string))
|
||
|
||
|
||
if __name__ == '__main__':
|
||
# text = "example string//with spaces---and////slashes"
|
||
text = 'uBlock-filters-–-Resource-abuse-'
|
||
text = 'Dandelion-Sprouts-Anti-Malware-List-_for-AdGuard-Home-and-for-AdGuard-for-Android_Windows-DNS-filtering'
|
||
print(correct_name(text))
|
||
|
||
# text = '\x1b[36mINFO\x1b[0m[0000] parsed rules: 778/1450'
|
||
# print(remove_ansi_escape_codes(text).startswith('INFO'))
|