[大数据]python 如何实现数据库的连接

python 如何实现数据库的连接

记录一次phthon连接数据库的实例

一、源码

import pymysql
import re
def connectdb():
    print('连接到mysql服务器...')
    # 打开数据库连接
    # 用户名:hp, 密码:Hp12345.,用户名和密码需要改成你自己的mysql用户名和密码，并且要创建数据库TESTDB，并在TESTDB数据库中创建好表Student
    db = pymysql.connect(host="****", user="*******", password="******r", database="*****")

    print('连接上了!')
    return db
# 关闭数据连接
def closedb(db):
    db.close()

def getNumber(db,table):
    # 使用cursor()方法获取操作游标
    cursor = db.cursor()
    sql = "SELECT count(*) FROM "+table
    try:
        # 执行SQL语句
        cursor.execute(sql)
        # 获取所有记录数量
        num1 = cursor.fetchall()
        for row in num1:
            print(table+"表数据量:",row[0])
    except:
        print("获取失败！")

    return  row[0]

def getfjdcar_idtofoundvin(db):
    # 使用cursor()方法获取操作游标
    cursor = db.cursor()
    sql = "SELECT jdcar_id FROM car_mt_wiper limit 0,10000 "


    try:
        # 执行SQL语句
        cursor.execute(sql)
        # 获取所有记录列表
        results = cursor.fetchall()
        list=[]
        for row in results:
           jdcar_id  = row[0]
           list.append(foundvin(db,jdcar_id))
    except:
        print("数据获取失败！")

    return list
# 得到vin
def foundvin(db,str):
    # 使用cursor()方法获取操作游标
    cursor = db.cursor()
    sql = "SELECT vin FROM jd_car_vin where jdcar_id="+str
    try:
        # 执行SQL语句
        cursor.execute(sql)
        # 获取所有记录列表
        results = cursor.fetchall()
        for row in results:
           return row[0]

    except:
        print("vin获取失败!")



def main():
    db = connectdb()    # 连接MySQL数据库
    print(getNumber(db,"jd_car_vin"))
    for i in getfjdcar_idtofoundvin(db):
        print(i)
    closedb(db)         # 关闭数据库

if __name__ == '__main__':
    main()

总结