o tBh@s0ddlZddZ  d deded efd d ZdS) NcCs|r |r dt|S|S)N)float) rate_limituse_rate_limit_headerthrottle_by_secondsrq/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/sp_api/util/load_all_pages.pymake_sleep_times r  NextTokenFrrextrascssifdd}|S)a Load all pages if a next token is returned Args: throttle_by_seconds: float next_token_param: str | The param amazon expects to hold the next token use_rate_limit_header: if the function should try to use amazon's rate limit header extras: additional data to be sent with NextToken, e.g `dict(QueryType='NEXT_TOKEN')` for `FulfillmentInbound` Returns: Transforms the function in a generator, returning all pages cs"fddj_S)Nc?sz|i|}|V|jr9|||jit|j}|i|D]}|V|dkr8t|q*dSdS)Nr) next_tokenclearupdater rtimesleep)argskwargsres sleep_timex)r functionnext_token_paramrrwrapperrrrs z2load_all_pages..decorator..wrapper)__doc__)rr rrr)rrr decorators z!load_all_pages..decoratorr)rrrr rrrrload_all_pages s r)r r FN)rr rbooldictrrrrrs