From cb41c54135e8aa487a728de943ba243e9bea9043 Mon Sep 17 00:00:00 2001
From: Terrtia <or1994@hotmail.fr>
Date: Wed, 21 Nov 2018 08:49:58 +0100
Subject: [PATCH] fix: [lib_refine] add dns configuration

---
 bin/packages/config.cfg.sample |  3 +++
 bin/packages/lib_refine.py     | 13 ++++++++++++-
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/bin/packages/config.cfg.sample b/bin/packages/config.cfg.sample
index 6efb4be4..b5980766 100644
--- a/bin/packages/config.cfg.sample
+++ b/bin/packages/config.cfg.sample
@@ -206,6 +206,9 @@ dns = 8.8.8.8
 [Mail]
 dns = 8.8.8.8
 
+[Web]
+dns = 149.13.33.69
+
 # Indexer configuration
 [Indexer]
 type = whoosh
diff --git a/bin/packages/lib_refine.py b/bin/packages/lib_refine.py
index 5d2af0a9..32f56900 100644
--- a/bin/packages/lib_refine.py
+++ b/bin/packages/lib_refine.py
@@ -1,6 +1,8 @@
 #!/usr/bin/python3
 
 import re
+import os
+import configparser
 import dns.resolver
 
 from pubsublogger import publisher
@@ -101,11 +103,20 @@ def checking_MX_record(r_serv, adress_set, addr_dns):
 
 
 def checking_A_record(r_serv, domains_set):
+    configfile = os.path.join(os.environ['AIL_BIN'], 'packages/config.cfg')
+    if not os.path.exists(configfile):
+        raise Exception('Unable to find the configuration file. \
+                        Did you set environment variables? \
+                        Or activate the virtualenv.')
+    cfg = configparser.ConfigParser()
+    cfg.read(configfile)
+    dns_server = cfg.get("Web", "dns")
+
     score = 0
     num = len(domains_set)
     WalidA = set([])
     resolver = dns.resolver.Resolver()
-    resolver.nameservers = ['149.13.33.69']
+    resolver.nameservers = [dns_server]
     resolver.timeout = 5
     resolver.lifetime = 2