Python操作MySQL数据库方法、方式总结

import pandas as pd

import pymysql

import sqlalchemy

from sqlalchemy import create_engine

'''

通过 sqlalchemy 构建数据库链接engine获取数据

'''

def get_datas_by_sqlalchemy(db_user,db_pass,db_host,db_port,db_name,sql_cmd):

connect_msg = 'mysql+pymysql://{}:{}@{}:{}/{}?charset=utf8'.format(db_user,db_pass,db_host,db_port,db_name)

engine = create_engine(connect_msg)

return pd.read_sql(sql=sql_cmd, con=engine) # 返回类型 pandas.core.frame.DataFrame

'''

通过 dbaapi构建数据库连接引擎

'''

def get_datas_by_dbaapi(db_user,db_pass,db_host,db_port,db_name,sql_cmd):

connect = pymysql.connect(host=db_host,user=db_user,password=db_pass,database=db_name,charset='utf8',use_unicode=True)

return pd.read_sql(sql_cmd, connect) # 返回类型 pandas.core.frame.DataFrame

'''

将数据持久化到 MySQL 数据库中

'''

def persisst_datas_to_mysql(db_user,db_pass,db_host,db_port,db_name,tab_name,df):

connect_msg = 'mysql+pymysql://{}:{}@{}:{}/{}?charset=utf8'.format(db_user,db_pass,db_host,db_port,db_name)

engine = create_engine(connect_msg)

df.to_sql(name=tab_name,

con=engine,

if_exists='append',

index=False)

if __name__ == '__main__':

DB_USER,DB_PASS,DB_HOST,DB_PORT,DB_NAME='root','1234','localhost','3306','test'

SQL_CMD="select * from wo_user"

# 操作 1 查询方式一

al_df=get_datas_by_sqlalchemy(DB_USER,DB_PASS,DB_HOST,DB_PORT,DB_NAME,SQL_CMD)

print(al_df)

print ('------------------------')

# 操作 2 数据持久化

tab_name='wo_user'

df=pd.DataFrame([[6,'xiaoming','123'],[7,'xiaobai','456'],[8,'xiaohei','789']],columns=['id','userName','passWord'])

persisst_datas_to_mysql(DB_USER,DB_PASS,DB_HOST,DB_PORT,DB_NAME,tab_name,df)

# 操作 3 查询方式二

dba_df=get_datas_by_dbaapi(DB_USER,DB_PASS,DB_HOST,DB_PORT,DB_NAME,SQL_CMD)

print (dba_df)

print ('------------------------')