Example #1
0
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()())
Example #2
0
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()())