audit_names.py

import xml.etree.cElementTree as ET
from collections import defaultdict
import re
import pprint

OSMFILE = "mumbai_india.osm"
street_type_re = re.compile(r'\b\S+\.?$', re.IGNORECASE)

#expected = ["Slums", "Avenue", "Boulevard", "Drive", "Court", "Place", "Square", "Lane", "Road", 
            
# UPDATE THIS VARIABLE
mapping = { "Jhopadpatti": "Slums",
            "Marg,": "Marg",
            "Mumbai,":"Mumbai",
            "ROad":"Road",
            "Rd":"Road",
            "Raod":"Road",
            "Rd.":"Road",
            "road":"Road"
            }




# UPDATE THIS VARIABLE


def audit_street_type(street_types, street_name):
    m = street_type_re.search(street_name)
    if m:
        street_type = m.group()
        #if street_type not in expected:
        street_types[street_type].add(street_name)


def is_street_name(elem):
    return (elem.attrib['k'] == "addr:street")


def audit(osmfile):
    osm_file = open(osmfile, "r")
    street_types = defaultdict(set)
    for event, elem in ET.iterparse(osm_file, events=("start",)):

        if elem.tag == "node" or elem.tag == "way":
            for tag in elem.iter("tag"):
                if is_street_name(tag):
                    audit_street_type(street_types, tag.attrib['v'])
    osm_file.close()
    return street_types

def update_name(name, mapping):
    #print '\nSTART'
    #print name
    m = street_type_re.search(name)
    if m.group() in mapping.keys():        
        name=re.sub(street_type_re,mapping[m.group()],name)

    # YOUR CODE HERE

    return name

st_types = audit(OSMFILE)
#pprint.pprint(dict(st_types))

for st_type, ways in st_types.iteritems():
    for name in ways:
        better_name = update_name(name, mapping)
        print name, "=>", better_name