repos/pypilot
4
0
Fork 0
Code Issues Pull Requests Actions Packages Releases 11 Activity

Added checks to ensure API responses are as expected (#7)

Browse Source 
* Added checks to ensure API responses are as expected

* Add tests for broken IP/prefix API responses

* Add tests for bad IP/prefix status codes
This commit is contained in:
Luke Tainton
2022-06-26 00:15:27 +01:00
committed by GitHub
parent 03b85b3bef
commit af8fcad53a
9 changed files with 75 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
app/main.py
View File

@ -38,6 +38,9 @@ def main():
# Get information from API
ip_info = get_ip_information(ip_address)
if not ip_info:
print("ERROR: could not retrieve IP information from API.")
sys.exit(1)
as_number = get_autonomous_system_number(ip_info.get("as"))
# Assemble list for table generation
@ -56,7 +59,11 @@ def main():
# If wanted, get prefix information
if args.prefixes:
prefix_info = get_prefix_information(as_number)
table_data.append(["Prefixes", generate_prefix_string(prefix_info)])
if not prefix_info:
print("ERROR: could not retrieve prefix information from API.")
sys.exit(1)
else:
table_data.append(["Prefixes", generate_prefix_string(prefix_info)])
print_table(table_data)