def patch_all(): from solar.dblayer.model import ModelMeta if ModelMeta._defined_models: raise RuntimeError( "You should run patch_multi_get before defining models") from solar.dblayer.model import Model from solar.dblayer.gevent_helpers import get_local from solar.dblayer.gevent_helpers import multi_get from solar.dblayer.gevent_helpers import solar_map from solar import utils _patch(Model, 'multi_get', multi_get) _patch(utils, 'solar_map', solar_map) _patch(utils, 'get_local', get_local) _patch(Model, '_local', get_local()())
def patch_all(): from solar.config import C from solar.dblayer.model import ModelMeta if ModelMeta._defined_models: raise RuntimeError( "You should run patch_multi_get before defining models") from solar.dblayer.model import Model from solar.dblayer.gevent_helpers import get_current_ident from solar.dblayer.gevent_helpers import get_local from solar.dblayer.gevent_helpers import multi_get from solar.dblayer.gevent_helpers import solar_map from solar import utils if C.solar_db.startswith('riak'): # patching these methods on sql # dbs does not make sense _patch(Model, 'multi_get', multi_get) _patch(utils, 'solar_map', solar_map) _patch(utils, 'get_local', get_local) _patch(utils, 'get_current_ident', get_current_ident) _patch(Model, '_local', get_local()())