j06NddlmZddlZddlmZddlmZmZmZmZddlm Z m Z m Z ddl m Z ejdrddl mZeZndd l mZGd d eZGd d eZGddZGddejejjZdS))contextmanagerN)persist)ColumnIntegerStringBoolean) create_engineor_not_) sessionmakerz1.)declarative_base)DeclarativeBaseceZdZdS)BaseN)__name__ __module__ __qualname__k/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/casbin_sqlalchemy_adapter/adapter.pyrrs rrcPeZdZdZeedZeedZeedZ eedZ eedZ eedZ eedZ eedZdZdZdS) CasbinRule casbin_ruleT) primary_keyc|jg}|j|j|j|j|j|jfD]}|n||d|S)Nz, ) ptypev0v1v2v3v4v5appendjoin)selfarrvs r__str__zCasbinRule.__str__"s^zl'47DGTWdgtwG  Ay JJqMMMMyy~~rcRd|jt|S)Nz)formatidstrr&s r__repr__zCasbinRule.__repr__*s &--dgs4yyAAArN)rrr __tablename__rrr,rrrrr r!r"r#r)r/rrrrrs!M T * * *B F66#;;  E s  B s  B s  B s  B s  B s  BBBBBBrrc*eZdZgZgZgZgZgZgZgZ dS)FilterN) rrrrrrr r!r"r#rrrr2r2.s1 E B B B B B BBBrr2c heZdZdZ ddZedZdZdZdd Z d Z d d Z d Z dZ dZdZdZdZdededeedeed df dZdededeeedeeed df dZdeeed eeefdZd eeefdZdZdS)!Adapterz"the interface for Casbin adapters.NFTczt|trt||_n||_d|_|t }n|it|jtsOdttd}|dtt |jdz }t|||_dD]$}t||st|d%|j t_ ||_ t|j|_|r$tj |j||_dS)Nz6The type of db_class_softdelete_attribute needs to be z. zAn attribute of type z was given.)r,rrrr r!r"r#z# not found in custom DatabaseClass.)bind) isinstancer-r _enginesoftdelete_attributertyper ValueErrorhasattr Exceptionmetadatar _db_classr session_local create_all _filtered)r&enginedb_classdb_class_softdelete_attributefilteredcreate_all_modelsmsgattrs r__init__zAdapter.__init__;sP fc " " "(00DLL!DL$(!  !HH,8-2GBB8bsSZ||aaaks48U8Z3[3[/\/\kkkk oo%(ED %  R Rx..R#t$P$P$PQQQR$-DM!)t|<<<  3 M $ $T\ 2 2 2!rc#K|} |V|n(#t$r}||d}~wwxYw |dS#|wxYw)zz'Adapter.save_policy..sWWWG7CVCVCVCVr)r9rQrTr?deleterXkeysitemspolicyrrrUrVrbrrmrgr+countrrr r!r"r# has_policyrnname)r&rXrOrTsecrastrplines_before_changesraindexvaluev_valuerZfields_with_Nones r save_policyzAdapter.save_policys  $ ,$$&& Q' dn55 %QQC%+"2"2"4"444 &+k#&6&<&<&>&>QQ s$'JQQD 225$2PPPPQQQ Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q4 " "$ HgMM$.11E**511E$)99;; " Q Qek..0000"'+c"2"8"8":": Q QJE3 # QQ',||DN4HE4Q'R'R ,5dOOQQLE5&-dnell5>Q>Q&R&RG+7+>+>w%?O+P+PLL'--//144 225$2PPPQ Q- H H AhGGGGGG $ XW/?WWW''UD99HD$";"@$GGG H-$ H$ H$ H$ H$ H$ H$ H$ H$ H$ H$ H$ H$ H$ H$ HLts%BCCC!F5J##J'*J'c2|||dS)z"adds a policy rule to the storage.Nrr)r&rrrps r add_policyzAdapter.add_policys ud+++++rc<|D]}|||dS)z#adds a policy rules to the storage.Nr)r&rrrulesrps r add_policieszAdapter.add_policiess4 0 0D  " "5$ / / / / 0 0rc *|5}||j}||}||jj|k}t |D]D\}}|t|jd||k}E|j | }n| |j di}dddn #1swxYwY|dkrdndS)z'removes a policy rule from the storage.rlNTrF) rQrTr?rUrbrrmrgr+r9rzupdate) r&rrrprOrTrqr(rs r remove_policyzAdapter.remove_policysF  " " DgMM$.11E**511ELL!5!>??E!$ T T1 WT^U\\!__%M%MQR%RSS(0LLNNLL$";T!BCC D D D D D D D D D D D D D D D1uutt%'sCDDDc|sdS5}|j}|}|jj|k}t |}t|D]4\}|tfd|D}5j | n| j diddddS#1swxYwYdS)z%remove policy rules from the storage.Nc3pK|]0}tjd|kV1dS)rlN)rgr?r+)rwr(rqr&s r z*Adapter.remove_policies..s>TT! Q@@AETTTTTTrT) rQrTr?rUrbrziprmr r9rzr)r&rrrrOrTrprqs` @rremove_policieszAdapter.remove_policiessh  F  " " @gMM$.11E**511ELL!5!>??EKE$U++  4 TTTTTtTTTTT(0  d7>??? @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @sCDDDc |5}||j}||}||jj|k}d|cxkrdksn ddddSd|t |zcxkrdksn ddddSt|D]O\}}|dkrDt|jd ||z} || |k}P|j | } n| |j d i} dddn #1swxYwY| dkrd ndS) ztremoves policy rules that match the filter from the storage. This is part of the Auto-Save feature. rNFrlT) rQrTr?rUrbrrfrmrgr+r9rzr) r&rr field_index field_valuesrOrTrqr(rrs rremove_filtered_policyzAdapter.remove_filtered_policys " " DgMM$.11E**511ELL!5!>??E))))))))  D D D D D D D Ds<'8'88====A==== D D D D D D D D",// 7 7177%dnell;QR?6S6STTG!LLA66E(0LLNNLL$";T!BCC# D D D D D D D D D D D D D D D&1uutt%'sA#E E2BEEErrold_rulenew_rulec |5}||j}||}||jj|k}t |D]F\}}t|jd|} || |k}Gt|t|kr|n|} | } tt| D]?}|t|krtd|d|d,td|d@ ddddS#1swxYwYdS)a+ Update the old_rule with the new_rule in the database (storage). :param sec: section type :param ptype: policy type :param old_rule: the old rule that needs to be modified :param new_rule: the new rule to replace the old rule :return: None rlzold_rule_line.vz = new_rule[]z = NoneN) rQrTr?rUrbrrmrgr+rfonerangeexec) r&rrrrrOrTrrr longest_rule old_rule_lines r update_policyzAdapter.update_policy!s " " ;gMM$.11E**511ELL!5!>??E!*( 3 3 7 7 u!$.%,,u2E2EFF W%566(+8}}s8}}'D'D88(L!IIKKMs<0011 ; ;3x==((F5FFeFFFGGGG95999::::  ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;sD=E  E$'E$ old_rules new_rulesctt|D]&}|||||||'dS)a1 Update the old_rules with the new_rules in the database (storage). :param sec: section type :param ptype: policy type :param old_rules: the old rules that need to be modified :param new_rules: the new rules to replace the old rules :return: None N)rrfr)r&rrrrrqs rupdate_policieszAdapter.update_policiesDsT"s9~~&& G GA   sE9Q<1 F F F F G Grct}||_tt|D]<}||kr4||t|zkrt |d||||z <|||dS)zMupdate_filtered_policies updates all the policies on the basis of the filter.r(N)r2rrrfrn_update_filtered_policies)r&rrrrrrbrqs rupdate_filtered_policiesz Adapter.update_filtered_policiesXs  s<(())  AaA c,6G6G(G$G$GAa+o)FGGGG &&y&99999rc|5}||j}||}||jj|jk}|||}|}|d|j|| d|j||cdddS#1swxYwYdS)zN_update_filtered_policies updates all the policies on the basis of the filter.rtN) rQrTr?rUrbrrarVrr)r&rrbrOrTfiltered_queryrs rrz!Adapter._update_filtered_policiesis " " gMM$.11E**511ELL!5!EFFE!..uf==N&**,,I  flI > > >   c6< ; ; ;'                  sB;CC!$C!cf|}|j'|t|j}|Sr])r9wherer )r&rTquery_softdeletes rrUzAdapter._softdelete_querys6   $ 0/55d4;T6U6UVV r)NNFT)r_Nr])rrr__doc__rJrrQr[r^rdrarrrrrrrrr-listrrrrrUrrrr4r48s,, &* ."."."."`  ^ ;;; " " " "333""""888t,,,000 ((( @@@&(((2!;!;"!;.23i!;CG9!; !;!;!;!;FG GGS ? G S ? G  GGGG(:%)$s)_: d3i::::"d49o0     rr4) contextlibr sqlalchemycasbinrrrrrr r r sqlalchemy.ormr __version__ startswithr rrrr2r4adapters UpdateAdapterrrrrs%%%%%%777777777777//////////''''''$$T**  //////    DD/.....        BBBBBBBB0        M M M M M gow/=M M M M M r