lock tables decorator

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
def require_lock(*tables):
    def _lock(func):
        def _do_lock(*args,**kws):
            #lock tables
            cursor = connection.cursor()
            cursor.execute("LOCK TABLES %s WRITE" %' '.join(tables))
            try:
                result = func(*args,**kws)
            except Exception,e:
                #unlock tables
                cursor.execute("UNLOCK TABLES")
                if cursor:cursor.close()
                raise Exception(e)
            else:
                #unlock tables
                cursor.execute("UNLOCK TABLES")
                if cursor:cursor.close()
                return result
        return _do_lock
    return _lock

'''
    example:

    @require_lock(table_A,table_B)
    fuc(args):
        pass
'''

Comments

svetlyak (on September 4, 2008):

I think, that it would be better and shoter, to use try…finally block, like that: 

        try:
            return func(*args,**kws)
        finally:
            #unlock tables
            cursor.execute("UNLOCK TABLES")
            cursor.close()

#

(Forgotten your password?)

You may use Markdown syntax here, but raw HTML will be removed.