U <cR@sRddlZddlZddlmZmZddlmZddlmZddlZ ddl Z ddl Z ddl m Z ddl mZmZmZddlmZddlmZdd lmZdd lmZdd lmZdd l mZdd lmZmZddlm Z ddl!m"Z"ddl#m$Z$m%Z%m&Z&ddl'm(Z(m)Z)m*Z*m+Z+ddl,m-Z-m.Z.m/Z/m0Z0ddl1m2Z2Gddde-Z3GdddZ4dS)N) defaultdict namedtuple)copy)List)IntegrityError) ForeignKeyfuncinspect)json)Config)command)UNIQUE_VIOLATION)errors)close_all_sessions) CmfRelBaseCmfType)fields)models) CmfOrmErrorCmfOrmUniqueErrorCmfOrmIntegrityError) BaseModelMetaDEFAULT_DATASOURCE BaseModel CmfGM2MModel)BaseDataDriver BaseMapperMapperinc_select_count)imutable_deep_copycsBeZdZdZiZeZejj j eedZ iZ fddZ eddZeddZed d Zedid d ZdjddZddZddZeddZddZddZddZddZeddZed d!Zed"d#Zdkd%d&Z edle!ed(d)d*Z"dmd+d,Z#ed-d.Z$ej%d fd/d0Z&ed1d2Z'ed3d4Z(d5d6Z)ednd7d8Z*doed9d:d;Z+dd?d@Z-d d dAdBdCZ.d d d'd dDdEdFZ/dGdHZ0dIdJZ1dKdLZ2d'dMdNdOZ3dPdQZ4d dRdSdTZ5fdUdVZ6fdWdXZ7dYdZZ8d[d\Z9d]d^Z:d_d`Z;efdadbZdgdhZ?Z@S)pSQLAlchemyDataDriveru dataproviders['default'] = { 'type': 'sqlalchemy', 'sqlachemy.url': 'postgresql://user:password@host/db', 'sqlachemy.echo': True, # Достаточно для отладки запросов в develop mode } # Для отладки в продакшене логгеры: sqlalchemy.engine - controls SQL echoing. set to logging.INFO for SQL query output, logging. DEBUG for query + result set output. These settings are equivalent to echo=True and echo="debug" on create_engine.echo, respectively. sqlalchemy.pool - controls connection pool logging. set to logging.INFO to log connection invalidation and recycle events; set to logging.DEBUG to additionally log all pool checkins and checkouts. These settings are equivalent to pool_echo=True and pool_echo="debug" on create_engine.echo_pool, respectively. sqlalchemy.dialects - controls custom logging for SQL dialects, to the extend that logging is used within specific dialects, which is generally minimal. sqlalchemy.orm - controls logging of various ORM functions to the extent that logging is used within the ORM, which is generally minimal. Set to logging.INFO to log some top-level information on mapper configurations. Example: logging.basicConfig() logging.getLogger('sqlalchemy.engine').setLevel(logging.INFO) )Zclass_registryZmetadatacsTtj||tj|jfdd|D|_tjtjj |jd|_ | dS)NcSs*i|]"\}}|dr|ddd|qS)z sqlalchemy.r) startswithreplace).0kvr)./data_providers/sqlalchemy.py Ds z1SQLAlchemyDataDriver.__init__..)Zbind) super__init__ sqlalchemyZengine_from_configconfigitemsengineormZscoped_sessionZ sessionmakerSession init_metaselfargskwargs __class__r)r*r-?szSQLAlchemyDataDriver.__init__cCsVg}|jD]B\}}t|tr"q|jr:|d|ffq|jr|d|ffq|S)Nuniqueindex)rr0 issubclassrr;appendr<)cls cmf_modelZindexescmf_field_name cmf_fieldr)r)r*Z __get_indexesOs z"SQLAlchemyDataDriver.__get_indexescCs&|j|}|r"|j|=||j|<|SN)cached_queriesget)r?keyretr)r)r* get_query]s   zSQLAlchemyDataDriver.get_querycCs4||j|<t|jdkr0|jD]}|j|=q0qdS)N)rDlen)r?queryrFZtmpr)r)r* set_querygs   zSQLAlchemyDataDriver.set_queryNcCs|j}|jrdSt|tjr(t}nt|tjr>tj}nt|tj r^t |j |j }n~t|tj r|t|jg}|jr|t|jtj||j|jdSt|tjrt}nt|tjrt|j}nt|tjrdSt|tjrdSt|tjrdSt|tjr dSt|tjr2dSt|tjrNtjdd}nt|tj rft!}nvt|tj"r~t}n^t|tj#rdSt|tj$rdSt|tj%rtj&}n$t|tj'rtj(}nt)d|tj|||j*|j|jdS)u0Создадим SA поле для моделиN) primary_keynullableT)timezoneu"Не найден тип поля )defaultrNrM)+ class_nameZvirtualr=rZ CmfBigIntr.ZBIGINTZCmfIntZIntegerZ CmfNumericZNumericZ precisionZscaleCmfTUUIDStringZ max_lengthZ foreign_keyr>rZColumnrMrNZCmfTextZTEXTZCmfStr CmfRelationCmfGenericRelationZ CmfGenericM2MZ CmfDateRangeZCmfM2MZ CmfDateTime TIMESTAMPZCmfDateZDATEZCmfJson CmfBackrefCmfGenericBackrefZCmfBoolZBooleanZCmfTimeZTime ExceptionrP)r?Z _cmf_modelrBZ_db_field_name column_nameZsa_typer7r)r)r*_make_sa_columnrsl            z$SQLAlchemyDataDriver._make_sa_columncCst|jj||d}|S)N)schema)r r1Z get_columns)r6 table_namer\columnsr)r)r*inspect_table_columnssz*SQLAlchemyDataDriver.inspect_table_columnscCs>|d|j}|}|d|d|d||dS)Nz alter table z add column  )r[typer3executecommit)r6r]rZZcmf_field_typeZsa_column_typesr)r)r*add_custom_columnsz&SQLAlchemyDataDriver.add_custom_columnc Csdddd}tjtjdddd|d<tjtjdddd|d <tjtjd dd |d <tjtjd dd |d<t|tf|}tj|j j }| }z| || Wn@tk r}z"|jtd|d|W5d}~XYnXdS)u5 Создаем М2М таблицу Tzcustom.r#)abstract __module__ __qualname__uID Объекта)captionrNr<left_iduID Элементаright_iduИмя объекта)rirNZleft_name_cacheuИмя элементаZright_name_cacheu-Ошбика создания таблицы z: N)rZFieldrRZ CmfStr256rarr.r\Z CreateTabledp_model __table__r3rbrcrY transactionrollbackprint)r6 model_namer8r@Zsmtprder)r)r*add_custom_m2m_models    z)SQLAlchemyDataDriver.add_custom_m2m_modelcCs|j}|j|}|s|j}g}g}i}|jD].\}} |} ||| | } | dkrVq2|| q2||D]@\} } | dkr|t j d| ddiql|t j d| ql|||t |d}| dd|Dt ||jf|}||j|<|S) u] Вернём SA модель для CMF модели. TODO: db_name Nr;T)Z __tablename____table_args__cSsi|] }|j|qSr)name)r&cr)r)r*r+sz5SQLAlchemyDataDriver.dp_model_cls..)N)N)__name__models_registryrE tablenamerr0r[r>"_SQLAlchemyDataDriver__get_indexesr.ZIndextupleupdatera sa_base_model)r?r@rqsa_modelrzZ sa_columnsrtZ__table_kwargs__rArBZ sa_field_nameZ sa_columnZ index_typeZ index_tupleZsa_model_kwargsr)r)r* dp_model_clss:   z!SQLAlchemyDataDriver.dp_model_clscCs ||SrC)r)r6r@r)r)r*rlszSQLAlchemyDataDriver.dp_modelcCs tt|jS)uG Вернём CMF модель для SA модели. )getattrrrx)r6rr)r)r*r@szSQLAlchemyDataDriver.cmf_modelcCs t|tjSrC) isinstancer"r~)r6Z dp_instancer)r)r* is_instancesz SQLAlchemyDataDriver.is_instancecCs0i}|D]}|jj||jjj<qtj|d|S)N table_type)rlrmfullnamer.r2polymorphic_union)r6Z cmf_modelsaliasZtablesmodelr)r)r* _union_modelssz"SQLAlchemyDataDriver._union_modelscCsXd}|d|kr&|d|d|dfS|d|krH|d|d|dfStd||dS)N)><==>=<==IN!=><<>NOT INLIKENOT LIKEILIKE NOT ILIKEz SIMILAR TOzNOT SIMILAR TOEXISTS NOT EXISTSrrrzInvalid filter operation)r)paramsZop_listr)r)r* _parse_paramss   z"SQLAlchemyDataDriver._parse_paramscCsddlm}m}t|tjr:|jdkr2||jd|j}nXt|t rt }|D]>}t|tjr|jdkrt||jd| |jqN| |qN|}|dkrd}|dkr||kS|dkr||kS|dkr||kS|d kr||kS|d kr||kS|d kr||kS|d kr| |S|d kr*| |S|dkr>| |S|dkrT| |S|dkrh| |S|dkr~| |S|||SdS)Nr print_debugr/.u<collect_filter_exp: Значение не загруженоZNULLrrrr)rr)rrrrrrrrr) cmf.includerr/rrr_valueDEBUGvaluelistr>in_ZlikeZilikeop)operfieldvalrr/new_valr(r)r)r* _expression*sR                     z SQLAlchemyDataDriver._expressionJoinDatazJleft_table, right_table, alias, left_field_name, right_model, right_modelsTc Cs |r|j}|j}nT|j|}|s.|||t|tjtjfrLt|tjrft d|jd|jd|jd}|r|d|n|}| }|s|rt d|d|jd|jddS|d } t |d krt j j| jj|d } n |||} ||| ||| |} | S) NuXНедопустимый тип поля для вложенной фильтрации .up. Вложенная фильтрация возможно только для CmfRelation и CmfGenericRelationZ_sub__ujНе возможно построить запрос для вложенной фильтрации по u, т.к. у поля u# не указаны related_modelsrrru)rQinstance_classrrE_raise_invalid_fieldr=rTrUCmfSubclassedGenericRelationrrelated_modelsrJr.r2Zaliasedrlrmrr) r6r left_table field_name from_aliasis_models_required field_clsZ alias_partmodels_ right_model right_table join_datar)r)r*_calc_join_dataZs6      z$SQLAlchemyDataDriver._calc_join_dataF)joinscCsb|r tndd|jD}|D]:}|j|kr2q"||jt|jj|j d|jjj k}q"|S)uP TODO cmf_deleted Все данные для необходимых присоединений в joins, добавим недостающие присоединения в запрос. froms могут быть от филдов, при этом без джойна, чтобы принудительно создать join нужен force = True. Или более продвинутый способ определения наличия нужного джойна. Сейчас при построении основного запроса, джойны принудительно добавляются. А при расчёте фильтров, когда нет алиасов от полей, джойн делается только если нет нужного алиаса. cSs"h|]}t|tjjjr|jqSr))rr.sql selectableAliasrv)r&rr)r)r* sz3SQLAlchemyDataDriver._make_joins.._id) set statementlocate_all_fromsrZ outerjoinrrrrwZleft_field_nameid)rKrforceZavailable_joinsrr)r)r* _make_joinszs  z SQLAlchemyDataDriver._make_joinscCsdd}g}|dkrd}|jj}n|j}|dD]0}|||||}|||j}|j}|j}q.|S)Nr) rlrmrvsplitrr>rrr)r6 join_pathrrrrrrr)r)r*_calc_joins_by_paths  z(SQLAlchemyDataDriver._calc_joins_by_pathcCs0t|jdd}t|d|jd|dS)N,z, u+ не существует у модели u, но есть: )strrr%rrQ)rrZfields_r)r)r*rsz)SQLAlchemyDataDriver._raise_invalid_fieldc# Cs|s |dfSt|dtr`g}|D]0}|j|||||d\}} | dkrHq"|| q"|||fS|ddkr|j|dd||||dS|ddkr|j|dd|||tj|dS|ddkrt|dks|dd krtd |||d }|dfS| |\} } } t t | t r | j } t t | tr8| jj } t| ttfrg} | D]N}t t |t rr| |j n*t t |tr| |jj n | |qP| } d | kr| d \}}} |j|||d }|||}|d}|j}|j}d}n|dkr|jj}d}|j| }|s2td| d|jt |tjrV| d} |j| }|dkrl|| |t |tjrN| dkrt| tst| tstd| ||| d kr| dk rtd| ||| dkr<|j|d||d}|dkrtd|d|j r|t!j"}| d krP|j#$|%|j&|k|j'|jk(}||fS|j#$|%|j&|k|j'|jk|)d|j*| (}n|+}||}|j,r|j-}|j.}n"|j/r|j.}|j-}n td||| d kr|j#$|%||k(}||fS|j#$|%||k|)d|| (}|| dkr4|n|fStd| ||nft |tj0tj1fr| dkr| 2dr|d nd!} dd| g} | d"krtd#| |||j|d||d}|dkrtd|d$|j3|ddd|d%}|j}|j}|j4d}t5|j6|} |j#$| }!|d&|g}"t7|j6d'r8|"d'd&dgg}"| rF|"| g}"|j|"||j|!|d\}!}"|!%|"}!|!(}!| 2dr|!}!||!fSt |tjrtd(|d)| |||j8|j|| ||d}|dkrtd|d*| d+||)| || fSdS),Nr) field_tableZANDrOR)cmdrorder_by)rrutОперация order_by в фильтре должна быть в формате: ["order_by", "=", [поля..]]rr)rTFu Поле u не найдено у r)rruПравое значение для IN, NOT IN для m2m должно быть list, т.е. в квадратных скобках, например "field", "IN", "[obj.id]"uНедопустимая операция над m2m полем (сравнение = только на None), допустимые IN, NOT IN, = None)rrrrr with_aliasuВ таблице u3 нет поля id для фильтрации m2mruDНадо указать для поля либо left либо rightuhНедопустимая операция над m2m полем, допустимые IN, NOT IN, = NonezNOT rr)rrulНедопустимая операция над backref полем, допустимые EXISTS, NOT EXISTSu7 нет поля id для фильтрации backref)rr cmf_deleteduФильтрация по u+ полю не поддерживаетсяu нет поля u для фильтрации)9rrcollect_filter_expr>r.or_rJrY order_queryrr=rarrrrr| rpartitionrrrrrlrmrrErrQCmfRelationBaser CmfM2MBase _query_columnZ nested_fieldsrZ RelationCachesessionrKfilterZ parent_idZ parent_fieldexistsrZchild_id m2m_model_clsrightrkrjleftrWrXr$rbackrefrrwhasattr__mro__)#r6Z in_filterrrrKrrZ filter_resr(ZexprrrZ_rightir_rrrrZ field_column m2m_modelZ m2m_subqueryZ dp_m2m_modelZdp_m2m_field_selfZdp_m2m_field_theirZ self_columnZ backref_dataZ backref_modelZ backref_tableZbackref_column_nameZbackref_columnZ sub_queryZsub_filter_expr)r)r*rsJ             "                              z'SQLAlchemyDataDriver.collect_filter_expcCs |dS)NZ _m2m_directr)rr)r)r*_m2m_depth_alias_nameVsz*SQLAlchemyDataDriver._m2m_depth_alias_namecCs |dS)NZ_m2m_idr)rr)r)r*_m2m_alias_nameZsz$SQLAlchemyDataDriver._m2m_alias_namecCsh|dg}t|tr t|}|r8t|dts8|g}|D]"\}}||jkr@||d|gq@|S)u Преобразуем обычные kwargs в продвинутый фильтр для однообразной логики фильтрования :param model: :param kwargs: :return: rrr) rErrr loadsrr0rr>)r6rr8 smart_filterr'r(r)r)r* _get_filter^s    z SQLAlchemyDataDriver._get_filtercCsl|r4|jD]$}t|tjjjr|j|jkr|}qt|j |d}|dk rh|rd| t | ddS|SdS)uДля полей "основной" таблицы не нужен префикс таблицы, для этого нужно указать with_alias=FalseNrr) rrrr.rrrrvrrwZlabelrr%)tablerZrKrZfrom_columnr)r)r*rrsz"SQLAlchemyDataDriver._query_column parent_joincs|r |jn|jj}|D]\}|j|t|d}|dk rJ||q|r|r\|jp`gn|g} fdd| D} | s|q| dj} |r|j nd} |j | || dd} | sq|| |j |||| j | dqdS) u2Соберём поля и таблицы по fflrNcs>g|]6}|jrttjtjfrttjsqSr))rrEr=rTrUr)r&Z related_modelrrr)r* s  z?SQLAlchemyDataDriver._select_joined_columns..rF)rr) rrlrmr0rboolr>Z right_modelsrrr_select_joined_columnsr)r6Zfflrr^rrrZsub_fflrrZ fields_clsZ left_modelrrr)rr*r~s4   z+SQLAlchemyDataDriver._select_joined_columnscOs(||}|}ddlm}|d|j|}t||} | rX| } | |} nht } g} g} | || | ||j | } |j | | dd} t | dkr|  d} t|| |t|dkr|j}tt|D]}|||||<q|||}|r$||||| \} }| |} | S)Nr CMF_CACHEzquery: T)rg{Gz?)rlr3rrhashrxrarHZ_cloneZ with_sessiontimerrKrrLrJrMrangerrr)r6rfull_fields_loadr7r8rrdrrFrKZstart_calc_timerr^Zclone_qpkeyrrZ filter_expr)r)r* _create_querys4       z"SQLAlchemyDataDriver._create_query)rrslicegroup_by for_updatemappercOst|t } t|}||} || } |j|fd|i| } |j| ||d} |rb| j|} |rn| } | } |j | ||d}|r| |||| r| |S)Грубо селектим всю таблицу, фильтры: - в kwargs: поле -> значение rrrr)rr get_mappermap_args map_kwargsr with_labelsrrwith_for_updateall list_to_cmf _load_m2mpost_mapping_hook)r6rrrrrrrr7r8 mapper_is_my_argsqZsql_resZres_objsr)r)r*rs6      zSQLAlchemyDataDriver.list)rrcs ||}||}|j|fd|i|}|r||fdd|D}|jj|jj|t fj | d} t |j| } g} | D]<} i} t|D]\}}| || |<q| d| d<| | q| S|jj|jjt g d} |j| }|S)rrcsg|]}jj|qSr))rmrw)r&rZrr)r*rsz.SQLAlchemyDataDriver.count..Nrcount)rrrr rlrZ select_fromZfromsZwith_only_columnsrrrrrrrb enumerater>Zscalar)r6rrrr7r8rrr^Zcount_qres_listresrowZrow_resrrZrr)rr*rs>       "zSQLAlchemyDataDriver.count)rrrrc Ost|t }t|}||}||}|j|f|d|i|} | } |j| |ddgd} |rl| } | } | r|j | ||d} |j | g||d|r| | SdS)u Выбрать один объект по фильтру: - в args - первичный ключ(опционально) - в kwargs - другие поля: поле -> значениеrrrrr)rN) rrrrrrr rr first object_to_cmfr r) r6rrrrrr7r8rr sa_instanceZres_objr)r)r*rEs     zSQLAlchemyDataDriver.getc: s|r|s dSt|tfdddd}t|}tt}tt}tt}|D]&} || | || j| qN|D]t} tt| } |D]`} || sqt| | d} | rt| t j rt| t j r|| qt| t j t j fr|| | qqz|D]\} }g}|D]8} tt| j} | j | } | r t| t j r rt| | }|jdk rt|jdk r ||jq d|_|jdkrd|_| }t| |}|j}|dkrtd|| n`t| | d}|dk r|r ||q t| | d| }t| |d}|dkrtd|| |r |dd }|||| q |D]\} }tt| d} | slqNt|j| || |d d t|gd d }|D]T}|}||D]<} rt| | }||_|jdkr||_n t| | |qqqN|r|||| |q|D]\} }|D]} | j} || }t| | } t| t j r| jrt}| jd}|d fdd|Dg}tt|| f|ii}| D]D} |}!d| j kr|ddd gg}!|j| ||!d}"| |"|| j<q|sd dlm }#m!}$|#|$j"d| jd| jdq nVt#|dkrHtt$t%|&'}%n,t|(t)j*+dd|Dd'}%tt}&|%D]}'t|'|}(|&|(|'qg})|D]} |&| }*r$t| | }|*sg|_|jdkrg|_q|,|*|| |_|jdkrt-|j|_|).|jn<|*s:t| | gq|,|*|| }t| | ||).|q|)r||)|| |q t| t j r | j/rd}+d},nd}+d},| 0}-|1|-}.t|.|+}/t|.|,}0|(|/|02|/3fdd|D}1tt}2tt}3t|1D]@\}4}5|5r|4s&q|3|5|4|2|5dd |5q|2D]\} }6tt| } t|j| || d d |6g|d d})|)r||)|| ||)D]} |3| D]t}4||4D]d}7r|7| }8|8jdkrg|8_|8j| n0t|7| d}|dkrg}t|7| ||| qƐqqqX|D]f} r| | }9|9jdkr`g|9_|9jdkrxt-|9j|9_|94nt| | ddkr.obj_idcSsdd}t|S)NcSsttSrC)rrr)r)r)r*id_to_list_factory5szWSQLAlchemyDataDriver._load_m2m..model_to_id_factory..id_to_list_factory)r)rr)r)r*model_to_id_factory4sz;SQLAlchemyDataDriver._load_m2m..model_to_id_factory.u#Не загружено id поле:rrrF)rrrrrcsg|] }|qSr)r)r&orr)r*rsz2SQLAlchemyDataDriver._load_m2m..rr)rrrzORM: skip load field rz die no related modelsrcSsi|]\}}||jqSr))r)r&r'mr)r)r*r+sz2SQLAlchemyDataDriver._load_m2m..rrjrkcsg|] }|qSr)r)r r"r)r*rs)rrrr)5rrrrr>rQrrr=rrrrZCmfBackrefBaser0rErZ_oldZ id_fieldnamersetattr partitionr r rdictr!rrrrzrrr/rrJnextitervaluesr rKr.r2rr rextendrrrlrrZ apply_changes):r6ZobjsrrrZto_load_mappingZ my_objs_by_idZmy_objs_by_model_nameZmodels_by_rel_fieldsobjrqrrrZ model_to_idZalready_loaded_obj_listrZ id_field_nameZid_fieldZid_field_valueZ field_valueZrel_model_nameZid_to_obj_listZ load_objsZload_objZ loaded_idZ models_listZobj_listZm2mqueryZbackref_field_nameZfilter_qZ field_fflZ rel_modelZmodel_query_filterrrr/rZres_dictrZm2m_idZsub_objsZ ref_obj_listZmy_id_field_nameZsub_id_field_namerrZ my_sa_fieldZ sub_sa_fieldZ m2m_queryZ id_by_modelsZmy_id_by_sub_idZmy_idZsub_idZ sub_id_listZmy_objZmy_fieldZ obj_fieldr))rrr*r s                                                        zSQLAlchemyDataDriver._load_m2mcCs6g}|D](}t|tjr&||jq||q|SrC)rrrr>r)r6r7Znew_argsargr)r)r*r,s   zSQLAlchemyDataDriver.map_argscCs:i}|D](\}}t|tjr,|j||<q |||<q |SrC)r0rrrr)r6r8Z new_kwargsr'r(r)r)r*r5s    zSQLAlchemyDataDriver.map_kwargs) no_reloadc Os||}||}|}||j}|} t} | || z|| |j | gdWnt k r} z4|j t | jttrt| | t| | W5d} ~ XYn0tk r} z|j | W5d} ~ XYnXd|_|rdS| j| |d} | d| _| S)um no_reload - не загружать объект из базы и не возвращать ZobjectsNT) cmf_entity)rrr3rlr:rr cmf_to_objectaddflushrrnrororigrlookupr rrrYis_newrr) r6instancer-r7r8r_kwargsrdrrrrrr+r)r)r*create>s2        zSQLAlchemyDataDriver.createc s||}||}|}|j}tj}|tfdd|D} || | } z| | Wn0t k r} z|j | W5d} ~ XYnXS)u*TODO логическое удалениеc3s|]}t|VqdSrCrr&fr6r)r* jsz.SQLAlchemyDataDriver.delete..N)rrr3rlr:rarMr|rKrEdeleterYrnro) r6r6r7r8rr7rdrrget_argsrrrr)r<r*r>bs     zSQLAlchemyDataDriver.delete)rc s.||}||}|}|t}tj}|tfdd|D}|||} t } | | z|j | gdWn~t k r} z0|jt| jttrt| t| W5d} ~ XYn2tk r} z|j| W5d} ~ XYnXj} | j| |d} | | | _| S)Nc3s|]}t|VqdSrCr9r:r<r)r*r=ysz.SQLAlchemyDataDriver.update..r.)r/r)rrr3rlrarMr|rKrErrr0r2rrnrorr3rr4r rrrYr5rr)r6r6rr7r8rdrrr?rrrrr5r+r)r<r*r}ss0      zSQLAlchemyDataDriver.updatecstdSrC)r,before_requestr6r9r)r*r@sz#SQLAlchemyDataDriver.before_requestcstdSrC)r,before_first_requestrAr9r)r*rBsz)SQLAlchemyDataDriver.before_first_requestcCs6ddlm}||||jdSNrr)rr cache_unlockZcache_transaction_startr3rcremover6rr)r)r*rcs   zSQLAlchemyDataDriver.commitcCs.ddlm}|||jdSrC)rrrDr3rorErFr)r)r*ros  zSQLAlchemyDataDriver.rollbackcOs|j||SrCr3rKr5r)r)r*rKszSQLAlchemyDataDriver.querycOs|j||SrCrGr5r)r)r*query_deprecatedsz%SQLAlchemyDataDriver.query_deprecatedcst|dSrC)r, make_models)r?rr9r)r*rIsz SQLAlchemyDataDriver.make_modelscCs^ttD]P}tt|}t|tr|jr(q|jr<|j|jkrr}r@rBrcrorKrHrIr4rQr __classcell__r)r)r9r*r"s     F  .   -    /   * $  $%   r"c@seZdZddZdS)SAModelAccessorcCs|dkr dSttt|S)N)Z_pytestfixturefunctionZ__test__)r"rrr)r6itemr)r)r* __getattr__szSAModelAccessor.__getattr__N)rxrgrhrZr)r)r)r*rXsrX)5rMr collectionsrrrtypingrZsqlalchemy.ext.declarativer.Zsqlalchemy.ormZsqlalchemy.sql.selectableZsqlalchemy.excrrrr Zflaskr Zalembic.configr Zalembicr Zpsycopg2.errorcodesr Zpsycopg2rrZcmf.fields.base_fieldsrrZcmfrrrZ base_errorrrrZmodels.base_modelrrrrbaserrrr Zutil.immutablesr!r"rXr)r)r)r*sB            D