mirror of
https://github.com/KaringX/karing-ruleset.git
synced 2025-06-08 13:34:12 +08:00
104 lines
3.0 KiB
Python
Executable File
104 lines
3.0 KiB
Python
Executable File
# list convert to json
|
|
import sys
|
|
import os
|
|
import json
|
|
|
|
MAP_RULES_KEY_DICT = {
|
|
'IP-CIDR': 'ip_cidr',
|
|
'IP-CIDR6': 'ip_cidr',
|
|
'DOMAIN': 'domain',
|
|
'DOMAIN-SUFFIX': 'domain_suffix',
|
|
'DOMAIN-KEYWORD': 'domain_keyword',
|
|
'PROCESS-NAME': 'process_name',
|
|
}
|
|
|
|
|
|
def deug_log(msg: str):
|
|
print(msg)
|
|
pass
|
|
|
|
|
|
def main(src_path: str, out_path: str = None):
|
|
out_path = (
|
|
os.getcwd()
|
|
if out_path is None or len(out_path) == 0
|
|
else os.path.abspath(out_path)
|
|
)
|
|
src_path = os.path.abspath(src_path)
|
|
deug_log(f"{src_path} to {out_path}")
|
|
listdir_format(src_path, out_path=out_path)
|
|
|
|
|
|
def listdir_format(src_path: str, out_path: str):
|
|
for item in os.listdir(src_path):
|
|
if os.path.isdir(item):
|
|
out_path2 = os.path.join(out_path, item)
|
|
if os.path.exists(out_path2) is False:
|
|
os.mkdir(out_path2)
|
|
deug_log(f"mkdir {out_path2}")
|
|
listdir_format(os.path.join(src_path, item), out_path=out_path2)
|
|
else:
|
|
converto_json(os.path.join(src_path, item), out_path=out_path)
|
|
|
|
|
|
def converto_json(src_file: str, out_path: str):
|
|
splits = os.path.splitext(src_file)
|
|
if splits[1] != ".list":
|
|
return
|
|
if not os.path.isfile(src_file):
|
|
deug_log(f"ERROR FILE: {src_file}")
|
|
return
|
|
|
|
out_file = os.path.join(out_path, os.path.basename(splits[0]) + ".json")
|
|
content = {"version": 1, "rules": [{}]}
|
|
|
|
with open(src_file, "r") as f:
|
|
line = f.readline()
|
|
while line:
|
|
# line.strip()
|
|
in_key = False
|
|
row = line.split(",")
|
|
if len(row) >= 2:
|
|
rule, cont = row[0].upper(), row[1].strip()
|
|
if 'le.com' == cont:
|
|
deug_log(f"row:{row} drop le.com/match google.com")
|
|
rule = 'PASS'
|
|
|
|
if rule in MAP_RULES_KEY_DICT:
|
|
rkey = MAP_RULES_KEY_DICT[rule]
|
|
if rkey in content['rules'][0]:
|
|
content['rules'][0][rkey].append(cont)
|
|
else:
|
|
content['rules'][0][rkey] = [cont]
|
|
|
|
in_key = True
|
|
# END if
|
|
|
|
#! DEBUG
|
|
if in_key is False:
|
|
if row[0] not in ["USER-AGENT", "URL-REGEX"] and line[0] not in [
|
|
'#',
|
|
'\n',
|
|
]:
|
|
deug_log(f"row:{row} first:{line[0]} UNKOWN")
|
|
line = f.readline()
|
|
# END while
|
|
|
|
if len(content['rules'][0]) == 0:
|
|
return
|
|
|
|
json_string = json.dumps(content)
|
|
with open(out_file, "w") as json_file:
|
|
json_file.write(json_string)
|
|
deug_log(f"wirte to {out_file}")
|
|
# END converto_josn
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) < 3:
|
|
print("Usage: python script.py <first_parameter> <second_parameter>")
|
|
else:
|
|
src_path = sys.argv[1]
|
|
out_path = sys.argv[2]
|
|
main(src_path, out_path)
|