46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
import geonamescache
|
|
from geopy.distance import geodesic
|
|
|
|
gc = geonamescache.GeonamesCache()
|
|
cities = gc.get_cities()
|
|
us_cities = {k: c for k, c in cities.items() if c.get("countrycode") == "US"}
|
|
|
|
print(gc.search_cities("Jamaica"), "\n")
|
|
print(gc.search_cities("Manhattan"), "\n")
|
|
print("lengths:", len(cities), len(us_cities))
|
|
|
|
|
|
def get_coordinates(city_name, country_code="US"):
|
|
search_results = gc.search_cities(city_name, case_sensitive=True)
|
|
for city in search_results:
|
|
print(f"searching {city}")
|
|
possible_matches = city.get("alternatenames") + [city_name]
|
|
if city_name in possible_matches and city.get("countrycode") == country_code:
|
|
return city.get("latitude"), city.get("longitude")
|
|
return None
|
|
|
|
|
|
def get_distance(city1, city2, country1="US", country2="US"):
|
|
city1_coords = get_coordinates(city1, country1)
|
|
city2_coords = get_coordinates(city2, country2)
|
|
|
|
if city1_coords is None or city2_coords is None:
|
|
return None
|
|
|
|
return geodesic(city1_coords, city2_coords).km
|
|
|
|
|
|
MAX_DISTANCE = 20_037.5
|
|
|
|
city1 = "New York"
|
|
city2 = "Jamaica"
|
|
country1 = "US"
|
|
country2 = "US"
|
|
|
|
distance = get_distance(city1, city2, country1, country2)
|
|
|
|
if distance is not None:
|
|
print(f"Distance between {city1} and {city2} is {distance:.2f} km.")
|
|
else:
|
|
print("One or both city names were not found.")
|