import pymysql

#获得数据库的链接
def getdb():
    conn_db = pymysql.connect(host="127.0.0.1", port=3306, user="root", passwd="4399", db="py_us_pre_ele", charset="utf8")
    return conn_db

#得到数据库中的所有的城市
def get_all_city(db):
    set={""}#创建一个set集合，用来放置所有的城市
    str_sql="select state from tab"#查找到所有的城市语句
    cur=db.cursor()#得到cursor对象
    cur.execute(str_sql)#执行sql语句
    rows=cur.fetchall()#得到结果集
    for row in rows:#将城市加入到不重复的set集合中
        set.add(row)
    set.remove("")
    with open("all_cities.txt", "w") as f:
        for ele in set:
            f.write(str(ele)+"\n")

def main():
    get_all_city(getdb())

if __name__ == '__main__':
    main()
    print("程序完成")