관리-도구
편집 파일: decl_api.cpython-38.pyc
U -?�f�� � @ s� d Z ddlmZ ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlZddlmZ ddlmZ ddlm Z ddlm Z ddlm Z dd l m Z dd l m!Z! dd l m"Z" dd l m#Z# dd l m$Z$ dd!lm%Z% dd"l&m'Z' dd#l&m(Z( dd$l&m)Z) dd%l&m*Z* dd&l+m,Z, dd'l+m-Z- dd(l+m.Z. dd)l+m/Z/ dd*l+m0Z0 dd+l+m1Z1 dd,l+m2Z2 dd-l3m4Z4 dd.l3m5Z5 dd.l3m5Z6 dd/l7m8Z8 dd0l9m:Z: dd1l;m<Z< dd2l=m>Z> d3d4lm?Z? d3d5lm@Z@ d3d6lmAZA d3d7lBmCZC d3d8lDmEZE d3d9lFmGZG d3d:lHmIZI d3d;lJmKZK d3d<lAmLZL d3d=lAmMZM d3d>lAmZN d3d?lOmPZP d3d@lOmQZQ d3dAlOmRZR d3dBlOmSZS d3dClOmTZT d3dDlOmUZU d3dElOmVZV d3dFlOmWZW e�r�ddGlXmYZY ddHlXmZZZ ddIl+m[Z[ ddJl m\Z\ ddKl m]Z] dd2l=m>Z> d3dLl^m_Z_ d3dMl`maZa edNedO�ZbedPedO�ZceedQf Zde edQf ZeedRee)eb e*eb eGeb f f ZfdSdTdU�dVdW�ZgG dXdY� dYeh�ZiG dZd[� d[eie@jje8e �ZkeNjle:e<e4e5e"e#e e$e!f d\�G d]d^� d^ek��ZmG d_d`� d`ek�Znd�dbdTdcdd�dedf�ZoG dgdh� dh�ZpG didj� djepe eb �ZqG dkdl� dle jreb ep�ZsG dmdn� dneseb �ZtdododU�dpdq�ZudrdsdU�dtdu�ZvG dvdw� dwemdx�ZwG dydz� dze@jje>e ekdx�Zxdrdrdsd{�d|d}�ZyG d~d� de@jje>e �ZzdSdbd�dsd��d�d��Z{dde|d�dde/end��d�d�drdbd�d�d�drd�d�� d�d��Z}G d�d�� d��Z~e~Ze�s�e~ZZd�d�d��d�d��Z�e@��enexek�drd�dU�d�d���Z�dS )�z1Public API functions and helpers for declarative.� )� annotationsN)�Any)�Callable)�cast)�ClassVar)�Dict)� FrozenSet)�Generic)�Iterable)�Iterator)�Mapping)�Optional)�overload)�Set)�Tuple)�Type)� TYPE_CHECKING)�TypeVar)�Union� )� attributes)� clsregistry)�instrumentation)� interfaces)� mapperlib)� composite)�deferred)� mapped_column)� relationship)�synonym)�InstrumentedAttribute)�_inspect_mapped_class)�_is_mapped_class)�Mapped)� ORMDescriptor��_add_attribute)�_as_declarative)�_ClassScanMapperConfig)�_declarative_constructor)�_DeferredMapperConfig)�_del_attribute��_mapper)� Composite)�Synonym)�Mapper)� MappedColumn)�RelationshipProperty)� InstanceState� )�exc)� inspection)�util)�sqltypes)�_NoArg)�SQLCoreOperations)�MetaData)� FromClause)� hybridmethod)�hybridproperty)�typing)�CallableReference)�flatten_newtype)� is_generic)� is_literal)� is_newtype)� is_pep695)�Literal)�Self)�_O)� _RegistryType)�_DataclassArguments)� ClassManager)�MapperProperty)�_TypeEngineArgument)�_MatchedOnType�_T)�bound�_TTz_TypeEngineArgument[Any].�Type[_O]�bool��cls�returnc C s. | j dd� D ]}t|dd�dk r dS qdS )aK Given a class, return True if any of the classes it inherits from has a mapped table, otherwise return False. This is used in declarative mixins to build attributes that behave differently for the base class vs. a subclass in an inheritance hierarchy. .. seealso:: :ref:`decl_mixin_inheritance` r N� __table__TF)�__mro__�getattr)rU �class_� r[ �G/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/orm/decl_api.py�has_inherited_tablel s r] c @ s. e Zd Zdddd�dd�Zddd�dd �Zd S ) �_DynamicAttributesType�strr �None)�key�valuerV c C s* d| j krt| ||� nt�| ||� d S �N� __mapper__)�__dict__r& �type� __setattr__)rU ra rb r[ r[ r\ rg � s z"_DynamicAttributesType.__setattr__)ra rV c C s& d| j krt| |� n t�| |� d S rc )re r+ rf � __delattr__�rU ra r[ r[ r\ rh � s z"_DynamicAttributesType.__delattr__N)�__name__� __module__� __qualname__rg rh r[ r[ r[ r\ r^ s r^ c @ s e Zd ZdZdS )� DeclarativeAttributeInterceptz�Metaclass that may be used in conjunction with the :class:`_orm.DeclarativeBase` class to support addition of class attributes dynamically. N�rj rk rl �__doc__r[ r[ r[ r\ rm � s rm ��field_specifiersc @ s e Zd ZdZdS )�DCTransformDeclarativez-metaclass that includes @dataclass_transformsNrn r[ r[ r[ r\ rr � s rr c @ s4 e Zd ZU ded< ded< dddddd�dd �Zd S ) �DeclarativeMetar; �metadata� RegistryType�registryr r` )� classname�bases�dict_�kwrV c K sp | j }t| dd �}|d krB|�dd �}t|t�s<t�d��n|| _| j �dd�s\t|| |� t � | |||� d S )N� _sa_registryrv ziDeclarative base class has no 'registry' attribute, or registry is not a sqlalchemy.orm.registry() object� __abstract__F) re rY �get� isinstancerv r5 �InvalidRequestErrorr{ r' rf �__init__)rU rw rx ry rz �regr[ r[ r\ r� � s � zDeclarativeMeta.__init__N)rj rk rl �__annotations__r� r[ r[ r[ r\ rs � s rs Fr_ z,Callable[[Callable[..., Any]], Synonym[Any]])�name� map_columnrV c s ddd�� �fdd� }|S )a Decorator that produces an :func:`_orm.synonym` attribute in conjunction with a Python descriptor. The function being decorated is passed to :func:`_orm.synonym` as the :paramref:`.orm.synonym.descriptor` parameter:: class MyClass(Base): __tablename__ = 'my_table' id = Column(Integer, primary_key=True) _job_status = Column("job_status", String(50)) @synonym_for("job_status") @property def job_status(self): return "Status: %s" % self._job_status The :ref:`hybrid properties <mapper_hybrids>` feature of SQLAlchemy is typically preferred instead of synonyms, which is a more legacy feature. .. seealso:: :ref:`synonyms` - Overview of synonyms :func:`_orm.synonym` - the mapper-level function :ref:`mapper_hybrids` - The Hybrid Attribute extension provides an updated approach to augmenting attribute behavior more flexibly than can be achieved with synonyms. �Callable[..., Any]z Synonym[Any]��fnrV c s t �� | d�S )N)r� Z descriptor)� _orm_synonym)r� �r� r� r[ r\ �decorate� s z synonym_for.<locals>.decorater[ )r� r� r� r[ r� r\ � synonym_for� s $r� c @ s@ e Zd Zddddd�dd�Zdd�d d �Zd d d d �dd�ZdS )�_declared_attr_commonFr� rS )r� � cascading�quietc C s. t |t�r|j}|| _|| _|| _|j| _d S �N)r~ � classmethod�__func__�fgetZ _cascadingZ_quietro )�selfr� r� r� r[ r[ r\ r� � s z _declared_attr_common.__init__zOptional[Type[Any]]�rV c C s t �| j��d�S )NrV )r7 Zget_annotationsr� r} �r� r[ r[ r\ �_collect_return_annotation s z0_declared_attr_common._collect_return_annotation�Optional[object]r ��instance�ownerrV c C s� |}t �|�}|d krHt�d| jj�s>t�d| jj|jf � | �|�S |jrX| �|�S |� � }|d k slt �|j }| |kr�|| S | �|� || < }|S d S )Nz^__.+__$zEUnmanaged access of declarative attribute %s from non-mapped class %s) r Zopt_manager_of_class�re�matchr� rj r7 �warn� is_mapped�declarative_scan�AssertionErrorZdeclared_attr_reg)r� r� r� rU �managerr� r� �objr[ r[ r\ �__get__ s& �� z _declared_attr_common.__get__N)FF)rj rk rl r� r� r� r[ r[ r[ r\ r� � s �r� c @ sh e Zd Zejrddddd�dd�Zddd d �d d �Zddd d�dd�Zdd d�dd�Zddd�dd�Z dS )�_declared_directiveFzCallable[..., _T]rS �r� r� c C s d S r� r[ �r� r� r� r[ r[ r\ r� 8 s z _declared_directive.__init__r� r rO r� c C s d S r� r[ �r� r� r� r[ r[ r\ r� > � z_declared_directive.__get__r` �r� rb rV c C s d S r� r[ �r� r� rb r[ r[ r\ �__set__@ r� z_declared_directive.__set__�r� rV c C s d S r� r[ �r� r� r[ r[ r\ � __delete__B r� z _declared_directive.__delete__zCallable[..., _TT]z_declared_directive[_TT]r� c C s d S r� r[ �r� r� r[ r[ r\ �__call__D s z _declared_directive.__call__N)F) rj rk rl r? r r� r� r� r� r� r[ r[ r[ r\ r� 3 s �r� c @ s� e Zd ZdZejr�d%ddd�dd�Zddd d �d d �Zdd d �dd�Ze d ddd�dd��Z e dddd�dd��Z dddd�dd�Z e ddd�d d ��Z e d d �d d!��Ze dd �d"d#��Zd$S )&� declared_attra Mark a class-level method as representing the definition of a mapped property or Declarative directive. :class:`_orm.declared_attr` is typically applied as a decorator to a class level method, turning the attribute into a scalar-like property that can be invoked from the uninstantiated class. The Declarative mapping process looks for these :class:`_orm.declared_attr` callables as it scans classes, and assumes any attribute marked with :class:`_orm.declared_attr` will be a callable that will produce an object specific to the Declarative mapping or table configuration. :class:`_orm.declared_attr` is usually applicable to :ref:`mixins <orm_mixins_toplevel>`, to define relationships that are to be applied to different implementors of the class. It may also be used to define dynamically generated column expressions and other Declarative attributes. Example:: class ProvidesUserMixin: "A mixin that adds a 'user' relationship to classes." user_id: Mapped[int] = mapped_column(ForeignKey("user_table.id")) @declared_attr def user(cls) -> Mapped["User"]: return relationship("User") When used with Declarative directives such as ``__tablename__``, the :meth:`_orm.declared_attr.directive` modifier may be used which indicates to :pep:`484` typing tools that the given method is not dealing with :class:`_orm.Mapped` attributes:: class CreateTableName: @declared_attr.directive def __tablename__(cls) -> str: return cls.__name__.lower() :class:`_orm.declared_attr` can also be applied directly to mapped classes, to allow for attributes that dynamically configure themselves on subclasses when using mapped inheritance schemes. Below illustrates :class:`_orm.declared_attr` to create a dynamic scheme for generating the :paramref:`_orm.Mapper.polymorphic_identity` parameter for subclasses:: class Employee(Base): __tablename__ = 'employee' id: Mapped[int] = mapped_column(primary_key=True) type: Mapped[str] = mapped_column(String(50)) @declared_attr.directive def __mapper_args__(cls) -> Dict[str, Any]: if cls.__name__ == 'Employee': return { "polymorphic_on":cls.type, "polymorphic_identity":"Employee" } else: return {"polymorphic_identity":cls.__name__} class Engineer(Employee): pass :class:`_orm.declared_attr` supports decorating functions that are explicitly decorated with ``@classmethod``. This is never necessary from a runtime perspective, however may be needed in order to support :pep:`484` typing tools that don't otherwise recognize the decorated function as having class-level behaviors for the ``cls`` parameter:: class SomethingMixin: x: Mapped[int] y: Mapped[int] @declared_attr @classmethod def x_plus_y(cls) -> Mapped[int]: return column_property(cls.x + cls.y) .. versionadded:: 2.0 - :class:`_orm.declared_attr` can accommodate a function decorated with ``@classmethod`` to help with :pep:`484` integration where needed. .. seealso:: :ref:`orm_mixins_toplevel` - Declarative Mixin documentation with background on use patterns for :class:`_orm.declared_attr`. F�_DeclaredAttrDecorated[_T]rS r� c C s d S r� r[ r� r[ r[ r\ r� � s zdeclared_attr.__init__r r` r� c C s d S r� r[ r� r[ r[ r\ r� � r� zdeclared_attr.__set__r� c C s d S r� r[ r� r[ r[ r\ r� � r� zdeclared_attr.__delete__zInstrumentedAttribute[_T]r� c C s d S r� r[ r� r[ r[ r\ r� � s zdeclared_attr.__get__�objectrO c C s d S r� r[ r� r[ r[ r\ r� � s r� z$Union[InstrumentedAttribute[_T], _T]c C s d S r� r[ r� r[ r[ r\ r� � s �_stateful_declared_attr[_T]�rz rV c K s t f |�S r� )�_stateful_declared_attr�rU rz r[ r[ r\ � _stateful� s zdeclared_attr._statefulz_declared_directive[Any]r� c C s t S r� )r� �rU r[ r[ r\ � directive� s zdeclared_attr.directivec C s | j dd�S )NT)r� )r� r� r[ r[ r\ r� � s zdeclared_attr.cascadingN)F)rj rk rl ro r? r r� r� r� r r� r= r� r> r� r� r[ r[ r[ r\ r� I s" [ �r� c @ sH e Zd ZU ded< dd�dd�Zeddd�d d ��Zd d d �dd�ZdS )r� zDict[str, Any]rz r �rz c K s || _ d S r� r� �r� rz r[ r[ r\ r� � s z _stateful_declared_attr.__init__r� r� c K s | j �� }|�|� tf |�S r� )rz �copy�updater� )r� rz Znew_kwr[ r[ r\ r� � s z!_stateful_declared_attr._statefulr� zdeclared_attr[_T]r� c C s t |f| j�S r� )r� rz r� r[ r[ r\ r� � s z _stateful_declared_attr.__call__N)rj rk rl r� r� r= r� r� r[ r[ r[ r\ r� � s r� �Type[_T]c C s | S )a, Mark a class as providing the feature of "declarative mixin". E.g.:: from sqlalchemy.orm import declared_attr from sqlalchemy.orm import declarative_mixin @declarative_mixin class MyMixin: @declared_attr def __tablename__(cls): return cls.__name__.lower() __table_args__ = {'mysql_engine': 'InnoDB'} __mapper_args__= {'always_refresh': True} id = Column(Integer, primary_key=True) class MyModel(MyMixin, Base): name = Column(String(1000)) The :func:`_orm.declarative_mixin` decorator currently does not modify the given class in any way; it's current purpose is strictly to assist the :ref:`Mypy plugin <mypy_toplevel>` in being able to identify SQLAlchemy declarative mixin classes when no other context is present. .. versionadded:: 1.4.6 .. seealso:: :ref:`orm_mixins_toplevel` :ref:`mypy_declarative_mixins` - in the :ref:`Mypy plugin documentation <mypy_toplevel>` r[ r� r[ r[ r\ �declarative_mixin� s 'r� � Type[Any]r` c C s� d| j kr| j d }nd }d| j kr0| j d }nd }| j �dd �}|d k rtt|t�s`t�d��q�|d k r�t�d��nt||d�}|| _|| _d| j kr�| jj| _t| dt j �t j kr�| jj | _ d S )Nrt �type_annotation_maprv zfDeclarative base class has a 'registry' attribute that is not an instance of sqlalchemy.orm.registry()z�Declarative base class has both a 'registry' attribute and a type_annotation_map entry. Per-base type_annotation_maps are not supported. Please apply the type_annotation_map to this registry directly.)rt r� r� ) re r} r~ rv r5 r r{ rt rY r� r� � constructor)rU rt r� r� r[ r[ r\ �_setup_declarative_base s4 �� � r� c sZ e Zd ZdZejejejejejejejejfddddddddddd� � fdd� Z� ZS ) �MappedAsDataclassa Mixin class to indicate when mapping this class, also convert it to be a dataclass. .. seealso:: :ref:`orm_declarative_native_dataclasses` - complete background on SQLAlchemy native dataclass mapping .. versionadded:: 2.0 �Union[_NoArg, bool]�'Union[_NoArg, Callable[..., Type[Any]]]r r` ) �init�repr�eq�order� unsafe_hash� match_args�kw_only�dataclass_callablerz rV c s� ||||||||d�} t | d�rN| j� t�� � � fdd�| �� D � | _} n | | _} t� jf | � t| �s�t�| �} t� | | | � d S )N�r� r� r� r� r� r� r� r� �_sa_apply_dc_transformsc s. i | ]&\}}||t jkr&� �|t j�n|�qS r[ )r9 �NO_ARGr} )�.0�k�v��currentr[ r\ � <dictcomp>` s � z7MappedAsDataclass.__init_subclass__.<locals>.<dictcomp>) �hasattrr� r( Z_assert_dc_arguments�items�super�__init_subclass__r" Z(_update_annotations_for_non_mapped_classZ _apply_dataclasses_to_any_class) rU r� r� r� r� r� r� r� r� rz Zapply_dc_transformsZcurrent_transformsZnew_anno�� __class__r� r\ r� @ s8 � ���� �z#MappedAsDataclass.__init_subclass__)rj rk rl ro r9 r� r� � __classcell__r[ r[ r� r\ r� 3 s �r� )� metaclassc s� e Zd ZU dZejr�dd�dd�Zdd�dd�Zd ed < d ed < d ed < ded< ded< ded< ded< ded< ded< dd�dd�Z ddd �� fd d � Z � Z S ) �DeclarativeBasea� Base class used for declarative class definitions. The :class:`_orm.DeclarativeBase` allows for the creation of new declarative bases in such a way that is compatible with type checkers:: from sqlalchemy.orm import DeclarativeBase class Base(DeclarativeBase): pass The above ``Base`` class is now usable as the base for new declarative mappings. The superclass makes use of the ``__init_subclass__()`` method to set up new classes and metaclasses aren't used. When first used, the :class:`_orm.DeclarativeBase` class instantiates a new :class:`_orm.registry` to be used with the base, assuming one was not provided explicitly. The :class:`_orm.DeclarativeBase` class supports class-level attributes which act as parameters for the construction of this registry; such as to indicate a specific :class:`_schema.MetaData` collection as well as a specific value for :paramref:`_orm.registry.type_annotation_map`:: from typing_extensions import Annotated from sqlalchemy import BigInteger from sqlalchemy import MetaData from sqlalchemy import String from sqlalchemy.orm import DeclarativeBase bigint = Annotated[int, "bigint"] my_metadata = MetaData() class Base(DeclarativeBase): metadata = my_metadata type_annotation_map = { str: String().with_variant(String(255), "mysql", "mariadb"), bigint: BigInteger() } Class-level attributes which may be specified include: :param metadata: optional :class:`_schema.MetaData` collection. If a :class:`_orm.registry` is constructed automatically, this :class:`_schema.MetaData` collection will be used to construct it. Otherwise, the local :class:`_schema.MetaData` collection will supercede that used by an existing :class:`_orm.registry` passed using the :paramref:`_orm.DeclarativeBase.registry` parameter. :param type_annotation_map: optional type annotation map that will be passed to the :class:`_orm.registry` as :paramref:`_orm.registry.type_annotation_map`. :param registry: supply a pre-existing :class:`_orm.registry` directly. .. versionadded:: 2.0 Added :class:`.DeclarativeBase`, so that declarative base classes may be constructed in such a way that is also recognized by :pep:`484` type checkers. As a result, :class:`.DeclarativeBase` and other subclassing-oriented APIs should be seen as superseding previous "class returned by a function" APIs, namely :func:`_orm.declarative_base` and :meth:`_orm.registry.generate_base`, where the base class returned cannot be recognized by type checkers without using plugins. **__init__ behavior** In a plain Python class, the base-most ``__init__()`` method in the class hierarchy is ``object.__init__()``, which accepts no arguments. However, when the :class:`_orm.DeclarativeBase` subclass is first declared, the class is given an ``__init__()`` method that links to the :paramref:`_orm.registry.constructor` constructor function, if no ``__init__()`` method is already present; this is the usual declarative constructor that will assign keyword arguments as attributes on the instance, assuming those attributes are established at the class level (i.e. are mapped, or are linked to a descriptor). This constructor is **never accessed by a mapped class without being called explicitly via super()**, as mapped classes are themselves given an ``__init__()`` method directly which calls :paramref:`_orm.registry.constructor`, so in the default case works independently of what the base-most ``__init__()`` method does. .. versionchanged:: 2.0.1 :class:`_orm.DeclarativeBase` has a default constructor that links to :paramref:`_orm.registry.constructor` by default, so that calls to ``super().__init__()`` can access this constructor. Previously, due to an implementation mistake, this default constructor was missing, and calling ``super().__init__()`` would invoke ``object.__init__()``. The :class:`_orm.DeclarativeBase` subclass may also declare an explicit ``__init__()`` method which will replace the use of the :paramref:`_orm.registry.constructor` function at this level:: class Base(DeclarativeBase): def __init__(self, id=None): self.id = id Mapped classes still will not invoke this constructor implicitly; it remains only accessible by calling ``super().__init__()``:: class MyClass(Base): def __init__(self, id=None, name=None): self.name = name super().__init__(id=id) Note that this is a different behavior from what functions like the legacy :func:`_orm.declarative_base` would do; the base created by those functions would always install :paramref:`_orm.registry.constructor` for ``__init__()``. � Mapper[Self]r� c C s d S r� r[ r� r[ r[ r\ �_sa_inspect_type� r� z DeclarativeBase._sa_inspect_type�InstanceState[Self]c C s d S r� r[ r� r[ r[ r\ �_sa_inspect_instance� r� z$DeclarativeBase._sa_inspect_instance�ClassVar[_RegistryType]r{ rv �ClassVar[MetaData]rt z ClassVar[str]rj �ClassVar[Mapper[Any]]rd zClassVar[FromClause]rW r � __tablename__�__mapper_args__�__table_args__r� c K s d S r� r[ r� r[ r[ r\ r� = r� zDeclarativeBase.__init__r` r� c s@ t | jkr t| t � t| � nt| j| | j� t� jf |� d S r� ) r� � __bases__�_check_not_declarativer� r' r{ re r� r� r� r� r[ r\ r� ? s z!DeclarativeBase.__init_subclass__) rj rk rl ro r? r r� r� r� r� r� r� r[ r[ r� r\ r� t s o r� )rU �baserV c C sR | j }d|kr(t|d �s(t|d d�r:t|�dd �t�rNt�d|j�d���d S )NrW r� r� z Cannot use zR directly as a declarative base class. Create a Base by creating a subclass of it.) re �callabler� r~ r} r_ r5 r rj )rU r� �cls_dictr[ r[ r\ r� H s � � �� �r� c s� e Zd ZU dZded< ded< ded< ded< d ed < ejr~d d �d d�Zdd �dd�Zded< ded< ded< dd�dd�Z ddd�� fdd � Z � Z S ) �DeclarativeBaseNoMetaz�Same as :class:`_orm.DeclarativeBase`, but does not use a metaclass to intercept new attributes. The :class:`_orm.DeclarativeBaseNoMeta` base may be used when use of custom metaclasses is desirable. .. versionadded:: 2.0 r� r{ rv r� rt r� rd �Optional[FromClause]rW r� r� c C s d S r� r[ r� r[ r[ r\ r� � r� z&DeclarativeBaseNoMeta._sa_inspect_typer� c C s d S r� r[ r� r[ r[ r\ r� � r� z*DeclarativeBaseNoMeta._sa_inspect_instancer r� r� r� r� c K s d S r� r[ r� r[ r[ r\ r� � r� z DeclarativeBaseNoMeta.__init__r` r� c s@ t | jkr t| t � t| � nt| j| | j� t� jf |� d S r� ) r� r� r� r� r' r{ re r� r� r� r� r[ r\ r� � s z'DeclarativeBaseNoMeta.__init_subclass__) rj rk rl ro r� r? r r� r� r� r� r� r[ r[ r� r\ r� W s r� zMapperProperty[Any])�targetra �attrrV c C s t | ||� dS )a' Add a new mapped attribute to an ORM mapped class. E.g.:: add_mapped_attribute(User, "addresses", relationship(Address)) This may be used for ORM mappings that aren't using a declarative metaclass that intercepts attribute set operations. .. versionadded:: 2.0 Nr% )r� ra r� r[ r[ r\ �add_mapped_attribute� s r� �Base�rt �mapperrU r� �class_registryr� r� r� �Optional[MetaData]�$Optional[Callable[..., Mapper[Any]]]�&Optional[clsregistry._ClsRegistryType]� Optional[_TypeAnnotationMapType]�Callable[(Ellipsis, None)]r ) rt r� rU r� r� r� r� r� rV c C s t | |||d�j||||d�S )a Construct a base class for declarative class definitions. The new base class will be given a metaclass that produces appropriate :class:`~sqlalchemy.schema.Table` objects and makes the appropriate :class:`_orm.Mapper` calls based on the information provided declaratively in the class and any subclasses of the class. .. versionchanged:: 2.0 Note that the :func:`_orm.declarative_base` function is superseded by the new :class:`_orm.DeclarativeBase` class, which generates a new "base" class using subclassing, rather than return value of a function. This allows an approach that is compatible with :pep:`484` typing tools. The :func:`_orm.declarative_base` function is a shorthand version of using the :meth:`_orm.registry.generate_base` method. That is, the following:: from sqlalchemy.orm import declarative_base Base = declarative_base() Is equivalent to:: from sqlalchemy.orm import registry mapper_registry = registry() Base = mapper_registry.generate_base() See the docstring for :class:`_orm.registry` and :meth:`_orm.registry.generate_base` for more details. .. versionchanged:: 1.4 The :func:`_orm.declarative_base` function is now a specialization of the more generic :class:`_orm.registry` class. The function also moves to the ``sqlalchemy.orm`` package from the ``declarative.ext`` package. :param metadata: An optional :class:`~sqlalchemy.schema.MetaData` instance. All :class:`~sqlalchemy.schema.Table` objects implicitly declared by subclasses of the base will share this MetaData. A MetaData instance will be created if none is provided. The :class:`~sqlalchemy.schema.MetaData` instance will be available via the ``metadata`` attribute of the generated declarative base class. :param mapper: An optional callable, defaults to :class:`_orm.Mapper`. Will be used to map subclasses to their Tables. :param cls: Defaults to :class:`object`. A type to use as the base for the generated declarative base class. May be a class or tuple of classes. :param name: Defaults to ``Base``. The display name for the generated class. Customizing this is not required, but can improve clarity in tracebacks and debugging. :param constructor: Specify the implementation for the ``__init__`` function on a mapped class that has no ``__init__`` of its own. Defaults to an implementation that assigns \**kwargs for declared fields and relationships to an instance. If ``None`` is supplied, no __init__ will be provided and construction will fall back to cls.__init__ by way of the normal Python semantics. :param class_registry: optional dictionary that will serve as the registry of class names-> mapped classes when string names are used to identify classes inside of :func:`_orm.relationship` and others. Allows two or more declarative base classes to share the same registry of class names for simplified inter-base relationships. :param type_annotation_map: optional dictionary of Python types to SQLAlchemy :class:`_types.TypeEngine` classes or instances. This is used exclusively by the :class:`_orm.MappedColumn` construct to produce column types based on annotations within the :class:`_orm.Mapped` type. .. versionadded:: 2.0 .. seealso:: :ref:`orm_declarative_mapped_column_type_map` :param metaclass: Defaults to :class:`.DeclarativeMeta`. A metaclass or __metaclass__ compatible callable to use as the meta type of the generated declarative base class. .. seealso:: :class:`_orm.registry` )rt r� r� r� )r� rU r� r� )rv � generate_baser� r[ r[ r\ �declarative_base� s n��r c @ s~ e Zd ZU dZded< ded< ded< ded < d ed < d ed < ded< ded< ded< ddded�ddddd�dd�Zdd d �d d �Zd d!d"�d#d$�Ze d%d&�d'd(��Z d)d d*�d+d,�Z d-d d.�d/d0�Z e d1d2d3�d4d5��Ze d1d2d3�d6d7��Zd8d&�d9d:�Zd-d d;�d<d=�Zd>d d?�d@dA�ZdBd dC�dDdE�Zdrdd dG�dHdI�Zdsdd dG�dJdK�ZdBd dC�dLdM�ZdedNefdOdPdQdPdRdS�dTdU�Zeje e e e e e!e"e#e$f dV�e%d>d>dW�dXdY���Z&e%dtdZdZdZdZdZdZdZdZd[�d\d]d]d]d]d]d]d]d^d_d`� dadY��Z&due'j(e'j(e'j(e'j(e'j(e'j(e'j(e'j(d[�dbd]d]d]d]d]d]d]d^dcd`� dddY�Z&d>d>d?�dedf�Z)dRdgdh�didj�Z*d>dkd?�dldm�Z+dvd>dndRdkdo�dpdq�Z,dS )wrv a� Generalized registry for mapping classes. The :class:`_orm.registry` serves as the basis for maintaining a collection of mappings, and provides configurational hooks used to map classes. The three general kinds of mappings supported are Declarative Base, Declarative Decorator, and Imperative Mapping. All of these mapping styles may be used interchangeably: * :meth:`_orm.registry.generate_base` returns a new declarative base class, and is the underlying implementation of the :func:`_orm.declarative_base` function. * :meth:`_orm.registry.mapped` provides a class decorator that will apply declarative mapping to a class without the use of a declarative base class. * :meth:`_orm.registry.map_imperatively` will produce a :class:`_orm.Mapper` for a class without scanning the class for declarative class attributes. This method suits the use case historically provided by the ``sqlalchemy.orm.mapper()`` classical mapping function, which is removed as of SQLAlchemy 2.0. .. versionadded:: 1.4 .. seealso:: :ref:`orm_mapping_classes_toplevel` - overview of class mapping styles. z clsregistry._ClsRegistryType�_class_registryz;weakref.WeakKeyDictionary[ClassManager[Any], Literal[True]]� _managersz5weakref.WeakKeyDictionary[Mapper[Any], Literal[True]]�_non_primary_mappersr; rt z-CallableReference[Callable[(Ellipsis, None)]]r� � _MutableTypeAnnotationMapTyper� zSet[_RegistryType]� _dependents� _dependenciesrS � _new_mappersN)rt r� r� r� r� r� r� r� c C s� |pt � }|dkrt�� }|| _t�� | _t�� | _|| _|| _i | _ |dk rX| � |� t � | _ t � | _ d| _tj� dtj| < W 5 Q R X dS )a� Construct a new :class:`_orm.registry` :param metadata: An optional :class:`_schema.MetaData` instance. All :class:`_schema.Table` objects generated using declarative table mapping will make use of this :class:`_schema.MetaData` collection. If this argument is left at its default of ``None``, a blank :class:`_schema.MetaData` collection is created. :param constructor: Specify the implementation for the ``__init__`` function on a mapped class that has no ``__init__`` of its own. Defaults to an implementation that assigns \**kwargs for declared fields and relationships to an instance. If ``None`` is supplied, no __init__ will be provided and construction will fall back to cls.__init__ by way of the normal Python semantics. :param class_registry: optional dictionary that will serve as the registry of class names-> mapped classes when string names are used to identify classes inside of :func:`_orm.relationship` and others. Allows two or more declarative base classes to share the same registry of class names for simplified inter-base relationships. :param type_annotation_map: optional dictionary of Python types to SQLAlchemy :class:`_types.TypeEngine` classes or instances. The provided dict will update the default type mapping. This is used exclusively by the :class:`_orm.MappedColumn` construct to produce column types based on annotations within the :class:`_orm.Mapped` type. .. versionadded:: 2.0 .. seealso:: :ref:`orm_declarative_mapped_column_type_map` NFT)r; �weakref�WeakValueDictionaryr �WeakKeyDictionaryr r rt r� r� �update_type_annotation_map�setr r r r Z_CONFIGURE_MUTEXZ_mapper_registries)r� rt r� r� r� Z lcl_metadatar[ r[ r\ r� { s / zregistry.__init__�_TypeAnnotationMapTyper` )r� rV c C s | j �dd� |�� D �� dS )zQupdate the :paramref:`_orm.registry.type_annotation_map` with new values.c S s, i | ]$\}}t j|d d d�D ] }||�q qS )T)Z include_unionZ discard_none)� compat_typingZ expand_unions)r� �typZsqltypeZsub_typer[ r[ r\ r� � s � � z7registry.update_type_annotation_map.<locals>.<dictcomp>N)r� r� r� )r� r� r[ r[ r\ r � s ��z#registry.update_type_annotation_maprN z"Optional[sqltypes.TypeEngine[Any]])� python_typerV c s �}t |�r|j}q�|k� t|�rZt|�rHtd|�}�|ft|ff}q�|j}�|ff}nPt|�rvt|�}�|ff}n4t |t �r�|}� �fdd�|j D �}n|}�|ff}|D ]V\}}| j � |�}|d kr�t�|�}|d k r�t�|�}|�|||�} | d k r�| S q�d S )Nr� c 3 s | ]}� r|n�|fV qd S r� r[ )r� �pt�Z check_is_ptr r[ r\ � <genexpr>� s �z)registry._resolve_type.<locals>.<genexpr>)rE � __value__rB rC r rF � __origin__rD rA r~ rf rX r� r} r8 Z _type_map_getZ to_instanceZ_resolve_for_python_type) r� r Zpython_type_to_checkZpython_type_type�searchr Z flattenedZsql_typeZ sql_type_instZresolved_sql_typer[ r r\ � _resolve_type� sH � � � zregistry._resolve_typezFrozenSet[Mapper[Any]]r� c C s t dd� | jD ���| j�S )z9read only collection of all :class:`_orm.Mapper` objects.c s s | ] }|j V qd S r� )r� �r� r� r[ r[ r\ r s z#registry.mappers.<locals>.<genexpr>)� frozensetr �unionr r� r[ r[ r\ �mappers s �zregistry.mappersru )rv rV c C s( || kr d S |j �| � | j�|� d S r� )r �addr )r� rv r[ r[ r\ �_set_depends_on s zregistry._set_depends_onz Mapper[Any])r� rV c C s, d|_ | jrd S | �| h�D ] }d|_q d S �NT)�_ready_for_configurer �_recurse_with_dependents)r� r� r� r[ r[ r\ �_flag_new_mapper s zregistry._flag_new_mapperzSet[RegistryType]zIterator[RegistryType])� registriesrV c c sP |}t � }|rL|�� }|�|� |�|j�|�� |V |�|j�|�� q d S r� )r �popr r� r � difference�rU r" �todo�doner� r[ r[ r\ r # s z!registry._recurse_with_dependentsc c sP |}t � }|rL|�� }|�|� |�|j�|�� |V |�|j�|�� q d S r� )r r# r r� r r$ r% r[ r[ r\ �_recurse_with_dependencies6 s z#registry._recurse_with_dependencieszIterator[Mapper[Any]]c C s, t �dd� t| j�D �dd� t| j�D ��S )Nc s s* | ]"}|j r|jjs|jjr|jV qd S r� )r� r� � configuredr r r[ r[ r\ r L s �z1registry._mappers_to_configure.<locals>.<genexpr>c s s | ]}|j s|jr|V qd S r� )r) r )r� Znpmr[ r[ r\ r S s �)� itertools�chain�listr r r� r[ r[ r\ �_mappers_to_configureJ s ���z registry._mappers_to_configure)� np_mapperrV c C s d| j |< d S r )r )r� r. r[ r[ r\ �_add_non_primary_mapperZ s z registry._add_non_primary_mapperrR rT c C s t �|j|| j� d S r� )r Z remove_classrj r �r� rU r[ r[ r\ � _dispose_cls] s zregistry._dispose_clszClassManager[Any])r� rV c C s8 d| j |< |jr t�d|j ��|jd ks.t�| |_d S )NTz1Class '%s' already has a primary mapper defined. )r r� r5 � ArgumentErrorrZ rv r� )r� r� r[ r[ r\ � _add_manager` s ��zregistry._add_managerF)�cascaderV c C s t j| h|d� dS )a] Configure all as-yet unconfigured mappers in this :class:`_orm.registry`. The configure step is used to reconcile and initialize the :func:`_orm.relationship` linkages between mapped classes, as well as to invoke configuration events such as the :meth:`_orm.MapperEvents.before_configured` and :meth:`_orm.MapperEvents.after_configured`, which may be used by ORM extensions or user-defined extension hooks. If one or more mappers in this registry contain :func:`_orm.relationship` constructs that refer to mapped classes in other registries, this registry is said to be *dependent* on those registries. In order to configure those dependent registries automatically, the :paramref:`_orm.registry.configure.cascade` flag should be set to ``True``. Otherwise, if they are not configured, an exception will be raised. The rationale behind this behavior is to allow an application to programmatically invoke configuration of registries while controlling whether or not the process implicitly reaches other registries. As an alternative to invoking :meth:`_orm.registry.configure`, the ORM function :func:`_orm.configure_mappers` function may be used to ensure configuration is complete for all :class:`_orm.registry` objects in memory. This is generally simpler to use and also predates the usage of :class:`_orm.registry` objects overall. However, this function will impact all mappings throughout the running Python process and may be more memory/time consuming for an application that has many registries in use for different purposes that may not be needed immediately. .. seealso:: :func:`_orm.configure_mappers` .. versionadded:: 1.4.0b2 �r4 N)r Z_configure_registries�r� r4 r[ r[ r\ � configurej s 'zregistry.configurec C s t j| h|d� dS )a� Dispose of all mappers in this :class:`_orm.registry`. After invocation, all the classes that were mapped within this registry will no longer have class instrumentation associated with them. This method is the per-:class:`_orm.registry` analogue to the application-wide :func:`_orm.clear_mappers` function. If this registry contains mappers that are dependencies of other registries, typically via :func:`_orm.relationship` links, then those registries must be disposed as well. When such registries exist in relation to this one, their :meth:`_orm.registry.dispose` method will also be called, if the :paramref:`_orm.registry.dispose.cascade` flag is set to ``True``; otherwise, an error is raised if those registries were not already disposed. .. versionadded:: 1.4.0b2 .. seealso:: :func:`_orm.clear_mappers` r5 N)r Z_dispose_registriesr6 r[ r[ r\ �dispose� s zregistry.disposec C s8 d|j kr|j}|�� |j}| �|� tj�|� d S )Nr� )re r� Z_set_dispose_flagsrZ r1 r Z_instrumentation_factory� unregister)r� r� r� rZ r[ r[ r\ �_dispose_manager_and_mapper� s z$registry._dispose_manager_and_mapperr� r� r� r_ r )r� rU r� r� rV c C s� | j }t|t� r|fp|}t| |d�}t|t�r<|j|d< | jdk rP| j|d< d|d< |rd||d< t|d�r�d d d d �d d �}||d< ||||�S )a� Generate a declarative base class. Classes that inherit from the returned class object will be automatically mapped using declarative mapping. E.g.:: from sqlalchemy.orm import registry mapper_registry = registry() Base = mapper_registry.generate_base() class MyClass(Base): __tablename__ = "my_table" id = Column(Integer, primary_key=True) The above dynamically generated class is equivalent to the non-dynamic example below:: from sqlalchemy.orm import registry from sqlalchemy.orm.decl_api import DeclarativeMeta mapper_registry = registry() class Base(metaclass=DeclarativeMeta): __abstract__ = True registry = mapper_registry metadata = mapper_registry.metadata __init__ = mapper_registry.constructor .. versionchanged:: 2.0 Note that the :meth:`_orm.registry.generate_base` method is superseded by the new :class:`_orm.DeclarativeBase` class, which generates a new "base" class using subclassing, rather than return value of a function. This allows an approach that is compatible with :pep:`484` typing tools. The :meth:`_orm.registry.generate_base` method provides the implementation for the :func:`_orm.declarative_base` function, which creates the :class:`_orm.registry` and base class all at once. See the section :ref:`orm_declarative_mapping` for background and examples. :param mapper: An optional callable, defaults to :class:`_orm.Mapper`. This function is used to generate new :class:`_orm.Mapper` objects. :param cls: Defaults to :class:`object`. A type to use as the base for the generated declarative base class. May be a class or tuple of classes. :param name: Defaults to ``Base``. The display name for the generated class. Customizing this is not required, but can improve clarity in tracebacks and debugging. :param metaclass: Defaults to :class:`.DeclarativeMeta`. A metaclass or __metaclass__ compatible callable to use as the meta type of the generated declarative base class. .. seealso:: :ref:`orm_declarative_mapping` :func:`_orm.declarative_base` )rv rt ro Nr� Tr| Z__mapper_cls__�__class_getitem__r� r )rU ra rV c S s | S r� r[ ri r[ r[ r\ r; s z1registry.generate_base.<locals>.__class_getitem__)rt r~ �tuple�dictrf ro r� r� ) r� r� rU r� r� rt rx � class_dictr; r[ r[ r\ r� � s N zregistry.generate_baserp )�_registry__clsrV c C s d S r� r[ )r� r? r[ r[ r\ �mapped_as_dataclass s z registry.mapped_as_dataclass.r� z Literal[None]r� r� z Callable[[Type[_O]], Type[_O]]) r? r� r� r� r� r� r� r� r� rV c C s d S r� r[ ) r� r? r� r� r� r� r� r� r� r� r[ r[ r\ r@ . s zOptional[Type[_O]]z/Union[Type[_O], Callable[[Type[_O]], Type[_O]]]c s8 ddd�� ��������f dd� } |r0| |�S | S dS )a_ Class decorator that will apply the Declarative mapping process to a given class, and additionally convert the class to be a Python dataclass. .. seealso:: :ref:`orm_declarative_native_dataclasses` - complete background on SQLAlchemy native dataclass mapping .. versionadded:: 2.0 rR rT c s0 t | d�������� d�� t�| | j� | S )Nr� r� )�setattrr' re r� � r� r� r� r� r� r� r� r� r� r[ r\ r� [ s ��z.registry.mapped_as_dataclass.<locals>.decorateNr[ ) r� r? r� r� r� r� r� r� r� r� r� r[ rB r\ r@ = s $c C s t | ||j� |S )a� Class decorator that will apply the Declarative mapping process to a given class. E.g.:: from sqlalchemy.orm import registry mapper_registry = registry() @mapper_registry.mapped class Foo: __tablename__ = 'some_table' id = Column(Integer, primary_key=True) name = Column(String) See the section :ref:`orm_declarative_mapping` for complete details and examples. :param cls: class to be mapped. :return: the class that was passed. .. seealso:: :ref:`orm_declarative_mapping` :meth:`_orm.registry.generate_base` - generates a base class that will apply Declarative mapping to subclasses automatically using a Python metaclass. .. seealso:: :meth:`_orm.registry.mapped_as_dataclass` )r' re r0 r[ r[ r\ �mappedr s %zregistry.mapped� Callable[[Type[_T]], Type[_T]]r� c s ddd�� �fdd� }|S )a� Class decorator which will invoke :meth:`_orm.registry.generate_base` for a given base class. E.g.:: from sqlalchemy.orm import registry mapper_registry = registry() @mapper_registry.as_declarative_base() class Base: @declared_attr def __tablename__(cls): return cls.__name__.lower() id = Column(Integer, primary_key=True) class MyMappedClass(Base): # ... All keyword arguments passed to :meth:`_orm.registry.as_declarative_base` are passed along to :meth:`_orm.registry.generate_base`. r� rT c s | � d< | j � d< �jf � �S )NrU r� )rj r� r� �rz r� r[ r\ r� � s z.registry.as_declarative_base.<locals>.decorater[ )r� rz r� r[ rE r\ �as_declarative_base� s z registry.as_declarative_basez Mapper[_O]c C s t | ||j� |jS )a� Map a class declaratively. In this form of mapping, the class is scanned for mapping information, including for columns to be associated with a table, and/or an actual table object. Returns the :class:`_orm.Mapper` object. E.g.:: from sqlalchemy.orm import registry mapper_registry = registry() class Foo: __tablename__ = 'some_table' id = Column(Integer, primary_key=True) name = Column(String) mapper = mapper_registry.map_declaratively(Foo) This function is more conveniently invoked indirectly via either the :meth:`_orm.registry.mapped` class decorator or by subclassing a declarative metaclass generated from :meth:`_orm.registry.generate_base`. See the section :ref:`orm_declarative_mapping` for complete details and examples. :param cls: class to be mapped. :return: a :class:`_orm.Mapper` object. .. seealso:: :ref:`orm_declarative_mapping` :meth:`_orm.registry.mapped` - more common decorator interface to this function. :meth:`_orm.registry.map_imperatively` )r' re rd r0 r[ r[ r\ �map_declaratively� s -zregistry.map_declarativelyr� )rZ � local_tablerz rV c K s t | |||�S )a� Map a class imperatively. In this form of mapping, the class is not scanned for any mapping information. Instead, all mapping constructs are passed as arguments. This method is intended to be fully equivalent to the now-removed SQLAlchemy ``mapper()`` function, except that it's in terms of a particular registry. E.g.:: from sqlalchemy.orm import registry mapper_registry = registry() my_table = Table( "my_table", mapper_registry.metadata, Column('id', Integer, primary_key=True) ) class MyClass: pass mapper_registry.map_imperatively(MyClass, my_table) See the section :ref:`orm_imperative_mapping` for complete background and usage examples. :param class\_: The class to be mapped. Corresponds to the :paramref:`_orm.Mapper.class_` parameter. :param local_table: the :class:`_schema.Table` or other :class:`_sql.FromClause` object that is the subject of the mapping. Corresponds to the :paramref:`_orm.Mapper.local_table` parameter. :param \**kw: all other keyword arguments are passed to the :class:`_orm.Mapper` constructor directly. .. seealso:: :ref:`orm_imperative_mapping` :ref:`orm_declarative_mapping` r, )r� rZ rH rz r[ r[ r\ �map_imperatively� s 6zregistry.map_imperatively)F)F).)N)N)-rj rk rl ro r� r) r� r r �propertyr r r! r� r r( r- r/ r1 r3 r7 r8 r: r� rs r� r �dataclass_transformr1 r2 r. r/ r r r r r r r@ r9 r� rC rF rG rI r[ r[ r[ r\ rv P s� �D= ) �g�� ��& ��$5(#3 �rv rD r� c K s. | � dd�| � dd� }}t||d�jf | �S )a� Class decorator which will adapt a given class into a :func:`_orm.declarative_base`. This function makes use of the :meth:`_orm.registry.as_declarative_base` method, by first creating a :class:`_orm.registry` automatically and then invoking the decorator. E.g.:: from sqlalchemy.orm import as_declarative @as_declarative() class Base: @declared_attr def __tablename__(cls): return cls.__name__.lower() id = Column(Integer, primary_key=True) class MyMappedClass(Base): # ... .. seealso:: :meth:`_orm.registry.as_declarative_base` rt Nr� )rt r� )r# rv rF )rz rt r� r[ r[ r\ �as_declarative- s � ��rL zOptional[Mapper[Any]]c C s( t | �}|d kr$t�| �r$t�| � |S r� )r! r* Zhas_clsZraise_unmapped_for_cls)rU �mpr[ r[ r\ �_inspect_decl_metaS s rN )F)�ro � __future__r r* r� r? r r r r r r r r r r r r r r r r r r r � r r r r r Z_orm_constructorsr r r r r r r� r! r" r# r$ Z decl_baser& r' r( r) r* r+ r- Zdescriptor_propsr. r/ r� r� r0 � propertiesr1 Z relationshipsr2 �stater3 r5 r6 r7 �sqlr8 Zsql.baser9 Z sql.elementsr: Z sql.schemar; Zsql.selectabler< r= r> r Z util.typingr@ rA rB rC rD rE rF rG Z_typingrH rI rJ rK rL Z sql._typingrM Z sql.type_apirN rO rQ r r Z_DeclaredAttrDecoratedr] rf r^ Z Inspectablerm rK rr rs r� r� r� Z_MappedAttributer� r� r� r� r� r� r� r� r� r� r rv ru rL Z _inspectsrN r[ r[ r[ r\ �<module> s0 � � �� �*A *)A � U �k�"{ [& �