用python查询数据库中的数据并保存至文本(pymysql)

逃离我推掉我的手 2023-07-03 02:55 6阅读 0赞

#coding=utf-8
    
    import pymysql as MySQLdb  #这里是python3 如果你是python2.x的话，import MySQLdb
    
    host = xxxx
    user = xxxx
    passwd = xxxx
    port = xxxx
    db = xxxx
    class SelectMySQL(object):
        def select_data(self,sql):
            result = []
            try:
                conn = MySQLdb.connect(host=host,
                                   port=port,
                                   user=user,
                                   passwd=passwd,
                                   db=db,
                                   charset='utf8', )
                cur = conn.cursor()
                cur.execute(sql)
                alldata = cur.fetchall()
                # print(alldata)
                for rec in alldata:
                    result.append(rec[0]) #注意，我这里只是把查询出来的第一列数据保存到结果中了,如果是多列的话，稍微修改下就ok了
            except Exception as e:
                print('Error msg: ' + e)
            finally:
                cur.close()
                conn.close()
      
            return result
    
        def get_result(self, sql, filename):
            print(sql)
            results = self.select_data(sql)
            print('The amount of datas: %d' % (len(results)))
            with open(filename, 'w') as f:
                for result in results:
                    f.write(str(result) + '\n')
            print('Data write is over!')
            return results
    
    if __name__ == '__main__':
        sql = "select name from user"
        select = SelectMySQL()
        result1 = select.get_result(sql,'namemsg.txt')
        print(result1)