peewee 事物 回滚

peewee 事物 回滚

#!/usr/bin/env python
# coding=utf-8
from peewee import *

db = MySQLDatabase(host='123.57.229.66', user='root', passwd='password', database='czj', charset='utf8')

class tb_users(Model):
    id = PrimaryKeyField()  # 主键
    mobile = CharField(unique=True, max_length=64, null=False)  # 注册手机号/openid
    password = CharField(max_length=32)  # 密码
    truename = CharField(max_length=32)  # 真实姓名/nickname

    class Meta:
        database = db

# 一
def one():
    try:
        with db.atomic():
            tb_users.create(mobile='', password='', truename='')
            raise 'haha'
        print 'Success'
    except IntegrityError:
        print 'Failure: %s is already in use.' % username

# 二
@db.atomic()
def create_user():
    tb_users.create(mobile='', password='', truename='')
    raise 'haha'

def two():
    create_user()

# 三
def three():
    db.set_autocommit(False)
    db.begin()
    try:
        tb_users.create(mobile='', password='', truename='')
        raise 'haha'
    except:
        db.rollback()
        raise
    else:
        try:
            db.commit()
        except:
            db.rollback()
            raise
    finally:
        db.set_autocommit(True)

# 四
def four():
    db.set_autocommit(False)
    db.begin()
    tb_users.create(mobile='', password='', truename='')
    db.rollback()
    db.commit()
    db.set_autocommit(True)

实例1,2是使用 peewee封装的db.atomic()原子性，3,4是利用关闭自动提交和手动回滚来保证事物的原子性；