from decorator import decorator
import sqlalchemy
class Data:
    def __init__(self):
        self.pool =  None

data = Data()
#a valid dbDefer decorator

def setPool(pool): 
    data.pool = pool

@decorator
def runInDBPool(f,*args,**kwargs):
    """Decorator to run DB queries in Twisted's thread pool"""
    if isinstance(args[-1],sqlalchemy.orm.session.Session):
        return f(*args,**kwargs)
    else:
        return data.pool.run(f, *args, **kwargs)