o bi$ @sddlZddlmZmZddlmZmZmZmZddlZ ddl m Z ddl m Z ddlmZmZmZmZmZmZmZmZmZmZmZddlmZddlmZd ed ed efd d Z deegeefde 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#edee)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_typesuccess error_messageresultcKs$td|rtjntj||dd|S)NF) status_codemessagerconvert_google_style)rr OKINTERNAL_ERROR)rrrkwargsrrQ/home/ubuntu/.local/lib/python3.10/site-packages/ray/dashboard/state_api_utils.pydo_replysr" list_api_fnreqc sbz|t|dIdH}tddt|dWSty0}ztdt|ddWYd}~Sd}~ww)NoptionTrrrF)options_from_reqr"rrstr)r#r$rerrr!handle_list_api$sr,returncCsn|jdg}|jdg}|jdg}t|t|ksJg}t|||D] \}}}||||fq'|S)N filter_keysfilter_predicates filter_values)querygetalllenzipappend)r$r.r/r0filterskey predicatevalrrr!_get_filters_from_req3sr:cCst|jddur|jdnt}|tkr!td|dtdt|jdd}t|}t|jdd t}t|jd d t}t |||||d S) z1Obtain `ListApiOptions` from the aiohttp request.limitNz Given limit z exceeds the supported limit z. Use a lower limit.timeoutdetailFexclude_driverT)r;r<r6r>r?) intr1getr r ValueErrorr:rboolr )r$r;r<r6r>r?rrr!r)@s(r)cCs6t|jdt}t|}|jdd}t|||dS)Nr< summary_by)r<r6rD)r@r1rAr r:r)r$r<r6rDrrr!summary_options_from_reqZsrE summary_fncs(|t|dIdH}tddt|dS)Nr%Tr'r()rEr"r)rFr$rrrr!handle_summary_apiasrGfilterschemacCspg}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 TypeErrorr@rrBfloatrCr5)rHrI new_filtercolr8r9 column_typerrr!convert_filters_typemsf      r\datar6state_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_columnslowerrBrVr*rCrr@r5r) r]r6r^r>rdatummatch filter_columnfilter_predicate filter_valuer`rrr! do_filtersb      rg)+rSrrtypingrrrr aiohttp.webaiohttpray.dashboard.optional_utilsrray.dashboard.utilsr ray.util.state.commonr r r r rrrrrrrray.util.state.exceptionrray.util.state.utilrrCr*r"webRequestr,r:r)rErGr\dictrgrrrr!sZ  4         F