eris

- Elasticsearch Recon Ingestion Scripts (ERIS) 🔎
git clone git://git.acid.vegas/-c.git
Log | Files | Refs | Archive | README | LICENSE

commit 7f93a4d8de7da1c603a1ee077c4edd5e6780dac0
parent 4dc31a50900b91c74465d6bb3b172c2ab91426ed
Author: acidvegas <acid.vegas@acid.vegas>
Date: Wed, 13 Mar 2024 20:54:36 -0400

add ingest_rir_transfers.py

Diffstat:
Mingestors/ingest_rir_transfers.py | 20+++++++-------------

1 file changed, 7 insertions(+), 13 deletions(-)

diff --git a/ingestors/ingest_rir_transfers.py b/ingestors/ingest_rir_transfers.py
@@ -4,7 +4,6 @@
 
 import json
 import ipaddress
-import logging
 import time
 
 try:
@@ -77,19 +76,17 @@ async def process_data():
 			async with aiohttp.ClientSession(headers=headers) as session:
 				async with session.get(url) as response:
 					if response.status != 200:
-						logging.error(f'Failed to fetch {registry} delegation data: {response.status}')
-						raise Exception('errror') #continue
+						raise Exception(f'Failed to fetch {registry} delegation data: {response.status}')
   
 					data = await response.text()
 
 					try:
 						json_data = json.loads(data)
 					except aiohttp.ContentTypeError as e:
-						logging.error(f'Failed to parse {registry} delegation data: {e}')
-						raise Exception('errror') #continue
+						raise Exception(f'Failed to parse {registry} delegation data: {e}')
   
 					if 'transfers' not in json_data:
-						logging.error(f'Invalid {registry} delegation data: {json_data}')
+						raise Exception(f'Invalid {registry} delegation data: {json_data}')
       
 					for record in json_data['transfers']:
          
@@ -102,8 +99,7 @@ async def process_data():
 											asn = set_block[option]
 											if type(asn) != int:
 												if not asn.isdigit():
-													logging.error(f'Invalid {set_type} {option} ASN in {registry} data: {asn}')
-													raise Exception('errror') #continue
+													raise Exception(f'Invalid {set_type} {option} ASN in {registry} data: {asn}')
 												else:
 													record['asns'][set_type][count][option] = int(asn)
 										count += 1
@@ -126,18 +122,16 @@ async def process_data():
 															ipaddress.ip_address(normalized_ip)
 															record[ip_type][set_type][count][option] = normalized_ip
 														except ValueError as e:
-															logging.error(f'Invalid {set_type} {option} IP in {registry} data: {e}')
-															raise Exception('errror') #continue
+															raise Exception(f'Invalid {set_type} {option} IP in {registry} data: {e}')
 												count += 1
         
 						if record['type'] not in ('MERGER_ACQUISITION', 'RESOURCE_TRANSFER'):
-							logging.error(f'Invalid transfer type in {registry} data: {record["type"]}')
-							raise Exception('errror') #continue
+							raise Exception(f'Invalid transfer type in {registry} data: {record["type"]}')
         
 						yield {'_index': default_index, '_source': record}
 
 		except Exception as e:
-			logging.error(f'Error processing {registry} delegation data: {e}')
+			raise Exception(f'Error processing {registry} delegation data: {e}')
 
 
 async def test():