blob: 037873359591b8ad34469a3a5cea4153188ba5af (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
#!/usr/bin/env nix-shell
#!nix-shell -i python3 --pure -p 'python3.withPackages (p: [ p.requests ])'
import argparse
import requests
import socket
import sys
API = "http://ip-api.com/json"
def valid_ip(address: str) -> bool:
"""Check if the given address is a valid IPv4 address."""
try:
socket.inet_pton(socket.AF_INET, address)
except AttributeError: # No inet_pton.
try:
socket.inet_aton(address)
except socket.error:
return False
return address.count(".") == 3
except socket.error: # Not a valid address.
return False
return True
def main():
parser = argparse.ArgumentParser(
prog="geoip",
description="Geolocate an IPv4 address.",
)
parser.add_argument(dest="address", help="IPv4 address to locate")
args = parser.parse_args()
if not valid_ip(args.address):
print(f"'{args.address}' is not a valid IPv4 address", file=sys.stderr)
exit(1)
res = requests.get(f"{API}/{args.address}")
if res.status_code != 200:
print(
f"Something went wrong: returned {res.status_code}",
file=sys.stderr,
)
data = res.json()
if data["status"] == "fail":
print(f"Querying '{data['query']}' failed", file=sys.stderr)
exit(1)
# Not every address will have a zip code, so adjust the format accordingly.
loc = (
f"{data['query']} is located in {data['city']}, {data['regionName']}"
+ (f", {data['zip']} " if data["zip"] else " ")
+ data["country"]
)
print(loc)
if __name__ == "__main__":
main()
# vim:ft=python
|
