From 6038d3287d932dc5c30d33d801d3c81fa6fedc8d Mon Sep 17 00:00:00 2001
From: acidvegas <acid.vegas@acid.vegas>
Date: Tue, 25 Jul 2023 20:37:32 -0400
Subject: [PATCH] Updated

---
 avoidr/avoidr.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/avoidr/avoidr.py b/avoidr/avoidr.py
index 631d924..e608153 100644
--- a/avoidr/avoidr.py
+++ b/avoidr/avoidr.py
@@ -13,13 +13,15 @@ grand_total = {'4': 0, '6': 0}
 results     = dict()
 
 def calculate_hash(path):
+	''' Calculate the SHA1 hash of a file. '''
 	hash_sha1 = hashlib.sha1()
 	with open(path, 'rb') as f:
 		for chunk in iter(lambda: f.read(4096), b''):
 			hash_sha1.update(chunk)
 	return hash_sha1.hexdigest()
 
-def get_url(url, git=False):
+def get_url(url, git=False) -> str:
+	''' Get the contents of a URL. '''
 	data = {'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_9_3) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/35.0.1916.47 Safari/537.36'}
 	if git:
 		data['Accept'] = 'application/vnd.github.v3+json'
@@ -27,6 +29,7 @@ def get_url(url, git=False):
 	return urllib.request.urlopen(req, timeout=10).read().decode()
 
 def update_database():
+	''' Update the ASN database.  '''
 	DB = 'databases/fullASN.json.zip'
 	try:
 		os.mkdir('databases')
@@ -52,6 +55,7 @@ def update_database():
 			zObject.extract(DB[10:-4], 'databases')
 
 def process_asn(data):
+	''' Process an ASN. '''
 	if data['asn'] not in results:
 		title = data['descr'] if 'org' not in data else data['descr'] + ' / ' + data['org']
 		results[data['asn']] = {'name': title, 'ranges': dict()}