#!/usr/bin/env python3

import sys
import csv

"""
Strip unneeded data from world cities database.

https://simplemaps.com/data/world-cities
"""

us_cities = []
other_cities = []

with open(sys.argv[1]) as infile:
  reader = csv.reader(infile, quotechar='"')
  next(reader) # skip header
  for row in reader:
    pt = (row[2], row[3])
    if row[4] == "United States":
      us_cities.append(pt)
    else:
      other_cities.append(pt)

with open(sys.argv[2], "w") as outfile:
  for (lat, lng) in us_cities:
    outfile.write(f"{lat}, {lng}\n")

with open(sys.argv[3], "w") as outfile:
  for (lat, lng) in other_cities:
    outfile.write(f"{lat}, {lng}\n")