python - 链接mysql操作mysql代码实例

import pymysql
from pymysql.cursors import DictCursor
from loguru import logger


# 数据库配置选择
DATABASE_CONFIG = {
    "test": {
        'user': 'user_name',
        'passwd': 'xxxxxx',
        'host': '127.0.0.1',
        'database': 'test',
        'port': 3306
    },
    "product": {
        'user': 'user_name',
        'passwd': 'xxxxxx',
        'host': '127.0.0.1',
        'database': 'product',
        'port': 3306
    }
}


def inti_mysql(db_conf):
    """
    初始化数据库链接
    """
    conn = pymysql.connect(host=db_conf['host'],
                           user=db_conf['user'],
                           passwd=db_conf['passwd'],
                           db=db_conf['database'],
                           port=int(db_conf['port']),
                           cursorclass=DictCursor)

    cursor = conn.cursor()
    return conn, cursor


def get_data(db, cursor):
    """
    查询
    """
    sql = "select id, name, from users;"
    cursor.execute(sql)
    ret = cursor.fetchall()
    return ret


if __name__ == "__main__":
    pass
    # 链接mysql
    data_base_flag = "test"
    db_conf = DATABASE_CONFIG[data_base_flag]
    db, cursor = inti_mysql(db_conf)

    # 操作mysql
    res = get_data(db, cursor)
    logger.info(f"res = {res}")