Merge branch 'avinashshenoy97/patch-1'

site_list.py

@@ -1,14 +1,17 @@
 """Sherlock: Supported Site Listing
-
 This module generates the listing of supported sites.
 """
 import json
 import sys
 import requests
-from argparse import ArgumentParser, RawDescriptionHelpFormatter
+import threading
 from bs4 import BeautifulSoup as bs
+from datetime import datetime
+from argparse import ArgumentParser, RawDescriptionHelpFormatter
+
+pool = list()
 
-def get_rank(domain_to_query):
+def get_rank(domain_to_query, dest):
 	result = -1
 	url = "http://www.alexa.com/siteinfo/" + domain_to_query
 	page = requests.get(url).text
@@ -20,7 +23,7 @@ def get_rank(domain_to_query):
 					if strong.has_attr("class"):
 						if "metrics-data" in strong["class"]:
 							result = int(strong.text.strip().replace(',', ''))
-    return result
+							dest['rank'] = result
 
 parser = ArgumentParser(formatter_class=RawDescriptionHelpFormatter
                         )
@@ -37,20 +40,32 @@ with open("sites.md", "w") as site_file:
 	data_length = len(data)
 	site_file.write(f'## List Of Supported Sites ({data_length} Sites In Total!)\n')
 
-	index = 1
 	for social_network in data:
 		url_main = data.get(social_network).get("urlMain")
+		data.get(social_network)["rank"] = 0
+		if args.rank:
+			th = threading.Thread(target=get_rank, args=(url_main, data.get(social_network)))
+		else:
+			th = None
+		pool.append((url_main, url_main, th))
+		if args.rank:
+			th.start()
+
+	index = 1
+	for social_network, url_main, th in pool:
+		if args.rank:
+			th.join()
 		site_file.write(f'{index}. [{social_network}]({url_main})\n')
-		if args.rank == True:
-			data.get(social_network)["rank"] = get_rank(url_main)
 		sys.stdout.write("\r{0}".format(f"Updated {index} out of {data_length} entries"))
 		sys.stdout.flush()
 		index = index + 1
 
+	if args.rank:
+		site_file.write(f'\nAlexa.com rank data fetched at ({datetime.utcnow()} UTC)\n')
+
 sorted_json_data = json.dumps(data, indent=2, sort_keys=True)
 
 with open("data.json", "w") as data_file:
 	data_file.write(sorted_json_data)
 
-sys.stdout.write("\r{0}".format(f"Finished updating supported site listing!\n"))
-sys.stdout.flush()
+print("\nFinished updating supported site listing!")