blob: 4a5ca42e9edfc76a840cdf0de46f89968175a2d9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
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)
|
