python数据库实现

Question

我试图在python中实现一个简单的数据库程序.我到了我已经向db添加元素,更改值等的地步.

class db:
    def __init__(self):
            self.database ={}

    def dbset(self, name, value):
            self.database[name]=value

    def dbunset(self, name):
            self.dbset(name, 'NULL')

    def dbnumequalto(self, value):
            mylist = [v for k,v in self.database.items() if v==value]
            return mylist

def main():
    mydb=db()
    cmd=raw_input().rstrip().split(" ")
    while cmd[0]!='end':
            if cmd[0]=='set':
                    mydb.dbset(cmd[1], cmd[2])
            elif cmd[0]=='unset':
                    mydb.dbunset(cmd[1])
            elif cmd[0]=='numequalto':
                    print len(mydb.dbnumequalto(cmd[1]))
            elif cmd[0]=='list':
                    print mydb.database
            cmd=raw_input().rstrip().split(" ")

if __name__=='__main__':
    main()

现在,作为下一步,我希望能够在此python代码中执行嵌套事务.我使用BEGIN命令开始一组命令,然后使用COMMIT语句提交它们.提交应该提交所有开始的事务.但是,回滚应将更改还原为最近的BEGIN.我无法为此提出合适的解决方案.

Answer 1

一个简单的方法是保留一个"事务"列表,其中包含您能够回滚挂起的更改所需的所有信息:

def dbset(self, name, value):
    self.transaction.append((name, self.database.get(name)))
    self.database[name]=value

def rollback(self):
    # undo all changes
    while self.transaction:
        name, old_value = self.transaction.pop()
        self.database[name] = old_value

def commit(self):
    # everything went fine, drop undo information
    self.transaction = []