o [h]}@sdZddlZddlZddlmZddlmZddlmZddlmZdd lm Z dd l m Z ddl mZ dd l m Z dd l mZdd l mZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddl mZda a!ddZ"ddZ#ddZ$d2dd Z%d!d"Z&d#d$Z'Gd%d&d&e(Z)Gd'd(d(e)Z*d)d*Z+d+d,Z,d-d.Z-d/e-_.d0d1Z/dS)3z(Internal implementation for declarative.N)instrumentation) clsregistry)event)exc)util) class_mapper)mapper) mapperlib)synonym)QueryableAttribute)_is_mapped_class)InspectionAttr)MapperProperty)ColumnProperty)CompositeProperty)Column)Table) expression) topologicalcCs,t|r t|St|rt|ddSdS)NF) configure)_DeferredMapperConfighas_clsconfig_for_clsrr clsr/home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/sqlalchemy/ext/declarative/base.py_declared_mapping_info%s   rcCsZ|turdSt|dddr!|jD]}t|}|dur|SqdSt|}|dur+|S|S)N __abstract__Tstrict)object_get_immediate_cls_attr __bases__"_resolve_for_abstract_or_classical"_dive_for_classically_mapped_class)rsupZ classicalrrrr&0s r&cCsLt|drdSt|}|dur|S|jD]}t|}|dur#|SqdS)N_decl_class_registry)hasattrrZmanager_of_classr%r')rmanagerr(r rrrr'Cs   r'FcCstt|tsdS|jD]-}t|d}| ot|du}||jvr7||us0|r+||jvr7n |s7|s7t||Sq dS)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. Nr)) issubclassr#__mro__r*r'__dict__r%getattr)rattrnamer"baseZ_is_declarative_inheritsZ_is_classicial_inheritsrrrr$Ws&     r$cCsBtdurddlmattjfat|dddrdSt|||dS)Nr) declared_attrr Tr!)r2apir classpropertydeclarative_propsr$ _MapperConfig setup_mapping)r classnamedict_rrr_as_declarativeys   r:cCs0t|trt|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) isinstancer5r/rwarn)objnamerrrr_check_declared_props_nocascades  r@c@sxeZdZeddZddZddZddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZddZdS)r6cCs6t|dddp t|d}|rt}nt}||||dS)N_sa_decl_prepare_nocascadeTr!Z_sa_decl_prepare)r$r*rr6)rcls_r8r9Z defer_mapZcfg_clsrrrr7sz_MapperConfig.setup_mappingcCs||_t||_||_d|_t|_t|_ i|_ | i|j_ | tjz$t|j|j|||||WtjdStjwN)rdictr9r8persist_selectabler OrderedDict propertiessetdeclared_columns column_copies_setup_declared_events_sa_declared_attr_reg_scan_attributesr Z_CONFIGURE_MUTEXacquirerZ add_class_extract_mappable_attributes_extract_declared_columns _setup_table_setup_inheritance_early_mappingrelease)selfrBr8r9rrr__init__s&    z_MapperConfig.__init__cCs |dSrC)maprUrrrrSs z_MapperConfig._early_mappingcsPtjdrttdfdd}tjdr&ttdfdd}dSdS) N__declare_last__after_configuredcjdSrC)rrYrrXrrrZz>_MapperConfig._setup_declared_events..after_configured__declare_first__before_configuredcr[rC)rr]rrXrrr^r\z?_MapperConfig._setup_declared_events..before_configured)r$rrZ listens_forr )rUrZr^rrXrrKs    z$_MapperConfig._setup_declared_eventsc s|j|j}|j}d}d}}d}jD]}|uo(t|duo(t|ddd }|s4|ur4||t|D]\} } | dkrUt | | } |sT|rN| rTfdd}q:| dkrit | | } |sh|re| rhj }q:| dkrt | | } |s|ry| rj }t |t ttdfstd |urd}q:|rt | trtd |j| |fq:|ur0t | trq:t | trtd t | tr)t | tj} | s| jr| |vrtd | f| | || <|| <} t| | n(| rtd t| } t | tr| j rt | j!ts| j"} | || <|| <t | ttfr(| j#dur(| j$| _#q:|%|| | q:q|r:|s:d}||_&||_'||_(dS)NrATr!__mapper_args__cs tjSrC)rDr_rrrrmapper_args_fns z6_MapperConfig._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.zXAttribute '%s' on class %s cannot be processed due to @declared_attr.cascading; skippingzJUse of sqlalchemy.util.classproperty on declarative classes is deprecated.))rr9rJr-rr$_produce_column_copiesvarsitemsr@rarbr<tuplerDtyper ArgumentErrorr5rr=__name__rrInvalidRequestErrorr4r;__get__setattrZwarn_deprecatedr/rZ_is_internal_proxyZoriginal_propertyZ descriptordoc__doc___warn_for_decl_attributes table_args tablenamer`)rUr9rJr`rpZinherited_table_argsrqr1Z class_mappedr?r>Z check_declZ oldclasspropretrrrrMs                      r z_MapperConfig._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)r<rZ ColumnClauserr=)rUrkeycrrrrocs z'_MapperConfig._warn_for_decl_attributescCs|j}|j}|j}t|D]A\}}t|trPt|||ur q|jr(t d||vrPd|vr:|j p4||dj vsP| ||<}|j|_t||||||<qdS)NzvColumns with foreign keys to other columns must be declared as @declared_attr callables on declarative mixin classes. __table__)rr9rJrdrer<rr/Z foreign_keysrrjr?rtcopy_creation_orderrl)rUr1rr9rJr?r>Zcopy_rrrrcls(  z$_MapperConfig._produce_column_copiescCsN|j}|j}|j}t|ddd}t|D]}|dvrq||}t|tr:t|tr4|jr4t d|jt ||}nt|t rT|j |urT|j|krTt|j}t|||t|trpt|dkrpt|dttfrpt d|qt|ttfs|d s|||||||st|||q|d krtd t||}|||<qdS) NrATr!)rurar_zUse 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?__metadatazdAttribute name 'metadata' is reserved for the MetaData instance when using a declarative base class.)rr9rGr$listr<r5r2r;rr=r/r class_rsr rlrflenrr startswithpoprorrjr_deferred_relationship)rUrr9 our_stuffZ late_mappedkvalueproprrrrOsb             z*_MapperConfig._extract_mappable_attributesc s|jjfddd|j}tt}tD]R\}}t|t t frN|j D]#}t|t rL|j durLt||t|t sG||j|||q)qt|t rmt||||j|||||jkrm|=q|D]\}}t|dkrtd|j|dt|fqrdS)Ncs |jSrCrwrsrrrs z9_MapperConfig._extract_declared_columns..rrzzOn class %r, Column object %r named directly multiple times, only one will be used: %s. Consider using orm.synonym insteadz, )rGsortrI collections defaultdictrHrzrer<rrcolumnsrtable_undefer_column_namer?addrsr|rr=r8joinsorted)rUrIZname_to_prop_keyrsrtcolr?keysrrrrPs<          z'_MapperConfig._extract_declared_columnsc Cs4|j}|j}|j}|j}|j}t|ddd}|_d}t|dr(t|j }nt }d|vr}|dur|di}} |r[t |t rA|} nt |t r[t |dt rY|dd|d}} n|}|d } | rfd | d <|||jgt |t |Ri| |_}n|j}|r|D]} |j| std | jq||_dS) NcSs|jSrCr)rtrrrrsz,_MapperConfig._setup_table..r __table_cls__rurrZ __autoload__Tautoloadz8Can't add additional column %r when specifying __table__)rrqrpr9rIrr*runbound_method_to_callablerrr<rDrfgetryrurtZcontains_columnrrhrs local_table) rUrrqrpr9rIrZ table_clsargsZtable_kwrrtrrrrQsZ         z_MapperConfig._setup_tablec Csr|j}|j}|j}|j}g}|jD]}t|}|durqt|dur.t|ddds.||q|rFt |dkr@t d||f|d|_ nd|_ |dur^|j dur^t|ds^t d||j rt|j }|j}|j } |dur|rwt d |D]=}|j|jvr|j|j|urqyt d |||j|jf|jrt d ||| dur| |ur| |qydSdSdS) NrATr!rz&Class %s has multiple mapped bases: %rrZ __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.)rrrprIr%r&rr$appendr|rrjinheritsrErhr?rtZ primary_key append_columnZ_refresh_for_new_column) rUrrrprIrrtinherited_mapperinherited_tableZinherited_persist_selectablerrrrR's           z _MapperConfig._setup_inheritancec s>|j}|jr |}ni}dD]}||vr"||}|j||||<qd|vs+Jd|jr3|j|d<|jr|ddst|jj}d|vritfdd|jD j pXd |d<}| d d|j Dt |D]"\}}t|tjszqo|jvrj|}t|tr|g|j||<qo|} || d <| |_dS) N)Zversion_id_colZpolymorphic_onrz=Can't specify 'inherits' explicitly with declarative mappingsZconcreteFexclude_propertiescsg|] }|jvr|jqSr)Z_columntopropertyrs.0rtrrr s  z;_MapperConfig._prepare_mapper_arguments..rcSsg|]}|jqSrrrrrrrsrG)rGr`rJrrrrrHrtunionrdifference_updaterIrzrer<rZ ColumnElementZ_propsrrrv mapper_args) rUrGrrvrrrpZresult_mapper_argsrrr_prepare_mapper_argumentsrsN            z'_MapperConfig._prepare_mapper_argumentscCsR|t|jdrt|jj}nt}||j|jfi|j|j_ }|j` |S)N__mapper_cls__) rr*rrrrr rr __mapper__rL)rUZ mapper_clsZmp_rrrrWs z_MapperConfig.mapN)ri __module__ __qualname__ classmethodr7rVrSrKrMrorcrOrPrQrRrrWrrrrr6s   $  F'3K .cSsg|]}||jfqSrrrrrrrrscss|]}|j|fVqdSrCrrrrr sz9_DeferredMapperConfig.classes_for_base..c3s(|]}|vr|fVqdSrCr)rr) all_m_by_clsm_clsrrrs)rvaluesrDextendr%rzrr)rrrclasses_for_baseZtuplesr)rrrrrs  z&_DeferredMapperConfig.classes_for_basecs|j|jdtt|SrC)rr~rsuperrrWrX __class__rrrWsz_DeferredMapperConfig.map)T)rirrrrFrrSpropertyrsetterrrrrrrrW __classcell__rrrrrs$       rcCsd|jvrt|trt|||j||j||dSt|trG|j D]}t|tr=|j dur=t|||j|q&|j||dSt|t rY|j|t ||dSt|tru|j|krut|j}|j|t ||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. rN)r.r<rrrurrZ add_propertyrrrrrrr rsr rg __setattr___expire_memoizations)rrsrrrrr_add_attributes0            rcCsjd|jvr-||jvr-|jjs-|j|}t|ttttfr tdt |||j dSt ||dS)NrzsR                      " .B%