mirror of https://github.com/morpheus65535/bazarr
1142 lines
37 KiB
Python
1142 lines
37 KiB
Python
# orm/state.py
|
|
# Copyright (C) 2005-2023 the SQLAlchemy authors and contributors
|
|
# <see AUTHORS file>
|
|
#
|
|
# This module is part of SQLAlchemy and is released under
|
|
# the MIT License: https://www.opensource.org/licenses/mit-license.php
|
|
|
|
"""Defines instrumentation of instances.
|
|
|
|
This module is usually not directly visible to user applications, but
|
|
defines a large part of the ORM's interactivity.
|
|
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
from typing import Callable
|
|
from typing import Dict
|
|
from typing import Generic
|
|
from typing import Iterable
|
|
from typing import Optional
|
|
from typing import Set
|
|
from typing import Tuple
|
|
from typing import TYPE_CHECKING
|
|
from typing import Union
|
|
import weakref
|
|
|
|
from . import base
|
|
from . import exc as orm_exc
|
|
from . import interfaces
|
|
from ._typing import _O
|
|
from ._typing import is_collection_impl
|
|
from .base import ATTR_WAS_SET
|
|
from .base import INIT_OK
|
|
from .base import LoaderCallableStatus
|
|
from .base import NEVER_SET
|
|
from .base import NO_VALUE
|
|
from .base import PASSIVE_NO_INITIALIZE
|
|
from .base import PASSIVE_NO_RESULT
|
|
from .base import PASSIVE_OFF
|
|
from .base import SQL_OK
|
|
from .path_registry import PathRegistry
|
|
from .. import exc as sa_exc
|
|
from .. import inspection
|
|
from .. import util
|
|
from ..util.typing import Literal
|
|
from ..util.typing import Protocol
|
|
|
|
if TYPE_CHECKING:
|
|
from ._typing import _IdentityKeyType
|
|
from ._typing import _InstanceDict
|
|
from ._typing import _LoaderCallable
|
|
from .attributes import AttributeImpl
|
|
from .attributes import History
|
|
from .base import PassiveFlag
|
|
from .collections import _AdaptedCollectionProtocol
|
|
from .identity import IdentityMap
|
|
from .instrumentation import ClassManager
|
|
from .interfaces import ORMOption
|
|
from .mapper import Mapper
|
|
from .session import Session
|
|
from ..engine import Row
|
|
from ..ext.asyncio.session import async_session as _async_provider
|
|
from ..ext.asyncio.session import AsyncSession
|
|
|
|
if TYPE_CHECKING:
|
|
_sessions: weakref.WeakValueDictionary[int, Session]
|
|
else:
|
|
# late-populated by session.py
|
|
_sessions = None
|
|
|
|
|
|
if not TYPE_CHECKING:
|
|
# optionally late-provided by sqlalchemy.ext.asyncio.session
|
|
|
|
_async_provider = None # noqa
|
|
|
|
|
|
class _InstanceDictProto(Protocol):
|
|
def __call__(self) -> Optional[IdentityMap]:
|
|
...
|
|
|
|
|
|
class _InstallLoaderCallableProto(Protocol[_O]):
|
|
"""used at result loading time to install a _LoaderCallable callable
|
|
upon a specific InstanceState, which will be used to populate an
|
|
attribute when that attribute is accessed.
|
|
|
|
Concrete examples are per-instance deferred column loaders and
|
|
relationship lazy loaders.
|
|
|
|
"""
|
|
|
|
def __call__(
|
|
self, state: InstanceState[_O], dict_: _InstanceDict, row: Row[Any]
|
|
) -> None:
|
|
...
|
|
|
|
|
|
@inspection._self_inspects
|
|
class InstanceState(interfaces.InspectionAttrInfo, Generic[_O]):
|
|
"""tracks state information at the instance level.
|
|
|
|
The :class:`.InstanceState` is a key object used by the
|
|
SQLAlchemy ORM in order to track the state of an object;
|
|
it is created the moment an object is instantiated, typically
|
|
as a result of :term:`instrumentation` which SQLAlchemy applies
|
|
to the ``__init__()`` method of the class.
|
|
|
|
:class:`.InstanceState` is also a semi-public object,
|
|
available for runtime inspection as to the state of a
|
|
mapped instance, including information such as its current
|
|
status within a particular :class:`.Session` and details
|
|
about data on individual attributes. The public API
|
|
in order to acquire a :class:`.InstanceState` object
|
|
is to use the :func:`_sa.inspect` system::
|
|
|
|
>>> from sqlalchemy import inspect
|
|
>>> insp = inspect(some_mapped_object)
|
|
>>> insp.attrs.nickname.history
|
|
History(added=['new nickname'], unchanged=(), deleted=['nickname'])
|
|
|
|
.. seealso::
|
|
|
|
:ref:`orm_mapper_inspection_instancestate`
|
|
|
|
"""
|
|
|
|
__slots__ = (
|
|
"__dict__",
|
|
"__weakref__",
|
|
"class_",
|
|
"manager",
|
|
"obj",
|
|
"committed_state",
|
|
"expired_attributes",
|
|
)
|
|
|
|
manager: ClassManager[_O]
|
|
session_id: Optional[int] = None
|
|
key: Optional[_IdentityKeyType[_O]] = None
|
|
runid: Optional[int] = None
|
|
load_options: Tuple[ORMOption, ...] = ()
|
|
load_path: PathRegistry = PathRegistry.root
|
|
insert_order: Optional[int] = None
|
|
_strong_obj: Optional[object] = None
|
|
obj: weakref.ref[_O]
|
|
|
|
committed_state: Dict[str, Any]
|
|
|
|
modified: bool = False
|
|
expired: bool = False
|
|
_deleted: bool = False
|
|
_load_pending: bool = False
|
|
_orphaned_outside_of_session: bool = False
|
|
is_instance: bool = True
|
|
identity_token: object = None
|
|
_last_known_values: Optional[Dict[str, Any]] = None
|
|
|
|
_instance_dict: _InstanceDictProto
|
|
"""A weak reference, or in the default case a plain callable, that
|
|
returns a reference to the current :class:`.IdentityMap`, if any.
|
|
|
|
"""
|
|
if not TYPE_CHECKING:
|
|
|
|
def _instance_dict(self):
|
|
"""default 'weak reference' for _instance_dict"""
|
|
return None
|
|
|
|
expired_attributes: Set[str]
|
|
"""The set of keys which are 'expired' to be loaded by
|
|
the manager's deferred scalar loader, assuming no pending
|
|
changes.
|
|
|
|
see also the ``unmodified`` collection which is intersected
|
|
against this set when a refresh operation occurs."""
|
|
|
|
callables: Dict[str, Callable[[InstanceState[_O], PassiveFlag], Any]]
|
|
"""A namespace where a per-state loader callable can be associated.
|
|
|
|
In SQLAlchemy 1.0, this is only used for lazy loaders / deferred
|
|
loaders that were set up via query option.
|
|
|
|
Previously, callables was used also to indicate expired attributes
|
|
by storing a link to the InstanceState itself in this dictionary.
|
|
This role is now handled by the expired_attributes set.
|
|
|
|
"""
|
|
|
|
if not TYPE_CHECKING:
|
|
callables = util.EMPTY_DICT
|
|
|
|
def __init__(self, obj: _O, manager: ClassManager[_O]):
|
|
self.class_ = obj.__class__
|
|
self.manager = manager
|
|
self.obj = weakref.ref(obj, self._cleanup)
|
|
self.committed_state = {}
|
|
self.expired_attributes = set()
|
|
|
|
@util.memoized_property
|
|
def attrs(self) -> util.ReadOnlyProperties[AttributeState]:
|
|
"""Return a namespace representing each attribute on
|
|
the mapped object, including its current value
|
|
and history.
|
|
|
|
The returned object is an instance of :class:`.AttributeState`.
|
|
This object allows inspection of the current data
|
|
within an attribute as well as attribute history
|
|
since the last flush.
|
|
|
|
"""
|
|
return util.ReadOnlyProperties(
|
|
{key: AttributeState(self, key) for key in self.manager}
|
|
)
|
|
|
|
@property
|
|
def transient(self) -> bool:
|
|
"""Return ``True`` if the object is :term:`transient`.
|
|
|
|
.. seealso::
|
|
|
|
:ref:`session_object_states`
|
|
|
|
"""
|
|
return self.key is None and not self._attached
|
|
|
|
@property
|
|
def pending(self) -> bool:
|
|
"""Return ``True`` if the object is :term:`pending`.
|
|
|
|
|
|
.. seealso::
|
|
|
|
:ref:`session_object_states`
|
|
|
|
"""
|
|
return self.key is None and self._attached
|
|
|
|
@property
|
|
def deleted(self) -> bool:
|
|
"""Return ``True`` if the object is :term:`deleted`.
|
|
|
|
An object that is in the deleted state is guaranteed to
|
|
not be within the :attr:`.Session.identity_map` of its parent
|
|
:class:`.Session`; however if the session's transaction is rolled
|
|
back, the object will be restored to the persistent state and
|
|
the identity map.
|
|
|
|
.. note::
|
|
|
|
The :attr:`.InstanceState.deleted` attribute refers to a specific
|
|
state of the object that occurs between the "persistent" and
|
|
"detached" states; once the object is :term:`detached`, the
|
|
:attr:`.InstanceState.deleted` attribute **no longer returns
|
|
True**; in order to detect that a state was deleted, regardless
|
|
of whether or not the object is associated with a
|
|
:class:`.Session`, use the :attr:`.InstanceState.was_deleted`
|
|
accessor.
|
|
|
|
.. versionadded: 1.1
|
|
|
|
.. seealso::
|
|
|
|
:ref:`session_object_states`
|
|
|
|
"""
|
|
return self.key is not None and self._attached and self._deleted
|
|
|
|
@property
|
|
def was_deleted(self) -> bool:
|
|
"""Return True if this object is or was previously in the
|
|
"deleted" state and has not been reverted to persistent.
|
|
|
|
This flag returns True once the object was deleted in flush.
|
|
When the object is expunged from the session either explicitly
|
|
or via transaction commit and enters the "detached" state,
|
|
this flag will continue to report True.
|
|
|
|
.. versionadded:: 1.1 - added a local method form of
|
|
:func:`.orm.util.was_deleted`.
|
|
|
|
.. seealso::
|
|
|
|
:attr:`.InstanceState.deleted` - refers to the "deleted" state
|
|
|
|
:func:`.orm.util.was_deleted` - standalone function
|
|
|
|
:ref:`session_object_states`
|
|
|
|
"""
|
|
return self._deleted
|
|
|
|
@property
|
|
def persistent(self) -> bool:
|
|
"""Return ``True`` if the object is :term:`persistent`.
|
|
|
|
An object that is in the persistent state is guaranteed to
|
|
be within the :attr:`.Session.identity_map` of its parent
|
|
:class:`.Session`.
|
|
|
|
.. versionchanged:: 1.1 The :attr:`.InstanceState.persistent`
|
|
accessor no longer returns True for an object that was
|
|
"deleted" within a flush; use the :attr:`.InstanceState.deleted`
|
|
accessor to detect this state. This allows the "persistent"
|
|
state to guarantee membership in the identity map.
|
|
|
|
.. seealso::
|
|
|
|
:ref:`session_object_states`
|
|
|
|
"""
|
|
return self.key is not None and self._attached and not self._deleted
|
|
|
|
@property
|
|
def detached(self) -> bool:
|
|
"""Return ``True`` if the object is :term:`detached`.
|
|
|
|
.. seealso::
|
|
|
|
:ref:`session_object_states`
|
|
|
|
"""
|
|
return self.key is not None and not self._attached
|
|
|
|
@util.non_memoized_property
|
|
@util.preload_module("sqlalchemy.orm.session")
|
|
def _attached(self) -> bool:
|
|
return (
|
|
self.session_id is not None
|
|
and self.session_id in util.preloaded.orm_session._sessions
|
|
)
|
|
|
|
def _track_last_known_value(self, key: str) -> None:
|
|
"""Track the last known value of a particular key after expiration
|
|
operations.
|
|
|
|
.. versionadded:: 1.3
|
|
|
|
"""
|
|
|
|
lkv = self._last_known_values
|
|
if lkv is None:
|
|
self._last_known_values = lkv = {}
|
|
if key not in lkv:
|
|
lkv[key] = NO_VALUE
|
|
|
|
@property
|
|
def session(self) -> Optional[Session]:
|
|
"""Return the owning :class:`.Session` for this instance,
|
|
or ``None`` if none available.
|
|
|
|
Note that the result here can in some cases be *different*
|
|
from that of ``obj in session``; an object that's been deleted
|
|
will report as not ``in session``, however if the transaction is
|
|
still in progress, this attribute will still refer to that session.
|
|
Only when the transaction is completed does the object become
|
|
fully detached under normal circumstances.
|
|
|
|
.. seealso::
|
|
|
|
:attr:`_orm.InstanceState.async_session`
|
|
|
|
"""
|
|
if self.session_id:
|
|
try:
|
|
return _sessions[self.session_id]
|
|
except KeyError:
|
|
pass
|
|
return None
|
|
|
|
@property
|
|
def async_session(self) -> Optional[AsyncSession]:
|
|
"""Return the owning :class:`_asyncio.AsyncSession` for this instance,
|
|
or ``None`` if none available.
|
|
|
|
This attribute is only non-None when the :mod:`sqlalchemy.ext.asyncio`
|
|
API is in use for this ORM object. The returned
|
|
:class:`_asyncio.AsyncSession` object will be a proxy for the
|
|
:class:`_orm.Session` object that would be returned from the
|
|
:attr:`_orm.InstanceState.session` attribute for this
|
|
:class:`_orm.InstanceState`.
|
|
|
|
.. versionadded:: 1.4.18
|
|
|
|
.. seealso::
|
|
|
|
:ref:`asyncio_toplevel`
|
|
|
|
"""
|
|
if _async_provider is None:
|
|
return None
|
|
|
|
sess = self.session
|
|
if sess is not None:
|
|
return _async_provider(sess)
|
|
else:
|
|
return None
|
|
|
|
@property
|
|
def object(self) -> Optional[_O]:
|
|
"""Return the mapped object represented by this
|
|
:class:`.InstanceState`.
|
|
|
|
Returns None if the object has been garbage collected
|
|
|
|
"""
|
|
return self.obj()
|
|
|
|
@property
|
|
def identity(self) -> Optional[Tuple[Any, ...]]:
|
|
"""Return the mapped identity of the mapped object.
|
|
This is the primary key identity as persisted by the ORM
|
|
which can always be passed directly to
|
|
:meth:`_query.Query.get`.
|
|
|
|
Returns ``None`` if the object has no primary key identity.
|
|
|
|
.. note::
|
|
An object which is :term:`transient` or :term:`pending`
|
|
does **not** have a mapped identity until it is flushed,
|
|
even if its attributes include primary key values.
|
|
|
|
"""
|
|
if self.key is None:
|
|
return None
|
|
else:
|
|
return self.key[1]
|
|
|
|
@property
|
|
def identity_key(self) -> Optional[_IdentityKeyType[_O]]:
|
|
"""Return the identity key for the mapped object.
|
|
|
|
This is the key used to locate the object within
|
|
the :attr:`.Session.identity_map` mapping. It contains
|
|
the identity as returned by :attr:`.identity` within it.
|
|
|
|
|
|
"""
|
|
return self.key
|
|
|
|
@util.memoized_property
|
|
def parents(self) -> Dict[int, Union[Literal[False], InstanceState[Any]]]:
|
|
return {}
|
|
|
|
@util.memoized_property
|
|
def _pending_mutations(self) -> Dict[str, PendingCollection]:
|
|
return {}
|
|
|
|
@util.memoized_property
|
|
def _empty_collections(self) -> Dict[str, _AdaptedCollectionProtocol]:
|
|
return {}
|
|
|
|
@util.memoized_property
|
|
def mapper(self) -> Mapper[_O]:
|
|
"""Return the :class:`_orm.Mapper` used for this mapped object."""
|
|
return self.manager.mapper
|
|
|
|
@property
|
|
def has_identity(self) -> bool:
|
|
"""Return ``True`` if this object has an identity key.
|
|
|
|
This should always have the same value as the
|
|
expression ``state.persistent`` or ``state.detached``.
|
|
|
|
"""
|
|
return bool(self.key)
|
|
|
|
@classmethod
|
|
def _detach_states(
|
|
self,
|
|
states: Iterable[InstanceState[_O]],
|
|
session: Session,
|
|
to_transient: bool = False,
|
|
) -> None:
|
|
persistent_to_detached = (
|
|
session.dispatch.persistent_to_detached or None
|
|
)
|
|
deleted_to_detached = session.dispatch.deleted_to_detached or None
|
|
pending_to_transient = session.dispatch.pending_to_transient or None
|
|
persistent_to_transient = (
|
|
session.dispatch.persistent_to_transient or None
|
|
)
|
|
|
|
for state in states:
|
|
deleted = state._deleted
|
|
pending = state.key is None
|
|
persistent = not pending and not deleted
|
|
|
|
state.session_id = None
|
|
|
|
if to_transient and state.key:
|
|
del state.key
|
|
if persistent:
|
|
if to_transient:
|
|
if persistent_to_transient is not None:
|
|
persistent_to_transient(session, state)
|
|
elif persistent_to_detached is not None:
|
|
persistent_to_detached(session, state)
|
|
elif deleted and deleted_to_detached is not None:
|
|
deleted_to_detached(session, state)
|
|
elif pending and pending_to_transient is not None:
|
|
pending_to_transient(session, state)
|
|
|
|
state._strong_obj = None
|
|
|
|
def _detach(self, session: Optional[Session] = None) -> None:
|
|
if session:
|
|
InstanceState._detach_states([self], session)
|
|
else:
|
|
self.session_id = self._strong_obj = None
|
|
|
|
def _dispose(self) -> None:
|
|
# used by the test suite, apparently
|
|
self._detach()
|
|
|
|
def _cleanup(self, ref: weakref.ref[_O]) -> None:
|
|
"""Weakref callback cleanup.
|
|
|
|
This callable cleans out the state when it is being garbage
|
|
collected.
|
|
|
|
this _cleanup **assumes** that there are no strong refs to us!
|
|
Will not work otherwise!
|
|
|
|
"""
|
|
|
|
# Python builtins become undefined during interpreter shutdown.
|
|
# Guard against exceptions during this phase, as the method cannot
|
|
# proceed in any case if builtins have been undefined.
|
|
if dict is None:
|
|
return
|
|
|
|
instance_dict = self._instance_dict()
|
|
if instance_dict is not None:
|
|
instance_dict._fast_discard(self)
|
|
del self._instance_dict
|
|
|
|
# we can't possibly be in instance_dict._modified
|
|
# b.c. this is weakref cleanup only, that set
|
|
# is strong referencing!
|
|
# assert self not in instance_dict._modified
|
|
|
|
self.session_id = self._strong_obj = None
|
|
|
|
@property
|
|
def dict(self) -> _InstanceDict:
|
|
"""Return the instance dict used by the object.
|
|
|
|
Under normal circumstances, this is always synonymous
|
|
with the ``__dict__`` attribute of the mapped object,
|
|
unless an alternative instrumentation system has been
|
|
configured.
|
|
|
|
In the case that the actual object has been garbage
|
|
collected, this accessor returns a blank dictionary.
|
|
|
|
"""
|
|
o = self.obj()
|
|
if o is not None:
|
|
return base.instance_dict(o)
|
|
else:
|
|
return {}
|
|
|
|
def _initialize_instance(*mixed: Any, **kwargs: Any) -> None:
|
|
self, instance, args = mixed[0], mixed[1], mixed[2:] # noqa
|
|
manager = self.manager
|
|
|
|
manager.dispatch.init(self, args, kwargs)
|
|
|
|
try:
|
|
manager.original_init(*mixed[1:], **kwargs)
|
|
except:
|
|
with util.safe_reraise():
|
|
manager.dispatch.init_failure(self, args, kwargs)
|
|
|
|
def get_history(self, key: str, passive: PassiveFlag) -> History:
|
|
return self.manager[key].impl.get_history(self, self.dict, passive)
|
|
|
|
def get_impl(self, key: str) -> AttributeImpl:
|
|
return self.manager[key].impl
|
|
|
|
def _get_pending_mutation(self, key: str) -> PendingCollection:
|
|
if key not in self._pending_mutations:
|
|
self._pending_mutations[key] = PendingCollection()
|
|
return self._pending_mutations[key]
|
|
|
|
def __getstate__(self) -> Dict[str, Any]:
|
|
state_dict = {
|
|
"instance": self.obj(),
|
|
"class_": self.class_,
|
|
"committed_state": self.committed_state,
|
|
"expired_attributes": self.expired_attributes,
|
|
}
|
|
state_dict.update(
|
|
(k, self.__dict__[k])
|
|
for k in (
|
|
"_pending_mutations",
|
|
"modified",
|
|
"expired",
|
|
"callables",
|
|
"key",
|
|
"parents",
|
|
"load_options",
|
|
"class_",
|
|
"expired_attributes",
|
|
"info",
|
|
)
|
|
if k in self.__dict__
|
|
)
|
|
if self.load_path:
|
|
state_dict["load_path"] = self.load_path.serialize()
|
|
|
|
state_dict["manager"] = self.manager._serialize(self, state_dict)
|
|
|
|
return state_dict
|
|
|
|
def __setstate__(self, state_dict: Dict[str, Any]) -> None:
|
|
inst = state_dict["instance"]
|
|
if inst is not None:
|
|
self.obj = weakref.ref(inst, self._cleanup)
|
|
self.class_ = inst.__class__
|
|
else:
|
|
self.obj = lambda: None # type: ignore
|
|
self.class_ = state_dict["class_"]
|
|
|
|
self.committed_state = state_dict.get("committed_state", {})
|
|
self._pending_mutations = state_dict.get("_pending_mutations", {}) # type: ignore # noqa E501
|
|
self.parents = state_dict.get("parents", {}) # type: ignore
|
|
self.modified = state_dict.get("modified", False)
|
|
self.expired = state_dict.get("expired", False)
|
|
if "info" in state_dict:
|
|
self.info.update(state_dict["info"])
|
|
if "callables" in state_dict:
|
|
self.callables = state_dict["callables"]
|
|
|
|
self.expired_attributes = state_dict["expired_attributes"]
|
|
else:
|
|
if "expired_attributes" in state_dict:
|
|
self.expired_attributes = state_dict["expired_attributes"]
|
|
else:
|
|
self.expired_attributes = set()
|
|
|
|
self.__dict__.update(
|
|
[
|
|
(k, state_dict[k])
|
|
for k in ("key", "load_options")
|
|
if k in state_dict
|
|
]
|
|
)
|
|
if self.key:
|
|
self.identity_token = self.key[2]
|
|
|
|
if "load_path" in state_dict:
|
|
self.load_path = PathRegistry.deserialize(state_dict["load_path"])
|
|
|
|
state_dict["manager"](self, inst, state_dict)
|
|
|
|
def _reset(self, dict_: _InstanceDict, key: str) -> None:
|
|
"""Remove the given attribute and any
|
|
callables associated with it."""
|
|
|
|
old = dict_.pop(key, None)
|
|
manager_impl = self.manager[key].impl
|
|
if old is not None and is_collection_impl(manager_impl):
|
|
manager_impl._invalidate_collection(old)
|
|
self.expired_attributes.discard(key)
|
|
if self.callables:
|
|
self.callables.pop(key, None)
|
|
|
|
def _copy_callables(self, from_: InstanceState[Any]) -> None:
|
|
if "callables" in from_.__dict__:
|
|
self.callables = dict(from_.callables)
|
|
|
|
@classmethod
|
|
def _instance_level_callable_processor(
|
|
cls, manager: ClassManager[_O], fn: _LoaderCallable, key: Any
|
|
) -> _InstallLoaderCallableProto[_O]:
|
|
impl = manager[key].impl
|
|
if is_collection_impl(impl):
|
|
fixed_impl = impl
|
|
|
|
def _set_callable(
|
|
state: InstanceState[_O], dict_: _InstanceDict, row: Row[Any]
|
|
) -> None:
|
|
if "callables" not in state.__dict__:
|
|
state.callables = {}
|
|
old = dict_.pop(key, None)
|
|
if old is not None:
|
|
fixed_impl._invalidate_collection(old)
|
|
state.callables[key] = fn
|
|
|
|
else:
|
|
|
|
def _set_callable(
|
|
state: InstanceState[_O], dict_: _InstanceDict, row: Row[Any]
|
|
) -> None:
|
|
if "callables" not in state.__dict__:
|
|
state.callables = {}
|
|
state.callables[key] = fn
|
|
|
|
return _set_callable
|
|
|
|
def _expire(
|
|
self, dict_: _InstanceDict, modified_set: Set[InstanceState[Any]]
|
|
) -> None:
|
|
self.expired = True
|
|
if self.modified:
|
|
modified_set.discard(self)
|
|
self.committed_state.clear()
|
|
self.modified = False
|
|
|
|
self._strong_obj = None
|
|
|
|
if "_pending_mutations" in self.__dict__:
|
|
del self.__dict__["_pending_mutations"]
|
|
|
|
if "parents" in self.__dict__:
|
|
del self.__dict__["parents"]
|
|
|
|
self.expired_attributes.update(
|
|
[impl.key for impl in self.manager._loader_impls]
|
|
)
|
|
|
|
if self.callables:
|
|
# the per state loader callables we can remove here are
|
|
# LoadDeferredColumns, which undefers a column at the instance
|
|
# level that is mapped with deferred, and LoadLazyAttribute,
|
|
# which lazy loads a relationship at the instance level that
|
|
# is mapped with "noload" or perhaps "immediateload".
|
|
# Before 1.4, only column-based
|
|
# attributes could be considered to be "expired", so here they
|
|
# were the only ones "unexpired", which means to make them deferred
|
|
# again. For the moment, as of 1.4 we also apply the same
|
|
# treatment relationships now, that is, an instance level lazy
|
|
# loader is reset in the same way as a column loader.
|
|
for k in self.expired_attributes.intersection(self.callables):
|
|
del self.callables[k]
|
|
|
|
for k in self.manager._collection_impl_keys.intersection(dict_):
|
|
collection = dict_.pop(k)
|
|
collection._sa_adapter.invalidated = True
|
|
|
|
if self._last_known_values:
|
|
self._last_known_values.update(
|
|
{k: dict_[k] for k in self._last_known_values if k in dict_}
|
|
)
|
|
|
|
for key in self.manager._all_key_set.intersection(dict_):
|
|
del dict_[key]
|
|
|
|
self.manager.dispatch.expire(self, None)
|
|
|
|
def _expire_attributes(
|
|
self,
|
|
dict_: _InstanceDict,
|
|
attribute_names: Iterable[str],
|
|
no_loader: bool = False,
|
|
) -> None:
|
|
pending = self.__dict__.get("_pending_mutations", None)
|
|
|
|
callables = self.callables
|
|
|
|
for key in attribute_names:
|
|
impl = self.manager[key].impl
|
|
if impl.accepts_scalar_loader:
|
|
if no_loader and (impl.callable_ or key in callables):
|
|
continue
|
|
|
|
self.expired_attributes.add(key)
|
|
if callables and key in callables:
|
|
del callables[key]
|
|
old = dict_.pop(key, NO_VALUE)
|
|
if is_collection_impl(impl) and old is not NO_VALUE:
|
|
impl._invalidate_collection(old)
|
|
|
|
lkv = self._last_known_values
|
|
if lkv is not None and key in lkv and old is not NO_VALUE:
|
|
lkv[key] = old
|
|
|
|
self.committed_state.pop(key, None)
|
|
if pending:
|
|
pending.pop(key, None)
|
|
|
|
self.manager.dispatch.expire(self, attribute_names)
|
|
|
|
def _load_expired(
|
|
self, state: InstanceState[_O], passive: PassiveFlag
|
|
) -> LoaderCallableStatus:
|
|
"""__call__ allows the InstanceState to act as a deferred
|
|
callable for loading expired attributes, which is also
|
|
serializable (picklable).
|
|
|
|
"""
|
|
|
|
if not passive & SQL_OK:
|
|
return PASSIVE_NO_RESULT
|
|
|
|
toload = self.expired_attributes.intersection(self.unmodified)
|
|
toload = toload.difference(
|
|
attr
|
|
for attr in toload
|
|
if not self.manager[attr].impl.load_on_unexpire
|
|
)
|
|
|
|
self.manager.expired_attribute_loader(self, toload, passive)
|
|
|
|
# if the loader failed, or this
|
|
# instance state didn't have an identity,
|
|
# the attributes still might be in the callables
|
|
# dict. ensure they are removed.
|
|
self.expired_attributes.clear()
|
|
|
|
return ATTR_WAS_SET
|
|
|
|
@property
|
|
def unmodified(self) -> Set[str]:
|
|
"""Return the set of keys which have no uncommitted changes"""
|
|
|
|
return set(self.manager).difference(self.committed_state)
|
|
|
|
def unmodified_intersection(self, keys: Iterable[str]) -> Set[str]:
|
|
"""Return self.unmodified.intersection(keys)."""
|
|
|
|
return (
|
|
set(keys)
|
|
.intersection(self.manager)
|
|
.difference(self.committed_state)
|
|
)
|
|
|
|
@property
|
|
def unloaded(self) -> Set[str]:
|
|
"""Return the set of keys which do not have a loaded value.
|
|
|
|
This includes expired attributes and any other attribute that
|
|
was never populated or modified.
|
|
|
|
"""
|
|
return (
|
|
set(self.manager)
|
|
.difference(self.committed_state)
|
|
.difference(self.dict)
|
|
)
|
|
|
|
@property
|
|
def unloaded_expirable(self) -> Set[str]:
|
|
"""Return the set of keys which do not have a loaded value.
|
|
|
|
This includes expired attributes and any other attribute that
|
|
was never populated or modified.
|
|
|
|
"""
|
|
return self.unloaded
|
|
|
|
@property
|
|
def _unloaded_non_object(self) -> Set[str]:
|
|
return self.unloaded.intersection(
|
|
attr
|
|
for attr in self.manager
|
|
if self.manager[attr].impl.accepts_scalar_loader
|
|
)
|
|
|
|
def _modified_event(
|
|
self,
|
|
dict_: _InstanceDict,
|
|
attr: Optional[AttributeImpl],
|
|
previous: Any,
|
|
collection: bool = False,
|
|
is_userland: bool = False,
|
|
) -> None:
|
|
if attr:
|
|
if not attr.send_modified_events:
|
|
return
|
|
if is_userland and attr.key not in dict_:
|
|
raise sa_exc.InvalidRequestError(
|
|
"Can't flag attribute '%s' modified; it's not present in "
|
|
"the object state" % attr.key
|
|
)
|
|
if attr.key not in self.committed_state or is_userland:
|
|
if collection:
|
|
if TYPE_CHECKING:
|
|
assert is_collection_impl(attr)
|
|
if previous is NEVER_SET:
|
|
if attr.key in dict_:
|
|
previous = dict_[attr.key]
|
|
|
|
if previous not in (None, NO_VALUE, NEVER_SET):
|
|
previous = attr.copy(previous)
|
|
self.committed_state[attr.key] = previous
|
|
|
|
lkv = self._last_known_values
|
|
if lkv is not None and attr.key in lkv:
|
|
lkv[attr.key] = NO_VALUE
|
|
|
|
# assert self._strong_obj is None or self.modified
|
|
|
|
if (self.session_id and self._strong_obj is None) or not self.modified:
|
|
self.modified = True
|
|
instance_dict = self._instance_dict()
|
|
if instance_dict:
|
|
has_modified = bool(instance_dict._modified)
|
|
instance_dict._modified.add(self)
|
|
else:
|
|
has_modified = False
|
|
|
|
# only create _strong_obj link if attached
|
|
# to a session
|
|
|
|
inst = self.obj()
|
|
if self.session_id:
|
|
self._strong_obj = inst
|
|
|
|
# if identity map already had modified objects,
|
|
# assume autobegin already occurred, else check
|
|
# for autobegin
|
|
if not has_modified:
|
|
# inline of autobegin, to ensure session transaction
|
|
# snapshot is established
|
|
try:
|
|
session = _sessions[self.session_id]
|
|
except KeyError:
|
|
pass
|
|
else:
|
|
if session._transaction is None:
|
|
session._autobegin_t()
|
|
|
|
if inst is None and attr:
|
|
raise orm_exc.ObjectDereferencedError(
|
|
"Can't emit change event for attribute '%s' - "
|
|
"parent object of type %s has been garbage "
|
|
"collected."
|
|
% (self.manager[attr.key], base.state_class_str(self))
|
|
)
|
|
|
|
def _commit(self, dict_: _InstanceDict, keys: Iterable[str]) -> None:
|
|
"""Commit attributes.
|
|
|
|
This is used by a partial-attribute load operation to mark committed
|
|
those attributes which were refreshed from the database.
|
|
|
|
Attributes marked as "expired" can potentially remain "expired" after
|
|
this step if a value was not populated in state.dict.
|
|
|
|
"""
|
|
for key in keys:
|
|
self.committed_state.pop(key, None)
|
|
|
|
self.expired = False
|
|
|
|
self.expired_attributes.difference_update(
|
|
set(keys).intersection(dict_)
|
|
)
|
|
|
|
# the per-keys commit removes object-level callables,
|
|
# while that of commit_all does not. it's not clear
|
|
# if this behavior has a clear rationale, however tests do
|
|
# ensure this is what it does.
|
|
if self.callables:
|
|
for key in (
|
|
set(self.callables).intersection(keys).intersection(dict_)
|
|
):
|
|
del self.callables[key]
|
|
|
|
def _commit_all(
|
|
self, dict_: _InstanceDict, instance_dict: Optional[IdentityMap] = None
|
|
) -> None:
|
|
"""commit all attributes unconditionally.
|
|
|
|
This is used after a flush() or a full load/refresh
|
|
to remove all pending state from the instance.
|
|
|
|
- all attributes are marked as "committed"
|
|
- the "strong dirty reference" is removed
|
|
- the "modified" flag is set to False
|
|
- any "expired" markers for scalar attributes loaded are removed.
|
|
- lazy load callables for objects / collections *stay*
|
|
|
|
Attributes marked as "expired" can potentially remain
|
|
"expired" after this step if a value was not populated in state.dict.
|
|
|
|
"""
|
|
self._commit_all_states([(self, dict_)], instance_dict)
|
|
|
|
@classmethod
|
|
def _commit_all_states(
|
|
self,
|
|
iter_: Iterable[Tuple[InstanceState[Any], _InstanceDict]],
|
|
instance_dict: Optional[IdentityMap] = None,
|
|
) -> None:
|
|
"""Mass / highly inlined version of commit_all()."""
|
|
|
|
for state, dict_ in iter_:
|
|
state_dict = state.__dict__
|
|
|
|
state.committed_state.clear()
|
|
|
|
if "_pending_mutations" in state_dict:
|
|
del state_dict["_pending_mutations"]
|
|
|
|
state.expired_attributes.difference_update(dict_)
|
|
|
|
if instance_dict and state.modified:
|
|
instance_dict._modified.discard(state)
|
|
|
|
state.modified = state.expired = False
|
|
state._strong_obj = None
|
|
|
|
|
|
class AttributeState:
|
|
"""Provide an inspection interface corresponding
|
|
to a particular attribute on a particular mapped object.
|
|
|
|
The :class:`.AttributeState` object is accessed
|
|
via the :attr:`.InstanceState.attrs` collection
|
|
of a particular :class:`.InstanceState`::
|
|
|
|
from sqlalchemy import inspect
|
|
|
|
insp = inspect(some_mapped_object)
|
|
attr_state = insp.attrs.some_attribute
|
|
|
|
"""
|
|
|
|
__slots__ = ("state", "key")
|
|
|
|
state: InstanceState[Any]
|
|
key: str
|
|
|
|
def __init__(self, state: InstanceState[Any], key: str):
|
|
self.state = state
|
|
self.key = key
|
|
|
|
@property
|
|
def loaded_value(self) -> Any:
|
|
"""The current value of this attribute as loaded from the database.
|
|
|
|
If the value has not been loaded, or is otherwise not present
|
|
in the object's dictionary, returns NO_VALUE.
|
|
|
|
"""
|
|
return self.state.dict.get(self.key, NO_VALUE)
|
|
|
|
@property
|
|
def value(self) -> Any:
|
|
"""Return the value of this attribute.
|
|
|
|
This operation is equivalent to accessing the object's
|
|
attribute directly or via ``getattr()``, and will fire
|
|
off any pending loader callables if needed.
|
|
|
|
"""
|
|
return self.state.manager[self.key].__get__(
|
|
self.state.obj(), self.state.class_
|
|
)
|
|
|
|
@property
|
|
def history(self) -> History:
|
|
"""Return the current **pre-flush** change history for
|
|
this attribute, via the :class:`.History` interface.
|
|
|
|
This method will **not** emit loader callables if the value of the
|
|
attribute is unloaded.
|
|
|
|
.. note::
|
|
|
|
The attribute history system tracks changes on a **per flush
|
|
basis**. Each time the :class:`.Session` is flushed, the history
|
|
of each attribute is reset to empty. The :class:`.Session` by
|
|
default autoflushes each time a :class:`_query.Query` is invoked.
|
|
For
|
|
options on how to control this, see :ref:`session_flushing`.
|
|
|
|
|
|
.. seealso::
|
|
|
|
:meth:`.AttributeState.load_history` - retrieve history
|
|
using loader callables if the value is not locally present.
|
|
|
|
:func:`.attributes.get_history` - underlying function
|
|
|
|
"""
|
|
return self.state.get_history(self.key, PASSIVE_NO_INITIALIZE)
|
|
|
|
def load_history(self) -> History:
|
|
"""Return the current **pre-flush** change history for
|
|
this attribute, via the :class:`.History` interface.
|
|
|
|
This method **will** emit loader callables if the value of the
|
|
attribute is unloaded.
|
|
|
|
.. note::
|
|
|
|
The attribute history system tracks changes on a **per flush
|
|
basis**. Each time the :class:`.Session` is flushed, the history
|
|
of each attribute is reset to empty. The :class:`.Session` by
|
|
default autoflushes each time a :class:`_query.Query` is invoked.
|
|
For
|
|
options on how to control this, see :ref:`session_flushing`.
|
|
|
|
.. seealso::
|
|
|
|
:attr:`.AttributeState.history`
|
|
|
|
:func:`.attributes.get_history` - underlying function
|
|
|
|
.. versionadded:: 0.9.0
|
|
|
|
"""
|
|
return self.state.get_history(self.key, PASSIVE_OFF ^ INIT_OK)
|
|
|
|
|
|
class PendingCollection:
|
|
"""A writable placeholder for an unloaded collection.
|
|
|
|
Stores items appended to and removed from a collection that has not yet
|
|
been loaded. When the collection is loaded, the changes stored in
|
|
PendingCollection are applied to it to produce the final result.
|
|
|
|
"""
|
|
|
|
__slots__ = ("deleted_items", "added_items")
|
|
|
|
deleted_items: util.IdentitySet
|
|
added_items: util.OrderedIdentitySet
|
|
|
|
def __init__(self) -> None:
|
|
self.deleted_items = util.IdentitySet()
|
|
self.added_items = util.OrderedIdentitySet()
|
|
|
|
def append(self, value: Any) -> None:
|
|
if value in self.deleted_items:
|
|
self.deleted_items.remove(value)
|
|
else:
|
|
self.added_items.add(value)
|
|
|
|
def remove(self, value: Any) -> None:
|
|
if value in self.added_items:
|
|
self.added_items.remove(value)
|
|
else:
|
|
self.deleted_items.add(value)
|