U Zjʮ@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) topologicalcCs0t|rt|St|r(t|ddSdSdS)NF) configure)_DeferredMapperConfighas_clsconfig_for_clsr rclsr [/var/www/html/TRUCKING_PROJECT/venv/lib/python3.8/site-packages/sqlalchemy/orm/decl_base.py_declared_mapping_info%s    r"cCs\|tkr dS|jddrB|jD]}t|}|dk r |Sq dSt|}|rT|jS|SdS)N __abstract__F)object__dict__get __bases__"_resolve_for_abstract_or_classical_dive_for_cls_managerclass_)rsupZ clsmanagerr r r!r(0s  r(FcCs|dks tt|tsdS||jkr.t||S|jddD]@}t|}||jkr<||ksn|rj||jkr)rr^rrfr[r@Z mapper_clsr r r!r_s z_ImperativeMapperConfig.mapcCs|j}|dd}|dkrg}|jD]:}t|}|dkr:q$t|dk r$t|ddds$||q$|rt|dkrt d||f|d}nt |t r|j }||_ dS)NrgrPTrQr&Class %s has multiple mapped bases: %rr)rr&r'r(r"r3appendlenrrRrErr*rg)r[r@rrginherits_searchcr r r!rns>      z*_ImperativeMapperConfig._setup_inheritance) rXrbrcrdr\r EMPTY_DICTr_rn __classcell__r r ror!r=s r=cseZdZdZfddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dddZ ddZddZddZejfddZZS) rS) r;rfpersist_selectabledeclared_columns column_copies table_args tablename mapper_argsmapper_args_fnrgc s|r t|ni|_tt||||d|_t|_i|_| | t j Ht |j|j|j||||||||W5QRXdSr7)dictr;rirSr\rysetrzr{_setup_declared_events_scan_attributesrjrkrrlrLrrm_extract_mappable_attributes_extract_declared_columns _setup_tablernra)r[r:rTr;r?r@ror r!r\,s$   z_ClassScanMapperConfig.__init__csLtjdr$ttdfdd}tjdrHttdfdd}dS) N__declare_last__after_configuredcsjdSr7)rrr r[r r!rTszG_ClassScanMapperConfig._setup_declared_events..after_configured__declare_first__before_configuredcsjdSr7)rrr rr r!rZszH_ClassScanMapperConfig._setup_declared_events..before_configured)r3rrZ listens_forr)r[rrr rr!rQs     z-_ClassScanMapperConfig._setup_declared_eventscsntdddkr"fdd}nHfddtDfddtDtfdd}|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||k Sr7)r.)keyrFrr r!attribute_is_overriddeniszR_ClassScanMapperConfig._cls_attr_override_checker..attribute_is_overriddencs$i|]}|jkr|j|jqSr metadatarJ.0fsa_dataclass_metadata_keyr r! ns zE_ClassScanMapperConfig._cls_attr_override_checker..cs$i|]}|jkr|j|jqSr rrrr r!rss cst|r|j}|}t|r(|j}||kr4dS|k r@dS|}t||}||krddS|k rzt|trzdS||krdS|k rdSdS)NFT)rGfgetr&r.rEr )rrFretZ all_field)absentall_datacls_fieldsrlocal_datacls_fieldsr r!r{s.    )r3rZdataclass_fieldslocal_dataclass_fieldsr$)r[rrr )rrrrrr!_cls_attr_override_checker^s   (z1_ClassScanMapperConfig._cls_attr_override_checkercs<tdddkr"fdd}ntfdd}|S)zproduce a function to iterate the "attributes" of a class, adjusting for SQLAlchemy fields embedded in dataclass fields. rNc3s&tD]\}}||dfVq dS)NF)varsitems)rJrFrr r!local_attributes_for_classszM_ClassScanMapperConfig._cls_attr_resolver..local_attributes_for_classc3sjtD]0}|jkr |j|jt|jdfVq tD]\}}|krH||dfVqHdS)NTF)rrraddrJ_as_dc_declaredattrrr)fieldrJrFr field_namesrr r!rs  )r3r)r[rrr rr!_cls_attr_resolvers z)_ClassScanMapperConfig._cls_attr_resolverc sJ|j|j}|j}d}d}}d}||j}g}jD]b} | k o^t| dk o^t| ddd } || } | s| k r|| |} ni} | | | | | fq8|D]\} } } } | | | D]`\} }}| dkrt || }|s| r|rfdd}q| dkr(t || }|s | r |r j }q| dkrt || }|s | rP|r j }t|tttdfsvtd | k r d}q| rt|rtd | j| | fqq| k rt|trqq t|trtd nt|r|jr<| |krtd | f|||| <||<}t| |nj|rdt| d}t|tsn| }n t| }t|tr|j!rt|j"ts|j#}||| <||<t|ttfr|j$dkr|j%|_$n|&| | |q|r| |ks|| |k r|| |rt't|r| }||| <qq|r4|s4d}||_(||_)||_*dS) NrPTrQ__mapper_args__cs tjSr7)rrr rr r!r sz?_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"r3r_produce_column_copiesrsupdaterKrrrEtuplerrWr ArgumentErrorrGrrIrXrrrRrH__get__setattrr.r rZ_is_internal_proxyZoriginal_propertyZ descriptordoc__doc___warn_for_decl_attributesr,r|r}r)r[r;r{rr|Zinherited_table_argsr}rbasesr2Z class_mappedrZlocally_collected_columnsrJrF is_dataclassZ check_declrr rr!rs                             z'_ClassScanMapperConfig._scan_attributescCs"t|tjrtd||fdS)NzAttribute '%s' on class %s appears to be a non-schema 'sqlalchemy.sql.column()' object; this won't be part of the declarative mapping)rErZ ColumnClauserrI)r[rrrvr r r!rs  z0_ClassScanMapperConfig._warn_for_decl_attributesc Cs|j}|j}i}|j}|D]\}}} t|tr|||r>qq|jrPtdq||krd|krt|jph||dj ks| ||<} |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{rErZ foreign_keysrrRrJrvZ_copy_creation_orderr) r[Zattributes_for_classrrr;Zlocally_collected_attributesr{rJrFrZcopy_r r r!rs,    z-_ClassScanMapperConfig._produce_column_copiescCs@|j}|j}|j}t|ddd}t|D]}|dkr8q(||}t|rj|jr^td|jt ||}n4t |t r|j |k r|j |krt|j }t|||t |trt|dkrt |dttfrtd|q(nZt |ttfs|d s(|||||||s(t|||q(n|d kr2td |||<q(dS) NrPTrQ)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;rMr3listrGrHrrIr.rEr r*rrrrrtrr startswithpoprrrR)r[rr; our_stuffZ late_mappedkr]r r r!rsh        z3_ClassScanMapperConfig._extract_mappable_attributesc s |jtjfddd|j}tt}tD]\}}t |t t fr|j D]F}t |t rT|jdkrTt||t |t s||j|||qTq8t |t r8t||||j|||||jkr8|=q8|D]4\}}t|dkrtd|j|dt|fqdS)Ncs |jSr7rrrr r! 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, )rMrZsort_dictionaryrz collections defaultdictrrrrErr columnsrr?_undefer_column_namerJrrrtrIrLjoinsorted)r[rzZname_to_prop_keyrrvcolrJkeysr rr!rs8          z0_ClassScanMapperConfig._extract_declared_columnsNcCsn|j}|j}|j}|j}|j}t|}t|ddd}|_d|kr$|dkr$t|drjt |j }nt }|dk rddi} } |rt |tr|} n4t |trt |dtr|dd|d} } n|} |d } | r| | d <|d } | rd | d <|d||||ft|t| | }n@|dkr4|j}|rd|D]$} |j| s>td| jq>||_dS)NcSs|jSr7r)rvr r r!r7rz5_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;rzrr5rr4runbound_method_to_callablerrrErrr&r^_metadata_for_clsrhrvZcontains_columnrrrrf)r[r?rr}r|r;rzr6Z table_clsargsZtable_kwrrrvr r r!r-sf          z#_ClassScanMapperConfig._setup_tablecCs t|jdr|jjS|jjSdS)Nr)r4rrr:)r[r6r r r!ris z(_ClassScanMapperConfig._metadata_for_clsc Cs|j}|j}|j}|j}|dd}|dkrg}|jD]B}t|}|dkrLq6t|dk r6t|ddds6||kr6| |q6|rt |dkrt d||f|d}nt |tr|j}||_|dkr|jdkrt|dst d |n|jrt|j} | j} | j} |dkr|r t d |D]}|j| jkrh| j|j|krLq$t d ||| j|jf|jrzt d | || dk r$| | k r$| |q$dS) NrgrPTrQrrrrZ __no_table__zwClass %r does not have a __table__ or __tablename__ specified and does not inherit from an existing table-mapped class.z?Can't place __table_args__ on an inherited class with no table.z;Column '%s' on class %s conflicts with existing column '%s'zDCan't place primary key columns on an inherited class with no table.)rfrr|rzr&r'r(r"r3rsrtrrRrErr*rgryrrJrvZ primary_key append_columnZ_refresh_for_new_column) r[r@r?rr|rzrgrurvinherited_mapperinherited_tableZinherited_persist_selectabler r r!rnos         z)_ClassScanMapperConfig._setup_inheritancec s|j}|jr|}ni}|r(||d|krFt|}||ddD]&}||krJ||}|j||||<qJd|kr|d}t|tr|j}||j k rt d|d|j r|j |d<|j r|ddst |j j }d|kr2tfdd |jDjpd |d<}|d d |jDt|D]N\}} t| tjsXq>|jkr>j|} t| tr>| g| j||<q>|} || d<| |_dS) NrM)Zversion_id_colZpolymorphic_onrgz:mapper inherits argument given for non-inheriting class %sZconcreteFexclude_propertiescsg|]}|jkr|jqSr )Z_columntopropertyrrrvrr r! s zD_ClassScanMapperConfig._prepare_mapper_arguments..r cSsg|] }|jqSr rrr r r!rs)rMrrrr{r&rErr*rgrrRr"rfrrvunionrdifference_updaterzrrrZ ColumnElementZ_propsrrcopyr~) r[r@rMr~rvZ inherits_argrrrpZresult_mapper_argsr rr!_prepare_mapper_argumentssb             z0_ClassScanMapperConfig._prepare_mapper_argumentscCsF||t|jdr&t|jj}nt}|d||j|jf|j S)N__mapper_cls__r>) rr4rrrrrr^rfr~rqr r r!r_ s  z_ClassScanMapperConfig.map)N)rXrbrcrdr\rrrrrrrrrrrnrrrwr_rxr r ror!rSs  % GR !E) <RKrScCs6tjj}||}t|r.t||js.||S|SdSr7)rrBrCcallablerErD)Zfield_metadatarZdecl_apirFr r r!rs  rcseZdZeZddZeddZej ddZe ddZ e dd Z e d d Z e d d Ze dddZejffdd ZZS)rcCsdSr7r r`r r r!ra+sz$_DeferredMapperConfig._early_mappingcCs|Sr7)_clsrr r r!r.sz_DeferredMapperConfig.clscCs t||j|_||j|j<dSr7)weakrefref_remove_config_clsr_configs)r[r*r r r!r2scCs|j|ddSr7)rr)rrr r r!r7sz(_DeferredMapperConfig._remove_config_clscCst|tot||jkSr7)rErWrrrrr*r r r!r;sz_DeferredMapperConfig.has_clscCs.t|dr|tj|dt|ddS)N_sa_raise_deferred_configzOClass %s has a deferred mapping on it. It is not yet usable as a mapped class.)msg)r4rorm_excZUnmappedClassErrorZ_safe_cls_namerr r r!raise_unmapped_for_cls@s z,_DeferredMapperConfig.raise_unmapped_for_clscCs|jt|Sr7)rrrrr r r!rKsz$_DeferredMapperConfig.config_for_clsTcsvfdddd|jDD}|s*|Stdd|Dg}D] |fddjDqDtt||S)Ncs&g|]\}}|dk rt|r|qSr7)r-)rmrT)base_clsr r!rQs z:_DeferredMapperConfig.classes_for_base..cSsg|]}||jfqSr rrrr r r!rSscss|]}|j|fVqdSr7rrr r r! Zsz9_DeferredMapperConfig.classes_for_base..c3s&|]}|kr|fVqdSr7r )rr) all_m_by_clsm_clsr r!r^s)rvaluesrextendr'rrsort)rrrclasses_for_baseZtuplesr )rrrr!rOs  z&_DeferredMapperConfig.classes_for_basecs |j|jdtt||Sr7)rrrrirr_r`ror r!r_esz_DeferredMapperConfig.map)T)rXrbrcrrYrrapropertyrsetterrerrrrrrwr_rxr r ror!r(s"       rcCsd|jkrt|tr>t|||jj|dd|j||nt|tr|j D]2}t|trN|j dkrNt|||jj|ddqN|j||n`t|t r|j||nFt|t r|j |krt|j }|j||nt||||jnt|||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%rErrrhrr>Z add_propertyrrr?rr rrrW __setattr___expire_memoizations)rrr]rr r r!_add_attributejs&         rcCsjd|jkrZ||jkrZ|jjsZ|j|}t|ttttfrBtdqft |||j n t ||dS)Nr>zZ_dispose_calledrErrrr NotImplementedErrorrW __delattr__r)rrr]r r r!_del_attributes"    rcKsBt|}|D]0}t||s,td||jft||||q dS)a=A simple constructor that allows initialization from kwargs. Sets attributes on the constructed instance using the names and values in ``kwargs``. Only keys that are present as attributes of the instance's class are allowed. These could be, for example, any mapped columns or relationships. z(%r is an invalid keyword argument for %sN)rWr4 TypeErrorrXr)r[kwargsrTrr r r!_declarative_constructors   rr\cCs$|jdkr||_|jdkr ||_dSr7)rrJ)rcolumnr r r!rs  r)F)8r __future__rrrZsqlalchemy.ormrrrrrrrjr r r2r r Zdescriptor_propsr rZ interfacesrrrMrrrrsqlrZ sql.schemarrrr"r(r3r)r<rAZpreload_modulerGrKr$r8r=rSrrrrrrXrr r r r!sf                         %  @D B!