a `gb @sddlmZddlZddlZddlmZddl m Z ddl m Z ejrRddlmZGddde jZdd d d d d ZddddZdS)) annotationsN)app_ctx) SQLAlchemycsFeZdZdZddddfdd Zdd d d dd d fd d ZZS)SessionaA SQLAlchemy :class:`~sqlalchemy.orm.Session` class that chooses what engine to use based on the bind key associated with the metadata associated with the thing being queried. To customize ``db.session``, subclass this and pass it as the ``class_`` key in the ``session_options`` to :class:`.SQLAlchemy`. .. versionchanged:: 3.0 Renamed from ``SignallingSession``. rzt.AnyNone)dbkwargsreturnc s"tjfi|||_i|_dS)N)super__init___dbZ_model_changes)selfrr  __class__f/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/flask_sqlalchemy/session.pyr szSession.__init__Nz t.Any | Nonez.sa.engine.Engine | sa.engine.Connection | Nonez'sa.engine.Engine | sa.engine.Connection)mapperclausebindr r c s|dur |S|jj}|durzt|}Wn@tjyj}z&t|trTtj ||WYd}~n d}~00t |j |}|dur|S|durt ||}|dur|Sd|vr|dSt jf|||d|S)aSelect an engine based on the ``bind_key`` of the metadata associated with the model or table being queried. If no bind key is set, uses the default bind. .. versionchanged:: 3.0.3 Fix finding the bind for a joined inheritance model. .. versionchanged:: 3.0 The implementation more closely matches the base SQLAlchemy implementation. .. versionchanged:: 2.1 Support joining an external transaction. N)rrr)r enginessainspectsa_excZNoInspectionAvailable isinstancetypesa_ormexcZUnmappedClassError_clause_to_engineZ local_tabler get_bind)rrrrr reZenginerrrrs(   zSession.get_bind)NNN)__name__ __module__ __qualname____doc__r r __classcell__rrrrrs  rzsa.ClauseElement | Nonez't.Mapping[str | None, sa.engine.Engine]zsa.engine.Engine | None)rrr cCsd}|dur>t|tjr|}n t|tjr>t|jtjr>|j}|durd|jjvr|jjd}||vrxtd|d||SdS)zeIf the clause is a table, return the engine associated with the table's metadata's bind key. NZbind_keyz Bind key 'z&' is not in 'SQLALCHEMY_BINDS' config.) rrTableZ UpdateBasetablemetadatainforZUnboundExecutionError)rrr'keyrrrrQs   rint)r cCs ttS)zJGet the id of the current Flask application context for the session scope.)idrZ_get_current_objectrrrr _app_ctx_idmsr-) __future__rtypingtZ sqlalchemyrZsqlalchemy.excrrZsqlalchemy.ormZormrZ flask.globalsr TYPE_CHECKING extensionrrrr-rrrrs     C