Python script to get longitude and latitude from google maps API or transform from UTM (provided in file)

geolocorutm.py

# -*- coding: UTF-8 -*-
import csv
import os
import sys
import json
import time
import pyproj
import argparse
from urllib import parse, request
if sys.version_info[0] < 3: raise Exception("Python 3 required")

googleapi = '<YOUR_GOOGLEMAPS_APIKEY_HERE>'
_projections = {}

def zone(coordinates):
    # https://gist.github.com/twpayne/4409500
    if 56 <= coordinates[1] < 64 and 3 <= coordinates[0] < 12:
        return 32
    if 72 <= coordinates[1] < 84 and 0 <= coordinates[0] < 42:
        if coordinates[0] < 9:
            return 31
        elif coordinates[0] < 21:
            return 33
        elif coordinates[0] < 33:
            return 35
        return 37
    return int((coordinates[0] + 180) / 6) + 1


def letter(coordinates):
    # https://gist.github.com/twpayne/4409500
    return 'CDEFGHJKLMNPQRSTUVWXX'[int((coordinates[1] + 80) / 8)]


def project(coordinates):
    # https://gist.github.com/twpayne/4409500
    z = zone(coordinates)
    l = letter(coordinates)
    if z not in _projections:
        _projections[z] = pyproj.Proj(proj='utm', zone=z, ellps='WGS84')
    x, y = _projections[z](coordinates[0], coordinates[1])
    if y < 0:
        y += 10000000
    return z, l, x, y


def unproject(z, l, x, y):
    # https://gist.github.com/twpayne/4409500
    if z not in _projections:
        _projections[z] = pyproj.Proj(proj='utm', zone=z, ellps='WGS84')
    if l < 'N':
        y -= 10000000
    lng, lat = _projections[z](x, y, inverse=True)
    return (lng, lat)



def get_coordinates_from_address(address, key):
    debug_msg = False
    googleGeocodeUrl = 'https://maps.googleapis.com/maps/api/geocode/json?'

    address = address.encode('utf-8')
    params = {
        'address': address,
        'key': key,
    }
    url = googleGeocodeUrl + parse.urlencode(params)
    json_response = request.urlopen(url)
    response = json.loads(json_response.read().decode('utf-8'))

    if response['status'] == 'OK':
        location = response['results'][0]['geometry']['location']
        latitude, longitude = location['lat'], location['lng']
        if debug_msg: print (address, latitude, longitude)
    else:
        latitude, longitude = None, None
        if debug_msg: print (address, "<no results>")
        if debug_msg: print (response)

    return [response['status'], longitude, latitude]



def clean_white_spaces(strtoclean):
    while strtoclean.startswith(' '):
        strtoclean = strtoclean[1:]

    while strtoclean.endswith(' '):
        strtoclean = strtoclean[:-1]

    return strtoclean


def transform_file(inputfile, outputfile):
    with open(inputfile, 'r', encoding='utf-8') as csvfile:
        reader = csv.reader(csvfile, delimiter=",")
        next(reader)
        with open(outputfile, 'w') as outfile:
            writer = csv.writer(outfile)
            for row in reader:
                address = f'{clean_white_spaces(row[2])}, {clean_white_spaces(row[4])}, ESPAÑA'
                while True:
                    time.sleep(0.5)
                    status, lng, lat = get_coordinates_from_address(address, googleapi)
                    if status == 'OK':
                        row.append(lng)
                        row.append(lat)
                        writer.writerow(row)
                        print(status, lng, lat, address)
                        break
                    if status == 'ZERO_RESULTS':
                        lng, lat = unproject(30, 's', row[6], row[5])
                        row.append(lng)
                        row.append(lat)
                        writer.writerow(row)
                        print(status, lng, lat, address)
                        break


if __name__=="__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("-f", help="CSV file to parse")
    args = parser.parse_args()

    if not args.f:
        print('-f argument required')
        print('Example: python geolocorutm.py -f educapu.csv')
        sys.exit()
    if not args.f.endswith('.csv'):
        print('-f argument must end with ".csv"')
        print('Example: python geolocorutm.py -f educapu.csv')
        sys.exit()

    inputfile = os.path.normpath(args.f)
    if not os.path.exists(inputfile):
        print('File not found')
        sys.exit()

    outputfile = os.path.normpath(args.f[:-4]+'-output.csv')

    transform_file(inputfile, outputfile)
    

testfile.csv

Usage: python geolocorutm.py -f path/to/file.csv

• path/to/file.csv is the input file.
• output file will be path/to/file-output.csv