o jD @shddlmZddlZddlZddlmZddlZddlmZddlm Z ddl m Z zd?Z6e7d@kr2e1dSdS)F) annotationsN)partial)ClickException) current_app)with_appcontext)Consolegroup)Live)Pretty)Confirm)Table)Tree)get_filtered_endpoint limiter_theme render_func render_limit render_limitsrender_storagerender_strategyTF) TypedDict)Rule)Limiter) RuntimeLimit)Callable Generator) ConfigVars HeaderNamesz,Flask-Limiter maintenance & utility commands)helpreturnNonecCststddS)NziMissing dependencies for flask-limiter cli. Please install the cli extra (pip install flask-limiter[cli]))CLI_DEPS_AVAILABLErr"r"K/var/www/ai-app/venv/lib/python3.10/site-packages/flask_limiter/commands.pycli*s r$z View the extension configurationc Cstttd}ttjdt}|ot|d}|rtdd}| d| d| d| d t j t |j| d t jt|j| d t jt |jtt j}tt|j}|t j}|d |d |t j|d|t|| d|||jjr| dt jt dd|jjD| dt jt |j | dt j!t|j"| dt j#t|j$| dt j%t |j&n | dt jt g|jj'r| dt j(t dd|jj'D| dt j)t |j*| dt j+t|j,| dt j-t|j.| dt j/t|j0n | dt j(t g|j1r:| dt j2t ddt3j4|j1D|j5rtt j6}|t j7|t j8|t j9|t j:tt |j5}|t |j;t|t |j;tSzconfig..zApplicationLimits LimitszDefault LimitscSr1r"r2r3r"r"r#r6tr7z Meta LimitscSr1r"r2r3r"r"r#r6r7zHeader configurationFzFail on first breachzOn breach callback(No Flask-Limiter extension installed on bold redstyle)Frtest_request_contextrrlist extensionsgetsetr add_columnadd_rowrENABLEDr enabledKEY_FUNCr _key_func KEY_PREFIX _key_prefixr STRATEGYrr'add STORAGE_URISTORAGE_OPTIONSr limit_managerapplication_limitsAPPLICATION_LIMITSAPPLICATION_LIMITS_PER_METHOD_application_limits_per_methodAPPLICATION_LIMITS_EXEMPT_WHEN_application_limits_exempt_whenAPPLICATION_LIMITS_DEDUCT_WHEN_application_limits_deduct_whenAPPLICATION_LIMITS_COST_application_limits_costdefault_limitsDEFAULT_LIMITSDEFAULT_LIMITS_PER_METHOD_default_limits_per_methodDEFAULT_LIMITS_EXEMPT_WHEN_default_limits_exempt_whenDEFAULT_LIMITS_DEDUCT_WHEN_default_limits_deduct_whenDEFAULT_LIMITS_COST_default_limits_cost _meta_limits META_LIMITS itertoolschain_headers_enabledHEADERS_ENABLED HEADER_RESETHEADER_REMAININGHEADER_RETRY_AFTERHEADER_RETRY_AFTER_VALUE_header_mappingrRESET REMAINING RETRY_AFTER _retry_afterFAIL_ON_FIRST_BREACH_fail_on_first_breach ON_BREACH _on_breachprint) consolelimitersr'extension_detailslimiter_configlimiter_config_valuesnodeheader_configs header_valuesr"r"r#config2s                     $r~z3Enumerate details about all routes with rate limitsz --endpointzEndpoint to filter by)defaultrz--pathzPath to filter byz--methodzHTTP Method to filter byz--keyzTest the limitz--watch/--no-watchzCreate a live dashboardGETendpoint str | Nonepathmethodstrkeywatchboolcsttjdt}|rt|dndttd}rj}it t|}t tj |dddD]^} | j } | dkrAq7t| dd krvd| ddd } | gttt|t| | | | | |t| | d q7d gttt|t| d | d| |t| dd q7tdfdd } |s|| nMt| |ddd!} z | | tdWn tyYnwqWdn1swYn|jdtddWddSWddSWddS1swYdS)Nr'rr%cSt|SNrrr"r"r#zlimits..rstatic.r)exemption_scopertestrootrrc3sr-jjr-s-s-ttttjjgfddVttjjgfddVD]&}|dkr?tdtj dntd|dfd d |DVq/dS) Nz[gold3]Meta Limits[/gold3])rrlabelz![gold3]Application Limits[/gold3]rz[gold3]z[/gold3]z[blue]z[/blue]csg|]}|qSr")rJ)r4 renderable group_treer"r#r6sz6limits..console_renderable..) rMrNrrr=rdrerbr name)rrgroupsrr'rrrr#console_renderables2 z"limits..console_renderableg?T)rvrefresh_per_secondscreenr8r9r:)rr)rr<r>r?r@r=rrrMrsortedurl_map iter_rulesrlensplitjoin setdefaultappendrrresolve_limitsrrrur updatetimesleepKeyboardInterrupt)rrrrrrwrvmanagerfilter_endpointrule rule_endpoint bp_fullnamerliver"rr#limitss          O$rzClear limits for a specific keyzKey to reset the limits for)rrequiredrz-yzSkip prompt for confirmation)is_flagryc Cstbttjdt}|r|dnd}ttd}|rM|j}t t||||} Gdddt } i} t tj | dddD]:} | j} | d krKqAt| d d krnd | d dd }| | |t| |d | | <qA| | |t| dd | | <qAd}| s|jj}|s|r|tt||gfd|d| D]\}}|dr|tt||d||d|dq|std|drL|rtd}|D]}|j|j||d||t|dq||| D]`\}}|drKt|}|d\}}||D]C}|j r-|dr-|dj!r-|s-|dj!D]}|j|j||||qn |j|j|||||t|dq||qn|jdtddWddSWddS1skwYdS)Nr'rr%c@seZdZUded<ded<dS)zclear..Detailsrrztuple[list[RuntimeLimit], ...]rN)__name__ __module__ __qualname____annotations__r"r"r"r#DetailsFs  rcSrrrrr"r"r#rLrzclear..rrrrr)rrrr0)rrrr)rrz/Proceed with resetting limits for key: [danger]z [/danger]?z: [success]Cleared[/success]r8r9r:)"rr<r=r>r?r@rrrMrrrrrrrrrrrNruritemsr askr r'clearr5 scope_forrJr per_methodmethods)rrrrrrwr'rvrrr rule_limitsrrrrNdetailsr{r5r decorated rule_methodr"r"r#r0s                 $r__main__)rr )NNrNF) rrrrrrrrrrrr )NNrF) rrrrrrrrrrrr )8 __future__rrdr functoolsrclickrflaskr flask.clir rich.consolerr rich.liver rich.prettyr rich.promptr rich.tabler rich.treer _clirrrrrrrr! ImportErrortyping_extensionsrwerkzeug.routingr _extensionr_limitsr_typingrr constantsrrr$commandr~optionrrrr"r"r"r#sx          $          k k