python操作数据库demo

# coding:utf-8
import pymysql

def query_sql(sql):
    # 打开数据库连接
    db = pymysql.connect("localhost", "root", "root", "test")
    # 使用 cursor() 方法创建一个游标对象 cursor
    cursor = db.cursor()
    try:
        # 执行sql语句
        cursor.execute(sql)
        # 提交到数据库执行
        db.commit()
    except:
        # 如果发生错误则回滚
        db.rollback()
    # 关闭数据库连接
    db.close()

def main():
    sql="insert into test (name,number) values ('test1',12)"
    query_sql(sql)

if __name__ == '__main__':
    main()