20 lines
573 B
Python
20 lines
573 B
Python
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 |