Use link.sh; Add header analysis; #477

drivers/python/wappalyzer.py

@@ -18,7 +18,7 @@ class Wappalyzer(object):
     def __init__(self, url):
         self.file_dir = os.path.dirname(__file__)
 
-        f = open(os.path.join(self.file_dir, '../../share/apps.json'))
+        f = open(os.path.join(self.file_dir, 'apps.json'))
         data = json.loads(f.read())
         f.close()
 
@@ -30,7 +30,7 @@ class Wappalyzer(object):
         ctxt = PyV8.JSContext()
         ctxt.enter()
 
-        f1 = open(os.path.join(self.file_dir, '../../share/js/wappalyzer.js'))
+        f1 = open(os.path.join(self.file_dir, 'js/wappalyzer.js'))
         f2 = open(os.path.join(self.file_dir, '../php/js/driver.js'))
         ctxt.eval(f1.read())
         ctxt.eval(f2.read())
@@ -38,9 +38,11 @@ class Wappalyzer(object):
         f2.close()
 
         host = urlparse(self.url).hostname
-        html = requests.get(self.url).text
+        response = requests.get(self.url)
+        html = response.text
+        headers = dict(response.headers)
 
-        data = {'host': host, 'url': self.url, 'html': html, 'headers': {}}
+        data = {'host': host, 'url': self.url, 'html': html, 'headers': headers}
         apps = json.dumps(self.apps)
         categories = json.dumps(self.categories)
         return ctxt.eval("w.apps = %s; w.categories = %s; w.driver.data = %s; w.driver.init();" % (apps, categories, json.dumps(data)))

links.sh

@@ -22,3 +22,9 @@ ln -f share/js/wappalyzer.js      drivers/html/js
 
 ln -f share/apps.json             drivers/php
 ln -f share/js/wappalyzer.js      drivers/php/js
+
+ln -f share/apps.json             drivers/python
+if [ ! -d "drivers/python/js" ];then
+  mkdir drivers/python/js/
+fi
+ln -f share/js/wappalyzer.js      drivers/python/js