import argparse

import json

import os

import random

import sys

from urllib.request import urlopen, urlretrieve

import shapefile # pip install pyshp

# Optional, http://stackoverflow.com/a/1557906/724176

try:

import timing

assert timing # avoid flake8 warning

except ImportError:

pass

# Google Street View Image API

# 25,000 image requests per 24 hours

# See https://developers.google.com/maps/documentation/streetview/

API_KEY = "INSERT_YOUR_API_KEY_HERE"

GOOGLE_STREETVIEW_URL = (

"http://maps.googleapis.com/maps/api/streetview?sensor=false&"

"size=640x640&key=" + API_KEY

)

GOOGLE_METADATA_URL = (

"https://maps.googleapis.com/maps/api/streetview/metadata?key=" + API_KEY

)

IMG_PREFIX = "img_"

IMG_SUFFIX = ".jpg"

parser = argparse.ArgumentParser(

description="Get random Street View images from a given country",

formatter_class=argparse.ArgumentDefaultsHelpFormatter,

)

parser.add_argument(

"-n", "--images-wanted", type=int, default=10, help="Number of images wanted"

)

parser.add_argument("country", help="ISO 3166-1 Alpha-3 Country Code")

parser.add_argument(

"-hdg",

"--heading",

help="Heading in degrees: 0 and 360 north, 90 east, 180 south, 270 west",

)

parser.add_argument(

"-p",

"--pitch",

help="Pitch in degrees: 0 is default, 90 straight up, -90 straight down",

)

args = parser.parse_args()

# Determine if a point is inside a given polygon or not

# Polygon is a list of (x,y) pairs.

# http://www.ariel.com.au/a/python-point-int-poly.html

def point_inside_polygon(x, y, poly):

return inside

print("Loading borders")

shape_file = "TM_WORLD_BORDERS-0.3.shp"

if not os.path.exists(shape_file):

sys.exit(

f"Cannot find {shape_file}. Please download it from "

"http://thematicmapping.org/downloads/world_borders.php and try again."

)

sf = shapefile.Reader(shape_file, encoding="latin1")

shapes = sf.shapes()

print("Finding country")

for i, record in enumerate(sf.records()):

if record[2] == args.country.upper():

print(record[2], record[4])

print(shapes[i].bbox)

min_lon = shapes[i].bbox[0]

min_lat = shapes[i].bbox[1]

max_lon = shapes[i].bbox[2]

max_lat = shapes[i].bbox[3]

borders = shapes[i].points

break

print("Getting images")

attempts, country_hits, imagery_hits, imagery_misses = 0, 0, 0, 0

MAX_URLS = 25000

if not os.path.exists(args.country):

os.makedirs(args.country)

try:

while True:

attempts += 1

rand_lat = random.uniform(min_lat, max_lat)

rand_lon = random.uniform(min_lon, max_lon)

# Is (lat,lon) inside borders?

if point_inside_polygon(rand_lon, rand_lat, borders):

print(" In country")

country_hits += 1

lat_lon = str(rand_lat) + "," + str(rand_lon)

outfile = os.path.join(args.country, IMG_PREFIX + lat_lon + IMG_SUFFIX)

streetview_url = GOOGLE_STREETVIEW_URL + "&location=" + lat_lon

metadata_url = GOOGLE_METADATA_URL + "&location=" + lat_lon

with urlopen(metadata_url) as response:

body = response.read()

metadata_status = json.loads(body)["status"]

if metadata_status == "ZERO_RESULTS" or metadata_status == "NOT_FOUND":

imagery_misses += 1

print(f" Image not found: {metadata_status}")

continue

elif metadata_status != "OK":

raise RuntimeError(

f"Unexpected response from metadata query: {metadata_status}"

)

if args.heading:

streetview_url += "&heading=" + args.heading

if args.pitch:

streetview_url += "&pitch=" + args.pitch

try:

urlretrieve(streetview_url, outfile)

except KeyboardInterrupt:

sys.exit("exit")

if os.path.isfile(outfile):

print(lat_lon)

print(" ========== Got one! ==========")

imagery_hits += 1

if imagery_hits == args.images_wanted:

break

if country_hits == MAX_URLS:

break

except KeyboardInterrupt:

print("Keyboard interrupt")

print("Attempts:\t", attempts)

print("Country hits:\t", country_hits)

print("Imagery misses:\t", imagery_misses)

print("Imagery hits:\t", imagery_hits)

# End of file