stop_cloudflare/tool/anti-cf_hosts_generator/generate.py

48 lines
1.3 KiB
Python

import glob
import getopt
import os
import sys
print("Anti-CF hosts file generator v1.0")
print("Licensed under CC0 1.0")
print("")
#functions
def progress(value, filename):
sys.stdout.write('\033[2K\033[1G') #wipe the line before printing
sys.stdout.write("Compiling rules [%s] %s" % ("{:,}".format(value), filename))
sys.stdout.flush()
opts, argv = getopt.getopt(sys.argv[1:], "o:d:")
def process(outputf, inputd):
if not os.path.isdir(inputd):
print("Input not directory");
sys.exit(1)
with open(outputf, 'a') as outfile:
rulescom = 0 #counter for counting how much rules compiled
files = glob.glob(os.path.join(inputd, "*.txt")) #filter out anything but txt files
for f in files:
with open(f) as ruleso:
for line in ruleso:
outfile.write("0.0.0.0 {}\n".format(line.rstrip())) #this does the job
rulescom = rulescom + 1
progress(rulescom, f)
if len(sys.argv) < 5: #check if theres argv
print("Usage: generate.py -o <output file> -d <input directory>")
print("Example: generate.py -o filter.txt -d ../../cloudflare_users/domains/")
sys.exit(1)
#argv parsing
for k, v in opts:
if k == '-o':
outputfile = v
if k == '-d':
inputdirectory = v
process(outputfile, inputdirectory)