pypilot/app/main.py

73 lines
2.1 KiB
Python
Raw Normal View History

2022-05-02 20:08:27 +01:00
#!/usr/local/bin/python3
2022-06-25 22:27:01 +01:00
"""MODULE: Main application module."""
import sys
from app.args import parse_args
from app.print_table import print_table, generate_prefix_string
from app.query_normalisation import is_ip_address, resolve_domain_name
from app.ip_info import (
2022-06-25 22:27:01 +01:00
get_ip_information,
get_autonomous_system_number,
get_prefix_information,
)
HEADER = """-----------------------------------------------
| IP Address Information Lookup Tool (iPilot) |
| By Luke Tainton (@luketainton) |
-----------------------------------------------\n"""
2022-05-02 20:08:27 +01:00
def main():
2022-06-25 22:27:01 +01:00
"""Main function."""
args = parse_args()
if not args.noheader:
print(HEADER)
# Set IP to passed IP address, or resolve passed domain name to IPv4
ip_address = (
resolve_domain_name(args.query) if not is_ip_address(args.query) else args.query
)
# If not given an IPv4, and can't resolve to IPv4, then throw error and exit
if not ip_address:
print("ERROR: could not resolve query to IPv4 address.")
sys.exit(1)
# 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)
2022-06-25 22:27:01 +01:00
as_number = get_autonomous_system_number(ip_info.get("as"))
# Assemble list for table generation
country = ip_info.get("country")
region = ip_info.get("regionName")
city = ip_info.get("city")
2022-06-25 22:27:01 +01:00
table_data = [
["IP Address", ip_info.get("query")],
["Organization", ip_info.get("org")],
["Location", f"{country}/{region}/{city}"],
2022-06-25 22:27:01 +01:00
["Timezone", ip_info.get("timezone")],
["Internet Service Provider", ip_info.get("isp")],
["Autonomous System", as_number],
]
# If wanted, get prefix information
if args.prefixes:
prefix_info = get_prefix_information(as_number)
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)])
2022-06-25 22:27:01 +01:00
print_table(table_data)
2022-05-02 20:08:27 +01:00
if __name__ == "__main__":
main()