from .config import db
import pymysql
from xpinyin import Pinyin
from pymysql.cursors import Cursor

def get_cities():
    conn = pymysql.connect(**db)
    cursor = conn.cursor(pymysql.cursors.DictCursor)
    cursor.execute("SELECT DISTINCT City FROM Airports")
    cities = [row['City'] for row in cursor.fetchall()]
    cursor.close()
    conn.close()
    p = Pinyin()
    cities = [
        (row, p.get_pinyin(row).replace("-", ""))
        for row in cities
    ]
    cities = sorted(cities, key=lambda x: x[1])
    cities = [row[0] for row in cities]
    return cities