python連接mysql數據表查詢表獲取數據導入到txt中

import pymysql
'''連接mysql數據表查詢表獲取數據導入到txt中'''
#查詢結果寫入數據到txt
def get_loan_number(file_txt):
    connect = pymysql.Connect(
        host="100.100.100.0",
        port=3306,
        user="test",
        passwd="123",
        db="bigdata",
        charset='utf8'
    )
    print("寫入中，請等待……")
    cursor = connect.cursor()
    sql = "select * from ts_vehicle_load"
    cursor.execute(sql)
    number = cursor.fetchall()
    fp = open(file_txt, "w")
    loan_count = 0
    for loanNumber in number:
        loan_count += 1
        fp.write(str(loanNumber) + "\n")
    fp.close()
    cursor.close()
    connect.close()
    print("寫入完成,共寫入%d條數據！" % loan_count)


if __name__ == "__main__":
    file_txt = r"E:\Users\admin\PycharmProjects\untitled\d"
    get_loan_number(file_txt)