189 lines
5.8 KiB
Python
189 lines
5.8 KiB
Python
|
# event/api.py
|
||
|
# Copyright (C) 2005-2018 the SQLAlchemy authors and contributors
|
||
|
# <see AUTHORS file>
|
||
|
#
|
||
|
# This module is part of SQLAlchemy and is released under
|
||
|
# the MIT License: http://www.opensource.org/licenses/mit-license.php
|
||
|
|
||
|
"""Public API functions for the event system.
|
||
|
|
||
|
"""
|
||
|
from __future__ import absolute_import
|
||
|
|
||
|
from .. import util, exc
|
||
|
from .base import _registrars
|
||
|
from .registry import _EventKey
|
||
|
|
||
|
CANCEL = util.symbol('CANCEL')
|
||
|
NO_RETVAL = util.symbol('NO_RETVAL')
|
||
|
|
||
|
|
||
|
def _event_key(target, identifier, fn):
|
||
|
for evt_cls in _registrars[identifier]:
|
||
|
tgt = evt_cls._accept_with(target)
|
||
|
if tgt is not None:
|
||
|
return _EventKey(target, identifier, fn, tgt)
|
||
|
else:
|
||
|
raise exc.InvalidRequestError("No such event '%s' for target '%s'" %
|
||
|
(identifier, target))
|
||
|
|
||
|
|
||
|
def listen(target, identifier, fn, *args, **kw):
|
||
|
"""Register a listener function for the given target.
|
||
|
|
||
|
e.g.::
|
||
|
|
||
|
from sqlalchemy import event
|
||
|
from sqlalchemy.schema import UniqueConstraint
|
||
|
|
||
|
def unique_constraint_name(const, table):
|
||
|
const.name = "uq_%s_%s" % (
|
||
|
table.name,
|
||
|
list(const.columns)[0].name
|
||
|
)
|
||
|
event.listen(
|
||
|
UniqueConstraint,
|
||
|
"after_parent_attach",
|
||
|
unique_constraint_name)
|
||
|
|
||
|
|
||
|
A given function can also be invoked for only the first invocation
|
||
|
of the event using the ``once`` argument::
|
||
|
|
||
|
def on_config():
|
||
|
do_config()
|
||
|
|
||
|
event.listen(Mapper, "before_configure", on_config, once=True)
|
||
|
|
||
|
.. versionadded:: 0.9.4 Added ``once=True`` to :func:`.event.listen`
|
||
|
and :func:`.event.listens_for`.
|
||
|
|
||
|
.. note::
|
||
|
|
||
|
The :func:`.listen` function cannot be called at the same time
|
||
|
that the target event is being run. This has implications
|
||
|
for thread safety, and also means an event cannot be added
|
||
|
from inside the listener function for itself. The list of
|
||
|
events to be run are present inside of a mutable collection
|
||
|
that can't be changed during iteration.
|
||
|
|
||
|
Event registration and removal is not intended to be a "high
|
||
|
velocity" operation; it is a configurational operation. For
|
||
|
systems that need to quickly associate and deassociate with
|
||
|
events at high scale, use a mutable structure that is handled
|
||
|
from inside of a single listener.
|
||
|
|
||
|
.. versionchanged:: 1.0.0 - a ``collections.deque()`` object is now
|
||
|
used as the container for the list of events, which explicitly
|
||
|
disallows collection mutation while the collection is being
|
||
|
iterated.
|
||
|
|
||
|
.. seealso::
|
||
|
|
||
|
:func:`.listens_for`
|
||
|
|
||
|
:func:`.remove`
|
||
|
|
||
|
"""
|
||
|
|
||
|
_event_key(target, identifier, fn).listen(*args, **kw)
|
||
|
|
||
|
|
||
|
def listens_for(target, identifier, *args, **kw):
|
||
|
"""Decorate a function as a listener for the given target + identifier.
|
||
|
|
||
|
e.g.::
|
||
|
|
||
|
from sqlalchemy import event
|
||
|
from sqlalchemy.schema import UniqueConstraint
|
||
|
|
||
|
@event.listens_for(UniqueConstraint, "after_parent_attach")
|
||
|
def unique_constraint_name(const, table):
|
||
|
const.name = "uq_%s_%s" % (
|
||
|
table.name,
|
||
|
list(const.columns)[0].name
|
||
|
)
|
||
|
|
||
|
A given function can also be invoked for only the first invocation
|
||
|
of the event using the ``once`` argument::
|
||
|
|
||
|
@event.listens_for(Mapper, "before_configure", once=True)
|
||
|
def on_config():
|
||
|
do_config()
|
||
|
|
||
|
|
||
|
.. versionadded:: 0.9.4 Added ``once=True`` to :func:`.event.listen`
|
||
|
and :func:`.event.listens_for`.
|
||
|
|
||
|
.. seealso::
|
||
|
|
||
|
:func:`.listen` - general description of event listening
|
||
|
|
||
|
"""
|
||
|
def decorate(fn):
|
||
|
listen(target, identifier, fn, *args, **kw)
|
||
|
return fn
|
||
|
return decorate
|
||
|
|
||
|
|
||
|
def remove(target, identifier, fn):
|
||
|
"""Remove an event listener.
|
||
|
|
||
|
The arguments here should match exactly those which were sent to
|
||
|
:func:`.listen`; all the event registration which proceeded as a result
|
||
|
of this call will be reverted by calling :func:`.remove` with the same
|
||
|
arguments.
|
||
|
|
||
|
e.g.::
|
||
|
|
||
|
# if a function was registered like this...
|
||
|
@event.listens_for(SomeMappedClass, "before_insert", propagate=True)
|
||
|
def my_listener_function(*arg):
|
||
|
pass
|
||
|
|
||
|
# ... it's removed like this
|
||
|
event.remove(SomeMappedClass, "before_insert", my_listener_function)
|
||
|
|
||
|
Above, the listener function associated with ``SomeMappedClass`` was also
|
||
|
propagated to subclasses of ``SomeMappedClass``; the :func:`.remove`
|
||
|
function will revert all of these operations.
|
||
|
|
||
|
.. versionadded:: 0.9.0
|
||
|
|
||
|
.. note::
|
||
|
|
||
|
The :func:`.remove` function cannot be called at the same time
|
||
|
that the target event is being run. This has implications
|
||
|
for thread safety, and also means an event cannot be removed
|
||
|
from inside the listener function for itself. The list of
|
||
|
events to be run are present inside of a mutable collection
|
||
|
that can't be changed during iteration.
|
||
|
|
||
|
Event registration and removal is not intended to be a "high
|
||
|
velocity" operation; it is a configurational operation. For
|
||
|
systems that need to quickly associate and deassociate with
|
||
|
events at high scale, use a mutable structure that is handled
|
||
|
from inside of a single listener.
|
||
|
|
||
|
.. versionchanged:: 1.0.0 - a ``collections.deque()`` object is now
|
||
|
used as the container for the list of events, which explicitly
|
||
|
disallows collection mutation while the collection is being
|
||
|
iterated.
|
||
|
|
||
|
.. seealso::
|
||
|
|
||
|
:func:`.listen`
|
||
|
|
||
|
"""
|
||
|
_event_key(target, identifier, fn).remove()
|
||
|
|
||
|
|
||
|
def contains(target, identifier, fn):
|
||
|
"""Return True if the given target/ident/fn is set up to listen.
|
||
|
|
||
|
.. versionadded:: 0.9.0
|
||
|
|
||
|
"""
|
||
|
|
||
|
return _event_key(target, identifier, fn).contains()
|