From 2280b33c431b36f099163b009e0aa9a359390dc3 Mon Sep 17 00:00:00 2001
From: Paul Lecuq <paul@paulbsd.com>
Date: Mon, 25 Jan 2016 14:07:49 +0100
Subject: [PATCH] add py-squid-blacklists.py

---
 py-squid-blacklists.py | 63 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100755 py-squid-blacklists.py

diff --git a/py-squid-blacklists.py b/py-squid-blacklists.py
new file mode 100755
index 0000000..ca57b9a
--- /dev/null
+++ b/py-squid-blacklists.py
@@ -0,0 +1,63 @@
+#!/usr/bin/python
+
+import sys
+import os
+import re
+import logging
+import time
+from urlparse import urlparse
+
+blacklists_dir = "/home/paul/blacklists/"
+domain_files = [os.path.join(dp, f) for dp, dn, fn in os.walk(os.path.expanduser(blacklists_dir)) for f in fn if re.match(r"domains*", f)]
+
+def make_list(files):
+	blacklists=[]
+	for f in files:
+		splitlist = f.split("/")
+		list_type = splitlist[len(splitlist)-2]
+		blacklists.append([list_type,f])
+	return blacklists
+
+def make_db(blacklist_files):
+	lib = dict()
+	for blacklist in blacklist_files:
+		cache= dict()
+		values=[]
+		f = open(blacklist[1], "r")
+		for line in f:
+			cache[line.strip("\n")] = True
+		lib[blacklist[0]]=cache
+	return lib
+
+def compare(outline,blacklist_cache):
+	result = False
+	while not result and outline != "":
+		try:
+			result=blacklist_cache['adult'][outline]
+		except KeyError:
+			pass
+		outline = outline.partition('.')[2]
+	return result
+
+def grant():
+	sys.stdout.write( 'OK\n' )
+	sys.stdout.flush()
+
+def deny():
+	sys.stdout.write( 'ERR\n' )
+	sys.stdout.flush()
+
+
+blacklist_cache=[]
+
+blacklist_files = make_list(domain_files)
+blacklist_cache = make_db(blacklist_files)
+
+while True:
+	line = sys.stdin.readline().strip()
+	outline = urlparse(line).netloc
+	if line:
+		if compare(outline,blacklist_cache):
+			grant()
+		else:
+			deny()