python连接MySQL数据库查询并打印出表的primarykey

cxg0508

于 2020-04-11 14:16:05 发布

阅读量1.3k

点赞数 1

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/SecondRound93/article/details/105451832

本文介绍了一种使用Python脚本快速查找MySQL数据库中指定表的主键的方法。通过pymysql和sqlalchemy库连接数据库，执行SQL语句获取表结构，并利用正则表达式解析主键信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import pymysql
from sqlalchemy import create_engine
import pandas as pd
import re

def engine_mysql():
    connect_mysql = create_engine('mysql+pymysql://username:password@127.0.0.1:3306/dbname?charset=utf8')
    return connect_mysql
    
def main(table):
    mysql_engine=engine_mysql()
    sql = 'show create table {};'.format(table)
    df = pd.read_sql_query(sql,mysql_engine)
    df.columns = [i.lower() for i in df.columns]
    mysql_engine.dispose()

    crt =df['create table'].iloc[-1]
    pat = re.compile(r"(?=\bPRIMARY KEY\b)(.*)(?<=,)")
    result = pat.findall(crt)
    return result

if __name__=='__main__':
    table ='table_name'
    print(main(table))