python基础——使用pymysql连接数据库

使用pymysql连接数据库

"""
pymysql基本用法：
"""
import pymysql


def main():
    # 创建连接
    with pymysql.connect(host="127.0.0.1", port=3306,
                         user='root', password='root',
                         db='bms', charset='utf8') as conn:
        # 报错就回滚数据库
        try:
            # 获取一个游标对象，游标对象可以执行sql语句
            with conn.cursor() as cursor:
                result = cursor.execute(
                    'insert into python_test values(1, "贵永康", 25)')
                if result == 1:
                    print('添加成功')
                else:
                    raise Exception('添加失败')
                # 因为python是自带事物的，不提交的话就不能添加
                conn.commit()
        except Exception as error:
            print(error)
            conn.rollback()


if __name__ == '__main__':
    main()