vikarti.anatra/itg-allow-domains
1
0
Fork 0
mirror of https://github.com/itdoginfo/allow-domains.git synced 2025-09-01 18:19:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Fix

Browse source
This commit is contained in:
unidcml 2025-06-06 17:38:50 +03:00 • committed by GitHub
parent 7bb3ea4da3
commit 3c8b49c2bf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 22 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
convert.py
View file

@ -362,28 +362,35 @@ def generate_srs_combined(input_subnets_file, input_domains_file, output_json_di
print(f"Compile error {output_file_path}: {e}")
def prepare_dat_domains(domains_or_dirs, output_name):
def prepare_dat_domains(domains, output_name, dirs=[]):
output_lists_directory = 'geosite_data'
os.makedirs(output_lists_directory, exist_ok=True)
extracted_domains = []
domain_attrs = {domain: [] for domain in domains}
if all(os.path.isdir(d) for d in domains_or_dirs):
for directory in domains_or_dirs:
for filename in os.listdir(directory):
file_path = os.path.join(directory, filename)
for directory in dirs:
if not os.path.isdir(directory):
continue
for filename in os.listdir(directory):
file_path = os.path.join(directory, filename)
if not os.path.isfile(file_path):
continue
if os.path.isfile(file_path):
with open(file_path, 'r', encoding='utf-8') as file:
attribute = os.path.splitext(filename)[0]
extracted_domains.extend(f"{line.strip()} @{attribute}" for line in file if line.strip())
else:
extracted_domains = domains_or_dirs
attribute = os.path.splitext(filename)[0].replace('_', '-')
with open(file_path, 'r', encoding='utf-8') as f:
for line in f:
domain = line.strip()
if not domain:
continue
if domain in domain_attrs:
domain_attrs[domain].append(f" @{attribute}")
output_file_path = os.path.join(output_lists_directory, output_name)
with open(output_file_path, 'w', encoding='utf-8') as file:
file.writelines(f"{name}\n" for name in extracted_domains)
with open(output_file_path, 'w', encoding='utf-8') as out_f:
for domain, attrs in domain_attrs.items():
line = domain + "".join(attrs)
out_f.write(f"{line}\n")
def generate_dat_domains(data_path='geosite_data', output_name='geosite.dat', output_directory='DAT'):
os.makedirs(output_directory, exist_ok=True)