vikarti.anatra/itg-allow-domains
1
0
Fork 0
mirror of https://github.com/itdoginfo/allow-domains.git synced 2025-09-02 02:29:46 +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

33
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}") 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' output_lists_directory = 'geosite_data'
os.makedirs(output_lists_directory, exist_ok=True) 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 dirs:
for directory in domains_or_dirs: if not os.path.isdir(directory):
continue
for filename in os.listdir(directory): for filename in os.listdir(directory):
file_path = os.path.join(directory, filename) file_path = os.path.join(directory, filename)
if not os.path.isfile(file_path):
continue
if os.path.isfile(file_path): attribute = os.path.splitext(filename)[0].replace('_', '-')
with open(file_path, 'r', encoding='utf-8') as file:
attribute = os.path.splitext(filename)[0] with open(file_path, 'r', encoding='utf-8') as f:
extracted_domains.extend(f"{line.strip()} @{attribute}" for line in file if line.strip()) for line in f:
else: domain = line.strip()
extracted_domains = domains_or_dirs 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) output_file_path = os.path.join(output_lists_directory, output_name)
with open(output_file_path, 'w', encoding='utf-8') as file: with open(output_file_path, 'w', encoding='utf-8') as out_f:
file.writelines(f"{name}\n" for name in extracted_domains) 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'): def generate_dat_domains(data_path='geosite_data', output_name='geosite.dat', output_directory='DAT'):
os.makedirs(output_directory, exist_ok=True) os.makedirs(output_directory, exist_ok=True)