o $i& @sddlZddlmZmZddlmZmZmZmZddlZ ddl m Z ddl m Z ddlmZmZmZmZmZmZmZmZmZmZmZddlmZddlmZd e d ed efd d Zdeegeefde j j!fddZ"de j j!deeeeeffddZ#de j j!defddZ$de j j!defddZ%deegefde j j!fddZ&deeeeefdedeeeeeffddZ'd ee(d!eeeeefd"ed#e)dee(f d$d%Z*dS)&N)asdictfields) AwaitableCallableListTuple rest_response)HTTPStatusCode) DEFAULT_LIMITDEFAULT_RPC_TIMEOUTRAY_MAX_LIMIT_FROM_API_SERVERListApiOptionsListApiResponse PredicateType StateSchemaSummaryApiOptionsSummaryApiResponseSupportedFilterType filter_fields)DataSourceUnavailable)convert_string_to_type status_code error_messageresultcKstd|||dd|S)NF)rmessagerconvert_google_styler)rrrkwargsrrZ/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/dashboard/state_api_utils.pydo_replysr list_api_fnreqc sz|t|dIdH}ttjdt|dWSty2}zttjt|ddWYd}~Sd}~wtyM}zttj t|ddWYd}~Sd}~wwN)option)rrr) options_from_reqr r OKr ValueError BAD_REQUESTstrrINTERNAL_ERROR)r!r"rerrrhandle_list_api&s.r-returncCsn|jdg}|jdg}|jdg}t|t|ksJg}t|||D] \}}}||||fq'|S)N filter_keysfilter_predicates filter_values)querygetalllenzipappend)r"r/r0r1filterskey predicatevalrrr_get_filters_from_req?sr;cCst|jddur|jdnt}|tkr$td|d|dtdt|jdd}t|}t|jd d t}t|jd d t}t |||||d S)z1Obtain `ListApiOptions` from the aiohttp request.limitNz Given limit z# exceeds the supported Given limit z exceeds the supported limit zg. Use a lower limit, or set the `RAY_MAX_LIMIT_FROM_API_SERVER` environment variable to a larger value.timeoutdetailFexclude_driverT)r<r=r7r?r@) intr2getr r r(r;rboolr)r"r<r=r7r?r@rrrr&Ls,r&cCs6t|jdt}t|}|jdd}t|||dS)Nr= summary_by)r=r7rD)rAr2rBr r;r)r"r=r7rDrrrsummary_options_from_reqhsrE summary_fncs*|t|dIdH}ttjdt|dSr#)rEr r r'r)rFr"rrrrhandle_summary_apiosrGfilterschemacCspg}t|rddt|D}n|}|D]\}}}||vr||}zt||Wn ty4Ynywt||r;nr|tusC|dkr`zt|t}Wnbty_td|d|d|dw|t ush|dkrzt|t }Wn=tytd|d|d |d w|t us|d krzt|t }Wntytd|d|d |d |d w| |||fq|S)aConvert the given filter's type to SupportedFilterType. This method is necessary because click can only accept a single type for its tuple (which is string in this case). Args: filter: A list of filter which is a tuple of (key, val). schema: The state schema. It is used to infer the type of the column for filter. Returns: A new list of filters with correct types that match the schema. cSsi|]}|j|jqSr)nametype).0fieldrrr sz(convert_filters_type..integerzInvalid filter `--filter  zD` for a int type column. Please provide an integer filter `--filter z [int]`numberzF` for a float type column. Please provide an integer filter `--filter z [float]`booleanz6` for a boolean type column. Please provide `--filter z& [True|true|1]` for True or `--filter z [False|false|0]` for False.) dataclasses is_dataclassr schema_dict isinstance TypeErrorrArr(floatrCr6)rHrI new_filtercolr9r: column_typerrrconvert_filters_type{sf      r\datar7state_dataclassr?c Csxt||}g}|D]}d}|D]\}}} |} |}|| vr*td|d| ||vr1d}nx|dkrt| trLt||trL||| k}n]t| trbt||trb||t| tk}nGt| trxt||trx||t| tk}n1||| k}n*|dkrt| trt||tr||| k}n||| k}ntd|d|snq|r| t |||q |S) aReturn the filtered data given filters. Args: data: A list of state data. filters: A list of KV tuple to filter data (key, val). The data is filtered if data[key] != val. state_dataclass: The state schema. Returns: A list of filtered state data in dictionary. Each state data's unnecessary columns are filtered by the given state_dataclass schema. TzThe given filter column un is not supported. Enter filters with –-filter key=value or –-filter key!=value Supported filter columns: F=z!=zUnsupported filter predicate z' is given. Available predicates: =, !=.) r\filterable_columnslowerr(rVr*rCrrAr6r) r]r7r^r?rdatummatch filter_columnfilter_predicate filter_valuer`rrr do_filtersb      rg)+rSrrtypingrrrr aiohttp.webaiohttpray.dashboard.optional_utilsr ray.dashboard.utilsr ray.util.state.commonr r r rrrrrrrrray.util.state.exceptionrray.util.state.utilrr*r webRequestr-r;r&rErGr\dictrCrgrrrrsf  4          F