完成大作业报告

Project/src/Service/func/utils.py

@@ -0,0 +1,20 @@
+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