o ˳i5@shddlmZddlmZmZmZmZmZmZm Z ddl m Z m Z m Z mZmZddlmZddlmZddlmZddlmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"dd l#m$Z$m%Z%m&Z&dd l'm(Z(m)Z)dd l*m+Z+ee Z,Gd d d Z-GdddZ.GdddZ/GdddZ0Gdddee e-Z1Gdddee.Z2Gddde-ee Z3GdddZ4dS)) annotations)AnyGenericLiteralOptionalTypeVarUnionoverload) AsyncClient BasicAuthHeaders QueryParamsResponse)ValidationError)override)URL) APIResponseBaseFilterRequestBuilderBaseRPCRequestBuilderBaseSelectRequestBuilder CountMethod RequestConfigSingleAPIResponse pre_delete pre_insert pre_select pre_update pre_upsert)APIErrorAPIErrorFromJSONgenerate_default_error_message)JSON ReturnMethod)model_validate_jsonc@ eZdZd ddZd ddZd S) AsyncQueryRequestBuilderrequest ReqConfigcC ||_dSNr'selfr'r.T/home/ubuntu/.local/lib/python3.10/site-packages/postgrest/_async/request_builder.py__init__  z!AsyncQueryRequestBuilder.__init__returnrc sb|jIdH}z|jrt|WStt|j}tt |t y0}ztt |d}~ww)zExecute the query. .. tip:: This is the last method called, after the query is built. Returns: :class:`APIResponse` Raises: :class:`APIError` If the API raised an error. N) r'send is_successrfrom_http_request_responser$r contentrdictrr!r-rjson_objer.r.r/execute#s     z AsyncQueryRequestBuilder.executeNr'r()r2r__name__ __module__ __qualname__r0r<r.r.r.r/r& r&c@r%) AsyncSingleRequestBuilderr'r(cCr)r*r+r,r.r.r/r0;r1z"AsyncSingleRequestBuilder.__init__r2rc sv|jIdH}zd|jkrdkrnnt|WStt|j}tt |t y:}ztt |d}~ww)a1Execute the query. .. tip:: This is the last method called, after the query is built. Returns: :class:`SingleAPIResponse` na Raises: :class:`APIError` If the API raised an error. Ni+) r'r3 status_coderr5r$r r6rr7rr!r8r.r.r/r<>s     z!AsyncSingleRequestBuilder.executeNr=)r2rr>r.r.r.r/rC:rBrCc@r%) AsyncExplainRequestBuilderr'r(cCr)r*r+r,r.r.r/r0Xr1z#AsyncExplainRequestBuilder.__init__r2strc s^|jIdH}z|jr|jWStt|j}tt|t y.}ztt |d}~wwr*) r'r3r4textr$r r6rr7rr!r8r.r.r/r<[s   z"AsyncExplainRequestBuilder.executeNr=)r2rGr>r.r.r.r/rFWrBrFc@r%) AsyncMaybeSingleRequestBuilderr'r(cCr)r*r+r,r.r.r/r0hr1z'AsyncMaybeSingleRequestBuilder.__init__r2Optional[SingleAPIResponse]c s|d}z t|jIdH}Wn!ty0}z|jr&d|jvr&WYd}~dSWYd}~nd}~ww|sr.r.r.r/rIgrBrIc@eZdZdddZdS) AsyncFilterRequestBuilderr'r(r2NonecCt||t||dSr*)rr0r&r,r.r.r/r0 z"AsyncFilterRequestBuilder.__init__Nr'r(r2rRr?r@rAr0r.r.r.r/rQ~srQc@rP) AsyncRPCFilterRequestBuilderr'r(r2rRcCrSr*)rr0rCr,r.r.r/r0rTz%AsyncRPCFilterRequestBuilder.__init__NrUrVr.r.r.r/rWsrWc@seZdZd*ddZd+dd Zd,d d Zifd-ddZd+ddZe      d.d/d"d#Z e     d0d1d%d#Z      d.d2d(d#Z d)S)3AsyncSelectRequestBuilderr'r(r2rRcCrSr*)rr0r&r,r.r.r/r0rTz"AsyncSelectRequestBuilder.__init__rCcCd|jjd<t|jS)zSpecify that the query will only return a single row in response. .. caution:: The API will raise an error if the query returned more than one row. !application/vnd.pgrst.object+jsonAcceptr'headersrCr-r.r.r/singles  z AsyncSelectRequestBuilder.singlerIcCrY)zRetrieves at most one row from the result. Result must be at most one row (e.g. using `eq` on a UNIQUE column), otherwise this will result in an error.rZr[)r'r]rIr^r.r.r/ maybe_single  z&AsyncSelectRequestBuilder.maybe_singlecolumnrGqueryoptionsdict[str, Any]r&cCs|d}d}|dkrd}n |dkrd}n|dkrd}|d r)d |d d nd}|jj||d |d ||j_t|jS)Ntypeplainplphraseph web_searchwconfig()fts.)getr'paramsaddr&)r-rbrcrdtype_ type_part config_partr.r.r/ text_searchs   z%AsyncSelectRequestBuilder.text_searchcCrY)zASpecify that the query must retrieve data as a single CSV string.ztext/csvr[r\r^r.r.r/csvrazAsyncSelectRequestBuilder.csvFrHanalyzeboolverbosesettingsbufferswalformatLiteral['text']rFcCdSr*r.r-r{r}r~rrrr.r.r/explains z!AsyncSelectRequestBuilder.explainLiteral['json']cCrr*r.rr.r.r/rs Literal['text', 'json']6AsyncExplainRequestBuilder | AsyncSingleRequestBuilderc CsRddtD}d|}d|d||jjd<|dkr$t|jSt|jS)NcSs g|] \}}|dvr|r|qS))r-rr.).0keyvaluer.r.r/ s  z5AsyncSelectRequestBuilder.explain..|zapplication/vnd.pgrst.plan+z ; options=r[rH)localsitemsjoinr'r]rFrC) r-r{r}r~rrrrd options_strr.r.r/rs     NrU)r2rC)r2rI)rbrGrcrGrdrer2r&)FFFFFrH)r{r|r}r|r~r|rr|rr|rrr2rF)FFFFF)r{r|r}r|r~r|rr|rr|rrr2rC)r{r|r}r|r~r|rr|rr|rrr2r) r?r@rAr0r_r`ryrzr rr.r.r.r/rXs:      rXc@seZdZd1d d Zd d dd2ddZd ejdddd3d#d$Zd ejdd%dd&d4d)d*Zd ejd+d5d-d.Z d ejd+d6d/d0Z d S)7AsyncRequestBuildersessionr pathrr]r authBasicAuth | Noner2rRcCs||_||_||_||_dSr*)rrr]r)r-rrr]rr.r.r/r0s zAsyncRequestBuilder.__init__NcountheadcolumnsrGrOptional[CountMethod]rOptional[bool]rXc GsHt|||d\}}}}||jt|j|j|j||||d}t|S)zRun a SELECT query. Args: *columns: The names of the columns to fetch. count: The method to use to get the count of rows returned. Returns: :class:`AsyncSelectRequestBuilder` rrrrrt http_methodr]json)rupdater]rrrrrX) r-rrrmethodrtr]rr'r.r.r/selects  zAsyncRequestBuilder.selectFTr returningupsertdefault_to_nullrr"rr#rr|rr&c CsJt|||||d\}}}}||jt|j|j|j||||d} t| S)aRun an INSERT query. Args: json: The row to be inserted. count: The method to use to get the count of rows returned. returning: Either 'minimal' or 'representation' upsert: Whether the query should be an upsert. default_to_null: Make missing fields default to `null`. Otherwise, use the default value for the column. Only applies for bulk inserts. Returns: :class:`AsyncQueryRequestBuilder` rr)rrr]rrrrr&) r-rrrrrrrtr]r'r.r.r/insert s$  zAsyncRequestBuilder.insertrgrrignore_duplicates on_conflictrrrc CsLt||||||d\}}} }| |jt|j|j|j||| |d} t| S)a Run an upsert (INSERT ... ON CONFLICT DO UPDATE) query. Args: json: The row to be inserted. count: The method to use to get the count of rows returned. returning: Either 'minimal' or 'representation' ignore_duplicates: Whether duplicate rows should be ignored. on_conflict: Specified columns to be made to work with UNIQUE constraint. default_to_null: Make missing fields default to `null`. Otherwise, use the default value for the column. This only applies when inserting new rows, not when merging with existing rows under `ignoreDuplicates: false`. This also only applies when doing bulk upserts. Returns: :class:`AsyncQueryRequestBuilder` rr)rrr]rrrrr&) r-rrrrrrrrtr]r'r.r.r/r4s&  zAsyncRequestBuilder.upsertrrrQc CsFt|||d\}}}}||jt|j|j|j||||d}t|S)aRun an UPDATE query. Args: json: The updated fields. count: The method to use to get the count of rows returned. returning: Either 'minimal' or 'representation' Returns: :class:`AsyncFilterRequestBuilder` rr)rrr]rrrrrQ)r-rrrrrtr]r'r.r.r/ras   zAsyncRequestBuilder.updatec CsDt||d\}}}}||jt|j|j|j||||d}t|S)zRun a DELETE query. Args: count: The method to use to get the count of rows returned. returning: Either 'minimal' or 'representation' Returns: :class:`AsyncFilterRequestBuilder` rr)rrr]rrrrrQ)r-rrrrtr]rr'r.r.r/deletes  zAsyncRequestBuilder.delete) rr rrr]r rrr2rR)rrGrrrrr2rX) rr"rrrr#rr|rr|r2r&)rr"rrrr#rr|rrGrr|r2r&)rr"rrrr#r2rQ)rrrr#r2rQ) r?r@rAr0rr#representationrrrrr.r.r.r/rs,  -1$rN)5 __future__rtypingrrrrrrr httpxr r r r rpydanticrtyping_extensionsryarlrbase_request_builderrrrrrrrrrrrr exceptionsrr r!typesr"r#utilsr$r(r&rCrFrIrQrWrXrr.r.r.r/s. $   8   [