o 2h<@s>dZddlZddlZddlmZddlmZGdddZdS)zDMiddleware for injecting client-side feature flags into the Context.N)context)errorsc@s0eZdZdZddZddZddZdd Zd S) ClientFeatureFlagsMiddlewareaMiddleware for injecting client-side feature flags into the Context. The client webapp is expected to include a json-serialized version of its FeatureFlags in the `X-TensorBoard-Feature-Flags` header or the `tensorBoardFeatureFlags` query parameter. This middleware extracts the header or query parameter value and converts it into the client_feature_flags property for the DataProvider's Context object, where client_feature_flags is a Dict of string keys and arbitrary value types. In the event that both the header and query parameter are specified, the values from the header will take precedence. cCs ||_dS)zxInitializes this middleware. Args: application: The WSGI application to wrap (see PEP 3333). N) _application)self applicationrc/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorboard/backend/client_feature_flags.py__init__&s z%ClientFeatureFlagsMiddleware.__init__cCsx||d}||d}|s|s|||S|D]\}}|||<qt|j|d}t|||||S)N HTTP_X_TENSORBOARD_FEATURE_FLAGS QUERY_STRING)client_feature_flags) #_parse_potential_header_param_flagsget"_parse_potential_query_param_flagsritemsr from_environreplaceset_in_environ)renvironstart_responseheader_feature_flagsquery_string_feature_flagsflagvaluectxrrr __call__.s     z%ClientFeatureFlagsMiddleware.__call__cCsJ|siSzt|}Wn tjytdwt|ts#td|S)Nz3X-TensorBoard-Feature-Flags cannot be JSON decoded.z8X-TensorBoard-Feature-Flags cannot be decoded to a dict.)jsonloadsJSONDecodeErrorrInvalidArgumentError isinstancedict)r header_stringrrrr rDs z@ClientFeatureFlagsMiddleware._parse_potential_header_param_flagscCs|siSztj|}Wn tyiYSw|dg}|s"iSz t|d}Wn tjy8t dwt |t sCt d|S)NtensorBoardFeatureFlagsrz/tensorBoardFeatureFlags cannot be JSON decoded.z4tensorBoardFeatureFlags cannot be decoded to a dict.) urllibparseparse_qs ValueErrorrrrrrr r!r")r query_stringquery_string_jsonpotential_feature_flagsr rrr rVs0  z?ClientFeatureFlagsMiddleware._parse_potential_query_param_flagsN)__name__ __module__ __qualname____doc__r rrrrrrr rs   r)r/r urllib.parser% tensorboardrrrrrrr s