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()