o Rc@sdZddlmZddlZddlZddlmZddlmZddlm Z ddlm Z dd lm Z dd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddl mZ ddlmZddlmZddlmZddlm Z ddlmZddlmZddl m!Z!ddl m"Z"ddlm#Z#ddZ$ddZ%d@d d!Z&d"d#Z'd$d%Z(d&d'Z)e*d(d)d*Z+d+d,Z,Gd-d.d.e-Z.Gd/d0d0e.Z/Gd1d2d2e.Z0e*d(d3d4Z1Gd5d6d6e0Z2d7d8Z3d9d:Z4d;d<Z5d=e5_6d>d?Z7dS)Az(Internal implementation for declarative.)absolute_importN) attributes)instrumentation) clsregistry)exc)mapper)InstrumentedAttribute)QueryableAttribute)_is_mapped_class)InspectionAttr)CompositeProperty)SynonymProperty)MapperProperty)Mapper)ColumnProperty) class_mapper)event)util) expression)Column)Table) topologicalcCs,t|r t|St|rt|ddSdS)NF) configure)_DeferredMapperConfighas_clsconfig_for_clsr rclsr :D:\Flask\env\Lib\site-packages\sqlalchemy/orm/decl_base.py_declared_mapping_info%s   r"cCsX|turdS|jddr!|jD]}t|}|dur|SqdSt|}|r*|jS|S)N __abstract__F)object__dict__get __bases__"_resolve_for_abstract_or_classical_dive_for_cls_managerclass_)rsupZ clsmanagerr r r!r(0s r(FcCs|dksJt|ts dS||jvrt||S|jddD]"}t|}||jvr@||us9|r5||jvr@n  |s@t||SqdS)areturn an attribute of the class that is either present directly on the class, e.g. not on a superclass, or is from a superclass but this superclass is a non-mapped mixin, that is, not a descendant of the declarative base and is also not classically mapped. This is used to detect attributes that indicate something about a mapped class independently from any mapped classes that it may inherit from. r#Nr) issubclassr$r%getattr__mro__r)r')rattrnamestrictbaseZ_is_classicial_inheritsr r r!_get_immediate_cls_attrDs*     r2cCs4t|dsdS|jD] }t|}|r|Sq dS)Nr.)hasattrr.rmanager_of_class)rr1managerr r r!r)is   r)cCst|||diSN) _MapperConfig setup_mapping)registryrdict_r r r!_as_declarativeysr;cCst|||||jSr6)_ImperativeMapperConfig __mapper__)r9rtable mapper_kwr r r!_mappersr@zsqlalchemy.orm.decl_apicCstjjj}t||tjfSr6)r preloaded orm_decl_api declared_attr isinstanceZ classproperty)objrCr r r!_is_declarative_propss rFcCs.t|rt|ddrtd||fdSdS)N _cascadingFz~@declared_attr.cascading is not supported on the %s attribute on class %s. This attribute invokes for subclasses in any case.T)rFr-rwarn)rEnamerr r r!_check_declared_props_nocascades rJc@s4eZdZdZeddZddZddZdd Zd S) r7)r classname propertiesdeclared_attr_reg __weakref__c Csnt|}|r|j|urtd||jddrdSt|dddp't|d}|r-t }nt }||||||S)Nz4Class %r already has been instrumented declarativelyr#F_sa_decl_prepare_nocascadeTr0Z_sa_decl_prepare) rr4r*rInvalidRequestErrorr%r&r2r3r_ClassScanMapperConfig) rr9cls_r:r>r?r5Z defer_mapZcfg_clsr r r!r8s z_MapperConfig.setup_mappingcCszt|td|_|j|_t|_i|_| dds(t j |jd|||j ddSt |j}|r3|js;td|jdS)NrS non_primaryF)finalizer9Zdeclarative_scanZ init_methodztClass %s has no primary mapper configured. Configure a primary mapper first before setting up a non primary Mapper.)rZassert_arg_typetyper__name__rK OrderedDictrLrMr&rZregister_class constructorrr4Z is_mappedrrQ)selfr9rSr?r5r r r!__init__s(    z_MapperConfig.__init__cCst|j}||||Sr6)rr4rZinstall_member)rZr/valuer5r r r!set_cls_attributes  z_MapperConfig.set_cls_attributecCs||dSr6)maprZr?r r r!_early_mappingz_MapperConfig._early_mappingN) rW __module__ __qualname__ __slots__ classmethodr8r[r]r`r r r r!r7s  r7cs6eZdZdZfddZejfddZddZZ S)r<)r: local_tableinheritscstt||||i|_|d||_tj#|dds(t |j |j |j ||||WddS1s=wYdS)N __table__rTF)superr<r[r:r]rf mapperlib_CONFIGURE_MUTEXr&r add_classrKr_class_registry_setup_inheritancer`)rZr9rSr>r? __class__r r!r[s     "z _ImperativeMapperConfig.__init__cCs"t}|d||j|jfi|S)Nr=)rr]rrfrZr?Z mapper_clsr r r!r^s z_ImperativeMapperConfig.mapcCs|j}|dd}|durFg}|jD]}t|}|durqt|dur/t|ddds/||q|rEt|dkrAt d||f|d}nt |t rN|j }||_ dS)NrgrOTrPr&Class %s has multiple mapped bases: %rr)rr&r'r(r"r2appendlenrrQrDrr*rg)rZr?rrginherits_searchcr r r!rns>      z*_ImperativeMapperConfig._setup_inheritance) rWrbrcrdr[r EMPTY_DICTr^rn __classcell__r r ror!r<s  r<cseZdZdZfddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dddZ ddZddZddZejfddZZS) rR) r:rfpersist_selectabledeclared_columns column_copies table_args tablename mapper_argsmapper_args_fnrgcs|rt|ni|_tt||||d|_t|_i|_| | t j *t |j|j|j||||||||WddS1sUwYdSr6)dictr:rirRr[rysetrzr{_setup_declared_events_scan_attributesrjrkrrlrKrrm_extract_mappable_attributes_extract_declared_columns _setup_tablernr`)rZr9rSr:r>r?ror r!r[,s"     "z_ClassScanMapperConfig.__init__csPtjdrttdfdd}tjdr&ttdfdd}dSdS) N__declare_last__after_configuredcjdSr6)rrr rZr r!rTzG_ClassScanMapperConfig._setup_declared_events..after_configured__declare_first__before_configuredcrr6)rrr rr r!rZrzH_ClassScanMapperConfig._setup_declared_events..before_configured)r2rrZ listens_forr)rZrrr rr!rQs    z-_ClassScanMapperConfig._setup_declared_eventscsptdddurfdd}|SfddtDfddtDtfdd}|S) zProduce a function that checks if a class has overridden an attribute, taking SQLAlchemy-enabled dataclass fields into account. __sa_dataclass_metadata_key__Ncst||uSr6)r-)keyrErr r!attribute_is_overriddenirazR_ClassScanMapperConfig._cls_attr_override_checker..attribute_is_overriddenc$i|]}|jvr|j|jqSr metadatarI.0fsa_dataclass_metadata_keyr r! n  zE_ClassScanMapperConfig._cls_attr_override_checker..crr rrrr r!rsrcst|r|j}|}t|r|j}||urdS|ur dS|}t||}||ur2dS|ur=t|tr=dS||urCdS|urIdSdS)NFT)rFfgetr&r-rDr )rrEretZ all_field)absentall_datacls_fieldsrlocal_datacls_fieldsr r!r{s,    )r2rZdataclass_fieldslocal_dataclass_fieldsr$)rZrrr )rrrrrr!_cls_attr_override_checker^s :  (z1_ClassScanMapperConfig._cls_attr_override_checkercs>tdddurfdd}|Stfdd}|S)zproduce a function to iterate the "attributes" of a class, adjusting for SQLAlchemy fields embedded in dataclass fields. rNc3s(tD] \}}||dfVqdS)NF)varsitems)rIrErr r!local_attributes_for_classszM_ClassScanMapperConfig._cls_attr_resolver..local_attributes_for_classc3sltD]}|jvr|j|jt|jdfVqtD]\}}|vr3||dfVq%dS)NTF)rrraddrI_as_dc_declaredattrrr)fieldrIrEr field_namesrr r!rs   )r2r)rZrrr rr!_cls_attr_resolvers  z)_ClassScanMapperConfig._cls_attr_resolverc s*|j|j}|j}d}d}}d}||j}g}jD]1} | uo/t| duo/t| ddd } || } | sB| urB|| |} ni} | | | | | fq|D]0\} } } } | | | D] \} }}| dkr|t || }|s{| ru|r{fdd}q_| dkrt || }|s| r|rj }q_| dkrt || }|s| r|rj }t|tttdfstd | urd}q_| rt|rtd | j| | fq_| ur\t|trq_t|trtd t|rT|jr | |vrtd | f|||| <||<}t| |n5|r t| d}t|ts| }nt| }t|tr9|j!r9t|j"ts9|j#}||| <||<t|ttfrS|j$durS|j%|_$q_|&| | |q_|r| |vsk|| |ur|| |rsJt|r|| }||| <q_qP|r|sd}||_'||_(||_)dS) NrOTrP__mapper_args__cs tjSr6)rrr rr r!r s z?_ClassScanMapperConfig._scan_attributes..mapper_args_fn __tablename____table_args__z3__table_args__ value must be a tuple, dict, or NonezRegular (i.e. not __special__) attribute '%s.%s' uses @declared_attr, but owning class %s is mapped - not applying to subclass %s.zMapper properties (i.e. deferred,column_property(), relationship(), etc.) must be declared as @declared_attr callables on declarative mixin classes. For dataclass field() objects, use a lambda:zXAttribute '%s' on class %s cannot be processed due to @declared_attr.cascading; skipping)*rr:r{rr.r"r2r_produce_column_copiesrsupdaterJrrrDtuplerrVr ArgumentErrorrFrrHrWrrrQrG__get__setattrr-r rZ_is_internal_proxyZoriginal_propertyZ descriptordoc__doc___warn_for_decl_attributesr|r}r)rZr:r{rr|Zinherited_table_argsr}rbasesr1Z class_mappedrZlocally_collected_columnsrIrE is_dataclassZ check_declrr rr!rs                         z'_ClassScanMapperConfig._scan_attributescCs&t|tjrtd||fdSdS)NzAttribute '%s' on class %s appears to be a non-schema 'sqlalchemy.sql.column()' object; this won't be part of the declarative mapping)rDrZ ColumnClauserrH)rZrrrvr r r!rs z0_ClassScanMapperConfig._warn_for_decl_attributesc Cs|j}|j}i}|j}|D]@\}}} t|trN|||rq|jr&td||vrNd|vr8|jp2||dj vsN| ||<} |j | _ t ||| | ||<q|S)NzColumns with foreign keys to other columns must be declared as @declared_attr callables on declarative mixin classes. For dataclass field() objects, use a lambda:.rh) rr:r{rDrZ foreign_keysrrQrIrvZ_copy_creation_orderr) rZZattributes_for_classrrr:Zlocally_collected_attributesr{rIrErZcopy_r r r!rs.   z-_ClassScanMapperConfig._produce_column_copiescCs6|j}|j}|j}t|ddd}t|D]}|dvrq||}t|r4|jr.td|jt ||}nt |t rN|j |urN|j |krNt|j }t|||t |trjt|dkrjt |dttfrjtd|qt |ttfs|d s|||||||st|||q|d krtd |||<qdS) NrOTrP)rhrrzUse of @declared_attr.cascading only applies to Declarative 'mixin' and 'abstract' classes. Currently, this flag is ignored on mapped class %srrzIgnoring declarative-like tuple value of attribute '%s': possibly a copy-and-paste error with a comma accidentally placed at the end of the line?__rzdAttribute name 'metadata' is reserved for the MetaData instance when using a declarative base class.)rr:rLr2listrFrGrrHr-rDr r*rrrrrtrr startswithpoprrrQ)rZrr: our_stuffZ late_mappedkr\r r r!rsf         z3_ClassScanMapperConfig._extract_mappable_attributesc s |jtjfddd|j}tt}tD]R\}}t |t t frO|j D]#}t |t rM|jdurMt||t |t sH||j|||q*qt |t rnt||||j|||||jkrn|=q|D]\}}t|dkrtd|j|dt|fqsdS)Ncs |jSr6rrrr r! s zB_ClassScanMapperConfig._extract_declared_columns..rrzzOn class %r, Column object %r named directly multiple times, only one will be used: %s. Consider using orm.synonym insteadz, )rLrZsort_dictionaryrz collections defaultdictrrrrDrr columnsrr>_undefer_column_namerIrrrtrHrKjoinsorted)rZrzZname_to_prop_keyrrvcolrIkeysr rr!rs@           z0_ClassScanMapperConfig._extract_declared_columnsNcCsf|j}|j}|j}|j}|j}t|}t|ddd}|_d|vr|durt|dr3t |j }nt }|durdi} } |rbt |trH|} nt |trbt |dtr`|dd|d} } n|} |d } | rm| | d <|d } | rxd | d <|d||||gt|t| Ri| }n|dur|j}|r|D]} |j| std| jq||_dS)NcSs|jSr6r)rvr r r!r7sz5_ClassScanMapperConfig._setup_table..rrh __table_cls__r rZ__autoload_with__ autoload_withZ __autoload__Tautoloadz8Can't add additional column %r when specifying __table__)rr}r|r:rzrr4rr3runbound_method_to_callablerrrDrrr&r]_metadata_for_clsrhrvZcontains_columnrrrrf)rZr>rr}r|r:rzr5Z table_clsargsZtable_kwrrrvr r r!r-sh           z#_ClassScanMapperConfig._setup_tablecCst|jdr |jjS|jjS)Nr)r3rrr9)rZr5r r r!ris z(_ClassScanMapperConfig._metadata_for_clsc Cs|j}|j}|j}|j}|dd}|durSg}|jD]!}t|}|dur&qt|durrr|rzrgrurvinherited_mapperinherited_tableZinherited_persist_selectabler r r!rnos          z)_ClassScanMapperConfig._setup_inheritancec s|j}|jr |}ni}|r||d|vr#t|}||ddD]}||vr8||}|j||||<q%d|vrW|d}t|trI|j}||j urWt d|d|j r_|j |d<|j r|ddst |j j }d|vrtfdd |jDjpd |d<}|d d |jDt|D]"\}} t| tjsq|jvrj|} t| tr| g| j||<q|} || d<| |_dS) NrL)Zversion_id_colZpolymorphic_onrgz:mapper inherits argument given for non-inheriting class %sZconcreteFexclude_propertiescsg|] }|jvr|jqSr )Z_columntopropertyrrrvrr r! s zD_ClassScanMapperConfig._prepare_mapper_arguments..r cSsg|]}|jqSr rrr r r!rs)rLrrrr{r&rDrr*rgrrQr"rfrrvunionrdifference_updaterzrrrZ ColumnElementZ_propsrrcopyr~) rZr?rLr~rvZ inherits_argrrrpZresult_mapper_argsr rr!_prepare_mapper_argumentssf              z0_ClassScanMapperConfig._prepare_mapper_argumentscCsJ||t|jdrt|jj}nt}|d||j|jfi|j S)N__mapper_cls__r=) rr3rrrrrr]rfr~rqr r r!r^ s  z_ClassScanMapperConfig.mapr6)rWrbrcrdr[rrrrrrrrrrrnrrrwr^rxr r ror!rRs" % GR !E )<RKrRcCs2tjj}||}t|rt||js||S|Sr6)rrArBcallablerDrC)Zfield_metadatarZdecl_apirEr r r!rs  rcseZdZeZddZeddZej ddZe ddZ e dd Z e d d Z e d d Ze dddZejffdd ZZS)rcCsdSr6r r_r r r!r`+sz$_DeferredMapperConfig._early_mappingcCs|Sr6)_clsrr r r!r.sz_DeferredMapperConfig.clscCs t||j|_||j|j<dSr6)weakrefref_remove_config_clsr_configs)rZr*r r r!r2scCs|j|ddSr6)rr)rrr r r!r7sz(_DeferredMapperConfig._remove_config_clscCst|to t||jvSr6)rDrVrrrrr*r r r!r;sz_DeferredMapperConfig.has_clscCs*t|dr |tj|dt|d)N_sa_raise_deferred_configzOClass %s has a deferred mapping on it. It is not yet usable as a mapped class.)msg)r3rorm_excZUnmappedClassErrorZ_safe_cls_namerr r r!raise_unmapped_for_cls@s z,_DeferredMapperConfig.raise_unmapped_for_clscCs|jt|Sr6)rrrrr r r!rKsz$_DeferredMapperConfig.config_for_clsTcsvfdddd|jDD}|s|Stdd|Dg}D]|fddjDq"tt||S)Ncs&g|]\}}|durt|r|qSr6)r,)rmrS)base_clsr r!rQsz:_DeferredMapperConfig.classes_for_base..cSsg|]}||jfqSr rrrr r r!rSscss|]}|j|fVqdSr6rrr r r! Zsz9_DeferredMapperConfig.classes_for_base..c3s(|]}|vr|fVqdSr6r )rr) all_m_by_clsm_clsr r!r^s )rvaluesrextendr'rrsort)rrrclasses_for_baseZtuplesr )rrrr!rOs  z&_DeferredMapperConfig.classes_for_basecs |j|jdtt||Sr6)rrrrirr^r_ror r!r^esz_DeferredMapperConfig.map)T)rWrbrcrrXrr`propertyrsetterrerrrrrrwr^rxr r ror!r(s$       rcCsd|jvrt|tr t|||jj|dd|j||dSt|trK|j D]}t|trA|j durAt|||jj|ddq(|j||dSt|t rY|j||dSt|t rq|j |krqt|j }|j||dSt||||jdSt|||dS)zadd an attribute to an existing declarative class. This runs through the logic to determine MapperProperty, adds it to the Mapper, adds a column to the mapped Table, etc. r=T)Zreplace_existingN)r%rDrrrhrr=Z add_propertyrrr>rr rrrV __setattr___expire_memoizations)rrr\rr r r!_add_attributejs(        rcCsjd|jvr-||jvr-|jjs-|j|}t|ttttfr tdt |||j dSt ||dS)Nr=zsh                        %  @D  B!