def gen_state_name(filename):
"""
http://www.50states.com/tools/postal.htm
"""
state_names = set()
state_codes = set()
with open(filename) as f:
for line in f:
words = line.strip().split('\t')
try:
state_names.add(words[0])
state_codes.add(words[1])
except:
pass
state_names = cf.process_text(state_names)
state_codes = cf.process_text(state_codes)
return state_names, state_codes
def gen_airport_cityname(filename):
"""
http://openflights.org/data.html
"""
cities = set()
airports = set()
airport_codes = set()
with open(filename) as f:
reader = csv.reader(f)
for row in reader:
try:
# if row[3] == 'United States':
cities.add(row[2])
airports.add(row[1])
airport_codes.add(row[4])
except:
pass
cities = cf.process_text(cities)
airports = cf.process_text(airports)
airport_codes = cf.process_text(airport_codes)
return cities, airports, airport_codes
