U [h@sddlmZddlZddlZddlZddlZddlm Z ej rJddl m Z GdddZ Gdd d ZGd d d eZGd d d eZdddddZdddddZGdddeeejjZdS)) annotationsN)Query) SQLAlchemyc@sReZdZdZejddddddZejddddd dZd dddd dZd S) _QueryPropertyzRA class property that creates a query object for a model. :meta private: Nonez type[Model]r)objclsreturncCsdSNselfrr r r :/tmp/pip-unpacked-wheel-pdy8nmv4/flask_sqlalchemy/model.py__get__sz_QueryProperty.__get__ModelcCsdSr r r r r rrsz Model | NonecCs|j||jdS)N)session) query_class__fsa__rr r r rrsN)__name__ __module__ __qualname____doc__toverloadrr r r rrs rc@sBeZdZUdZded<eZded<eZded<dd d d Z d S) raThe base class of the :attr:`.SQLAlchemy.Model` declarative model class. To define models, subclass :attr:`db.Model <.SQLAlchemy.Model>`, not this. To customize ``db.Model``, subclass this and pass it as ``model_class`` to :class:`.SQLAlchemy`. To customize ``db.Model`` at the metaclass level, pass an already created declarative model class as ``model_class``. zt.ClassVar[SQLAlchemy]rzt.ClassVar[type[Query]]rzt.ClassVar[Query]querystr)r cCspt|}|dk st|jr.dt|d}n*|jrFdt|d}ndtt|j }dt |j d|dS)Nz (transient )z (pending z, < >) sainspectAssertionErrorZ transientidpendingjoinmapridentitytyper)rstatepkr r r__repr__@s  zModel.__repr__N) rrrr__annotations__rrrrr,r r r rr#s   rcs@eZdZUdZded<ded<dddd d d fd d ZZS) BindMetaMixinaMetaclass mixin that sets a model's ``metadata`` based on its ``__bind_key__``. If the model sets ``metadata`` or ``__table__`` directly, ``__bind_key__`` is ignored. If the ``metadata`` is the same as the parent model, it will not be set directly on the child model. rr sa.MetaDatametadatartuple[type, ...]dict[str, t.Any]t.Anyrnamebasesdkwargsr c s^d|jksFd|jksFt|dd}t|dd}|j|}||k rF||_tj|||f|dS)Nr0 __table__Z __bind_key__)__dict__getattrrZ_make_metadatar0super__init__)r r5r6r7r8Zbind_keyZparent_metadatar0 __class__r rr=Ys   zBindMetaMixin.__init__)rrrrr-r= __classcell__r r r>rr.Ns r.csZeZdZUdZded<ded<ded<ddd d d d fd d Zd d ddddZZS) NameMetaMixinaPMetaclass mixin that sets a model's ``__tablename__`` by converting the ``CamelCase`` class name to ``snake_case``. A name is set for non-abstract models that do not otherwise define ``__tablename__``. If a model does not define a primary key, it will not generate a name or ``__table__``, for single-table inheritance. r/r0r __tablename__zsa.Tabler9r1r2r3rr4c sRt|rt|j|_tj|||f|d|jkrNd|jkrN|jddkrN|`dS)NrBr9)should_set_tablenamecamel_to_snake_caserrBr<r=r:r9)r r5r6r7r8r>r rr=rs  zNameMetaMixin.__init__zsa.Table | None)argsr8r cOs|d}|dkr|d}n|d|d}||jjkrFtj||S|D]2}t|tjr`|jslt|tjrJtj||SqJ|j ddD]}d|j krqqtj||Sd|j kr|` dS) aThis is called by SQLAlchemy during mapper setup. It determines the final table object that the model will use. If no primary key is found, that indicates single-table inheritance, so no table will be created and ``__tablename__`` will be unset. schemaNr.rr9rB) getr0Ztablesr!ZTable isinstanceZColumnZ primary_keyZPrimaryKeyConstraint__mro__r:rB)r rEr8rFkeyargbaser r r __table_cls__s&       zNameMetaMixin.__table_cls__)rrrrr-r=rOr@r r r>rrAgs rAr)bool)r r cCs|jdds*tdd|jddDs.dS|jD]T}d|jkrDq4t|jdtjjr^dS||kp|jddpt|tjj SdS) aDetermine whether ``__tablename__`` should be generated for a model. - If no class in the MRO sets a name, one should be generated. - If a declared attr is found, it should be used instead. - If a name is found, it should be used if the class is a mixin, otherwise one should be generated. - Abstract models should not have one generated. Later, ``__table_cls__`` will determine if the model looks like single or joined-table inheritance. If no primary key is found, the name will be unset. Z __abstract__Fcss|]}t|tjjVqdSr )rJr!ormDeclarativeMeta).0br r r sz'should_set_tablename..rNrBT) r:rIanyrKrJr!rQZ declared_attrrR)r rNr r rrCs     rCr)r5r cCstdd|}|dS)z/Convert a ``CamelCase`` name to ``snake_case``.z(((?<=[a-z0-9])[A-Z]|(?!^)[A-Z](?=[a-z]))z_\1_)resublowerlstrip)r5r r rrDsrDc@seZdZdZdS) DefaultMetazgSQLAlchemy declarative metaclass that provides ``__bind_key__`` and ``__tablename__`` support. N)rrrrr r r rr\sr\) __future__rrXtypingrZ sqlalchemyr!Zsqlalchemy.ormrr TYPE_CHECKING extensionrrrr)r.rArCrDrQrRr\r r r rs   +E!