@OOfPddlmZddlZddlZddlZddlZddlZddlmZddlm Z ddlm Z ddl m Z ddl mZddlmZdd lmZdd lmZddlZdd lmZdd lmZdd lmZ eZGddZe GddZddZddZGddZ d dZ!d!d"dZ"dZ#d!d#dZ$d$dZ% d% d&dZ&d%d'dZ'd(dZ(y))) annotationsN)Callable)Iterable)Sequence) dataclass)Path)Any)Optional)TextIO) exceptions)signals) typecheckcteZdZdZ d dZdZedZd dZddZ ddZ ddZ dd Z d Z y )_Option)nametypespecvalue_defaultchoiceshelpctj|||||_||_||_t |_tj|jjdd|_ ||_ y)N  ) rcheck_option_typerrrunsetrtextwrapdedentstripreplacerrselfrrdefaultrrs U/var/www/premiumrankchecker/venv/lib/python3.12/site-packages/mitmproxy/optmanager.py__init__z_Option.__init__!s_ ##D'8<     OOD)//199$D  cB|jd|jdS)Nz [])currentrr!s r#__repr__z_Option.__repr__1s ,,.!DMM?!44r%c@tj|jSN)copydeepcopyrr)s r#r"z_Option.default4s}}T]]++r%c|jtur |j}n |j}tj|Sr,)rrr"r-r.)r!vs r#r(z_Option.current8s/ ::  A A}}Qr%chtj|j||j||_yr,)rrrrr)r!rs r#setz _Option.set?s###DIIudmmD r%ct|_yr,)rrr)s r#resetz _Option.resetCs  r%c<|j|jk7Sr,)r(r"r)s r# has_changedz_Option.has_changedFs||~--r%c\|jD]}t||t||k7syy)NFT) __slots__getattr)r!otheris r#__eq__z_Option.__eq__Is3 AtQ75!#44 r%ct|j|j|j|j|j }|j r|j|_|Sr,) rrrr"rrr6r(r)r!_os r# __deepcopy__z_Option.__deepcopy__OsF DIIt}}dllDIIt|| T    llnAGr%N rstrrz type | objectr"r rrBrzSequence[str] | NonereturnNone)rCr )rr rCrDrCrD)rCbool)__name__ __module__ __qualname__r8r$r*propertyr"r(r2r4r6r<r@r%r#rrs}LI     &   5,, . r%rceZdZUded<y)_UnconvertedStrings list[str]valN)rGrHrI__annotations__rKr%r#rMrMVs Nr%rMcyr,rKupdateds r#_sig_changed_specrT[r%cyr,rKexcs r#_sig_errored_specrY_rUr%ceZdZdZddZ d d dZejd!dZdZ ddZ dZ dd Z e Z d Zfd Zd Zd ZdZdZdZdZdZdZdZd"dZdZdZdZddd#dZddZd$dZd%dZ xZ!S)& OptManagera OptManager is the base class from which Options objects are derived. .changed is a Signal that triggers whenever options are updated. If any handler in the chain raises an exceptions.OptionsError exception, all changes are rolled back, the exception is suppressed, and the .errored signal is notified. Optmanager always returns a deep copy of options to ensure that mutation doesn't change the option state inadvertently. ci|_tjt|_|jj |j tjt|_g|_ i|_ yr,) deferredr SyncSignalrTchangedconnect_notify_subscribersrYerrored_subscriptions_optionsr)s r#r$zOptManager.__init__psX(* ))*;<  T556))*;< LN)+ r%cvt||||||j|<|jj|hy)NrR)rrdr_sendr s r# add_optionzOptManager.add_optionzs6&dHgtWM d 4&)r%Fc#*Ktj|j} dy#tj$rU}|j j |||jd<|jj ||r|Yd}~yd}~wwxYww)NrWrdrR) r-r.rdr OptionsErrorrbrf__dict__r_)r!rSreraiseoldes r#rollbackzOptManager.rollbackszmmDMM*  &&  LL  !  $(+DMM* % LL  g  .  s+ B(BBA B B BBc|D](}||jvstjd|z|jj t j |t|fy)a) Subscribe a callable to the .changed signal, but only for a specified list of options. The callable should accept arguments (options, updated), and may raise an OptionsError. The event will automatically be unsubscribed if the callable goes out of scope. No such option: %sN)rdr rircappendr make_weak_refr2)r!funcoptsr;s r# subscribezOptManager.subscribesb HA % --.BQ.FGG H ""G$9$9$$?T#KLr%cd}|jD] \}}|}|||zs|||d}"|r7|jDcgc]\}}|||fc}}|jd<yycc}}w)NFTrc)rcrj)r!rScleanuprefrtcallbacks r#razOptManager._notify_subscriberss,, ICuH#'>T7+   /3/B/B/ +dceFWd /DMM* + /s A,A,cVt|tr|j|jk(Sy)NF) isinstancer[rd)r!r:s r#r<zOptManager.__eq__s" eZ (==ENN2 2r%ctt}tj|j||jd<|S)Nrd)r[r-r.rdrj)r!memodictr?s r#r@zOptManager.__deepcopy__s+ L!%t}}h!G :r%ct||jvr|j|jStd|z)Nrp)rdr(AttributeError)r!attrs r# __getattr__zOptManager.__getattr__s7 4== ==&..0 0 !5!<= =r%c|jjd}|st| ||y|jdi||iy)NrdrK)rjgetsuper __setattr__update)r!rrrt __class__s r#rzOptManager.__setattr__s@}}  , G e , DKK (4- (r%cHt|jjSr,)r2rdkeysr)s r#rzOptManager.keyss4==%%'((r%c6|jjSr,)rditemsr)s r#rzOptManager.itemss}}""$$r%c||jvSr,)rd)r!ks r# __contains__zOptManager.__contains__sDMM!!r%c|jjD]}|j|jj t |jj y)z3 Restore defaults for all options. rRN)rdvaluesr4r_rfr2r)r!r?s r#r4zOptManager.resetsP%%' A GGI  #dmm&8&8&:";.setters D$ "r%)rdr)r!rrs`` r#rzOptManager.setters. t}} $/$67 7 # r%cjvrtdzj}|jtk7r t dfd}|S)zv Generate a toggler for a boolean attribute. This returns a callable that takes no arguments. rpz-Toggler can only be used with boolean optionsc6tt yr,)rr9)rr!sr#togglez"OptManager.toggler..toggles D$GD$$7 7 8r%)rdrrrF ValueError)r!rr?rs`` r#togglerzOptManager.togglersT t}} $/$67 7 MM$  :: LM M 9 r%c4|j|jSr,)rdr"r!options r#r"zOptManager.defaults}}V$,,,r%c<|j|jS)z: Has the option changed from the default? )rdr6rs r#r6zOptManager.has_changeds}}V$0022r%ci}|jD]6\}}| t|ttfrt |||z||<2|||<8|j di|y)z Merge a dict of options into this object. Options that have None value are ignored. Lists and tuples are appended to the current option value. NrK)rr{listtupler9r)r!rttosetrr0s r#mergezOptManager.mergesf JJL !DAq}a$/&tQ/!3E!H E!H  !  er%ctj|jdjd}d|vrd|zdz}dj t |j t |j|S)N)indentz {}rz z{mod}.{cls}({{{options}}}))modclsoptions)pprintpformatrdrformattyperHrG)r!rs r#r*zOptManager.__repr__.sj..q9??F 7?(4/G+22T %%4:+>+>3  r%)deferc i}|D]P}d|vr8|jdd\}}|j|gj|?|j|gRi}t|j D]B}||j vs|j |j ||j|||<D|rG|jj|jD cic]\}} |t| c} }n)|r'tjddj||jdi|ycc} }w)ae Takes a list of set specification in standard form (option=value). Options that are known are updated immediately. If defer is true, options that are not known are deferred, and will be set once they are added. May raise an `OptionsError` if a value is malformed or an option is unknown and defer is False. =)maxsplitzUnknown option(s): rNrK)split setdefaultrqrrrd _parse_setvalpopr]rrrMr rir) r!rspecs unprocessedspecrr processedrr0s r#r2zOptManager.set6s=-/  1Dd{"jjqj9 e&&tR077>&&tR0  1%' ))+, Dt}}$"&"4"4MM$')># $   MM 7B7H7H7JKtq!'**K ))%dii &<%=>    i Ls"D= cNi}|jjD]R\}}||jvst|tr)|j |j||j }|||<T|jdi||jD]}|j|=y)zs Processes options that were deferred in previous calls to set, and have since been added. NrK) r]rrdr{rMrrOrr)r!roptnamerrs r#process_deferredzOptManager.process_deferred]s "$"mm113 (NGU$--'e%89 ..t}}W/EuyyQE"'w  (  f !A a  !r%c<|jttk(r|St|dkDr%t j d|j d||r|d}nd}|jtttfvr9|jtk(r$|"t j d|j |S|jtttfvrD|r t|S|jtk(r"t j d|j y|jtk(r9|dk(r|j S|r|d k(ry |d k(ry t j d td|j#t$rt j d|wxYw)zN Convert a string to a value appropriate for the option type. rzReceived multiple values for : rNzOption is required: zNot an integer: rtrueTfalseFzRBoolean must be "true", "false", or have the value omitted (a synonym for "true").zUnsupported option type: ) rrrBlenr rirr intrrFr(NotImplementedError)r!r?roptstrs r#rzOptManager._parse_setvalls ::# &M v;?))/xr&B  AYFF ::#x}- -zzS V^ --0DQVVH.MNNM ZZC#/ /Ov;&s" --0DQVVH.MNN ZZ4 !99;&Vv-7" --h"$=ajj\"JKK#"O$114DVH2MNNOs  E99"Fc ||jvry|j|}d}|||}|jtk(r|jd}||d} |d|zd} |r"|jr |d|z|} n |||} |j | d|d|j | d||j d |jdi|diy|jtttfvr%|j |d t||j |d y|jtttfvr0|j |d t||j ||jd y|jttk(r3|j |d t||j dz||jd ytd|j)z Auto-Create a command-line parser entry for a named option. If the option does not exist, it is ignored. Ncb|jdd}d|zg}|r|jd|z|S)Nr>-z--%s)rrq)rsfs r#mkfz#OptManager.make_parser..mkfs5 #s#A! Aq!Hr%F)requiredzno- store_false)actiondest store_true)rrrstore)rrrrmetavar)rrrrrrrqz May be passed multiple times.zUnsupported option type: %srK)rdrrFadd_mutually_exclusive_groupr" add_argumentr set_defaultsrr rBrrr) r!parserrrshortr?rflagsgonfofffs r# make_parserzOptManager.make_parsers $-- '  MM' " GU# :: 33U3CAgt$Cuw-D99uw6Dgu-C ANN$  ANNC 7 P F   27D/ 2 ZZC#/ / F  VV  ZZC#/ / F  VV  ZZ8C= ( F  VV>>  :AJJG Gr%rEr,rAF)rrBrCr )rrBrrFrCrD)r?rrrNrCr )NN)"rGrHrI__doc__r$rg contextlibcontextmanagerrnrurar<r@__copy__rrrrrr4rrrrrr"r6rr*r2rrr __classcell__)rs@r#r[r[cs + )- * *  * *  * & *  *   M   H> ))%"=&&H  -3  .3%!N !*LXBHr%r[ctjjj}t |j D]}|j |}|j||<|jj}|jr)|ddjd|jDzz }n'tj|j}|d|zz }djtj |}|j#|d|ztjj%j'||S)z3 Dumps an annotated file with all options. z Valid values are %s.rc32K|]}t|ywr,)repr).0cs r# z dump_defaults..s6R1tAw6Rsz Type %s.r)before)ruamelyamlcomments CommentedMapsortedrrdr"rrrrrtypespec_to_strrrwrap!yaml_set_comment_before_after_keyYAMLdump)rtoutrrr?txtts r# dump_defaultsrs  ))+A DIIK B MM! yy!fflln 99 *TYY6R 6R-RR RC))!**5A ;? "Cii c*+ ++AdSj+A B ;;    " "1c **r%ci}||j}t|D]i}|j|}tj|j }||j |j|j|jd}|||<k|S)z Dumps the options into a list of dict object. Return: A list like: { "anticache": { type: "bool", default: false, value: true, help: "help text"} } )rr"rrr) rrrdrrrr"r(rr)rtr options_dictrr?rrs r# dump_dictsrs L |yy{ D\ ! MM!   % %ajj 1yyYY[FFyy  ! Q ! r%c |siS tjjdd}|j|}t|trtjd |iS|S#tjjj $r}t |drZ|jj}tjd|jjdzd|dt|d d tjd d}~wwxYw) NsafeT)typpure problem_markzConfig error at line rz: rproblemzCould not parse options.zConfig error - no keys found.)rrrloaderror YAMLErrorhasattrr  get_snippetr riliner9r{rB)textrdatar0snips r#parser s   F{{F6yy$%%&EFF  K ;;   & &F 1n %>>--/D))>>&&*D'!Y2KM  ))*DE EFs2A##'D  A;DD c t|}|jd}|;|9|Dcgc]*}ttt |t |,c}|d<|j di|ycc}w)z Load configuration from text, over-writing options already set in this object. May raise OptionsError if the config file is invalid. scriptsN) relative_torK)rrrB relative_pathrr)rtrcwdrrpaths r#rr sl ;Dhhy!GsNU FJC d4jd3i@ A YD  s/A-c  |D]}t|j}|js-|js>|j d5} |j } ddd t||jjy#t $r }tjd|d|d}~wwxYw#1swY_xYw#tj$r }tjd|d|d}~wwxYw)z Load paths in order. Each path takes precedence over the previous path. Paths that don't exist are ignored, errors raise an OptionsError. utf8encodingzError reading rN)r) r expanduserexistsis_fileopenreadUnicodeDecodeErrorr rirabsoluteparent)rtpathsprrrms r# load_pathsr,0s  I G    88:!))+( MAM&&(C M  IT3AJJL$7$78 I *M$11N1#Rs2KLLM M M** I --qcA3.GHH IsBCB/&C C!B<<CCC D#C>>DcFt|}|jD]%}|s|j|st||||<'t |jD]}||j vs||=t jjj||y)ar Performs a round-trip serialization. If text is not None, it is treated as a previous serialization that should be modified in-place. - If "defaults" is False, only options with non-default values are serialized. Default values in text are preserved. - Unknown options in text are removed. - Raises OptionsError if text is invalid. N) rrr6r9rrdrrrr)rtfilerdefaultsrrs r# serializer0Ds ;D YY[' t''*dA&DG'$))+  DMM !Q KKD$'r%ct|j}|jr>|jr.|j d5} |j } dddnd}|j dd5}t|||dddy#t $r }tjd|d|d}~wwxYw#1swY`xYw#1swYyxYw)z Save to path. If the destination file exists, modify it in-place. Raises OptionsError if the existing data is corrupt. rr zError trying to modify rNr w) rr"r#r$r%r&r'r rir0)rtrr/rrrms r#saver3\s : "D {{} YYY ' U1 Uvvx U U  3 (+A$4*++ & U --0GvRPQs.STT U U U++s6 CB=C  B>B99B>>CC  Cct|}|j|k7r.|js|jj}||jz jS)z Make relative paths found in config files relative to said config file, instead of relative to where the command is ran. )rr" is_absoluter() script_pathrs r#rrps^ {#K;.{7N7N7P!,,.779 +002 2 < < >>r%)rSzset[str]rCrD)rX ExceptionrCrD)rr r,)rzIterable[str] | NonerCdict)rtr[rrBrzPath | str | NonerCrD)rtr[r* Path | strrCrDr) rtr[r.r rrBr/rFrCrD)rtr[rr9r/rFrCrD)r6r9rr9rCr)) __future__rrr-rrweakrefcollections.abcrrr dataclassesrpathlibrtypingr r r ruamel.yamlr mitmproxyr mitmproxy.utilsr robjectrrrMrTrYr[rrrrr,r0r3rrKr%r#rDs" $$$! #% 55p  wHwHt +,.,  I*AF( ("(*-(9=( (0+( ?r%