o /h6@sZddlmZddlmZddlmZmZmZmZm Z ddl m Z m Z ddl mZddlmZmZmZmZmZmZmZmZmZmZmZddlmZmZdd lmZdd l m!Z!m"Z"ed Z#Gd d d ee#Z$Gdddee#Z%Gddde%e#Z&Gdddee#e$e#Z'Gdddee#e%e#Z(Gdddee#e$e#Z)Gdddee#Z*dS)) annotations)JSONDecodeError)AnyGenericOptionalTypeVarUnion)Headers QueryParams)ValidationError) APIResponseBaseFilterRequestBuilderBaseRPCRequestBuilderBaseSelectRequestBuilder CountMethodSingleAPIResponse pre_delete pre_insert pre_select pre_update pre_upsert)APIErrorgenerate_default_error_message) ReturnMethod) SyncClientget_origin_and_cast_ReturnTc@ eZdZdddZdddZdS)SyncQueryRequestBuildersessionrpathstr http_methodheadersr paramsr jsonUnion[dict, list]returnNonecC(||_||_||_||_||_||_dSNr r!r#r$r%r&selfr r!r#r$r%r&r/X/var/www/html/govbot/env/lib/python3.10/site-packages/postgrest/_sync/request_builder.py__init__   z SyncQueryRequestBuilder.__init__APIResponse[_ReturnT]c Cs|jj|j|j|j|j|jd}z=|jrH|jdkr@|j}|j ddkr'|WS|j dr@d|j dvr@d|j dvr@|WSt t  |WSt |ty`}zt ||d}~wtylt t|w)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. r&r%r$HEADAccepttext/csvzapplication/vnd.pgrst.planz+jsonN)r requestr#r!r&r%r$ is_successtextgetr rfrom_http_request_responserr rr)r.rbodyer/r/r0execute.s6     zSyncQueryRequestBuilder.executeN)r rr!r"r#r"r$r r%r r&r'r(r))r(r3__name__ __module__ __qualname__r1r@r/r/r/r0r rc@r)SyncSingleRequestBuilderr rr!r"r#r$r r%r r&dictr(r)cCr*r+r,r-r/r/r0r1Vr2z!SyncSingleRequestBuilder.__init__SingleAPIResponse[_ReturnT]c Cs|jj|j|j|j|j|jd}zd|jkrdkr&nntt  |WSt |t y>}zt ||d}~wt yJt t|w)zExecute the query. .. tip:: This is the last method called, after the query is built. Returns: :class:`SingleAPIResponse` Raises: :class:`APIError` If the API raised an error. r4i+N)r r8r#r!r&r%r$ status_coderrr<rr rrr.r=r?r/r/r0r@fs"    z SyncSingleRequestBuilder.executeNr rr!r"r#r"r$r r%r r&rGr(r))r(rHrAr/r/r/r0rFUrErFc@seZdZdddZdS)SyncMaybeSingleRequestBuilderr(%Optional[SingleAPIResponse[_ReturnT]]c Cstd}z tt|}Wn!ty,}z|jr"d|jvr"WYd}~dSWYd}~nd}~ww|s8tddddd|S)NzThe result contains 0 rowszMissing response204z"Please check traceback of the codezPostgrest couldn't retrieve response, please check traceback of the code. Please create an issue in `supabase-community/postgrest-py` if needed.)messagecodehintdetails)rFrr@rrSrKr/r/r0r@s"z%SyncMaybeSingleRequestBuilder.executeN)r(rN)rBrCrDr@r/r/r/r0rMrMc@eZdZdddZdS)SyncFilterRequestBuilderr rr!r"r#r$r r%r r&rGr(r)c C:ttt||||ttt|||||||dSr+)rrrr1rr-r/r/r0r1  z!SyncFilterRequestBuilder.__init__NrLrBrCrDr1r/r/r/r0rVrTrVc@rU)SyncRPCFilterRequestBuilderr rr!r"r#r$r r%r r&rGr(r)c CrWr+)rrrr1rFr-r/r/r0r1rXz$SyncRPCFilterRequestBuilder.__init__NrLrYr/r/r/r0rZsrZc@sBeZdZd!ddZd"ddZd#ddZifd$ddZd%ddZd S)&SyncSelectRequestBuilderr rr!r"r#r$r r%r r&rGr(r)c CrWr+)rrrr1rr-r/r/r0r1rXz!SyncSelectRequestBuilder.__init__"SyncSingleRequestBuilder[_ReturnT]cC.d|jd<tt|j|j|j|j|j|jdS)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+jsonr6r$r#r&r%r!r )r$rFrr#r&r%r!r r.r/r/r0singles zSyncSelectRequestBuilder.single'SyncMaybeSingleRequestBuilder[_ReturnT]cCr])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.r^r6r_)r$rMrr#r&r%r!r r`r/r/r0 maybe_single z%SyncSelectRequestBuilder.maybe_singlecolumnqueryoptionsdict[str, Any]"SyncFilterRequestBuilder[_ReturnT]cCs|d}d}|dkrd}n |dkrd}n|dkrd}|d r)d |d d nd}|j||d |d ||_tt|j|j|j|j|j|j dS)Ntypeplainplphraseph web_searchwconfig()fts.r_) r;r%addrrr$r#r&r!r )r.rerfrgtype_ type_part config_partr/r/r0 text_searchs$   z$SyncSelectRequestBuilder.text_searchSyncSingleRequestBuilder[str]cCs.d|jd<tt|j|j|j|j|j|jdS)zASpecify that the query must retrieve data as a single CSV string.r7r6r,)r$rFr"r r!r#r%r&r`r/r/r0csvrdzSyncSelectRequestBuilder.csvNrL)r(r\)r(rb)rer"rfr"rgrhr(ri)r(r|)rBrCrDr1rarcr{r}r/r/r/r0r[s    r[c@seZdZd+ddZd d d,ddZd ejdddd-ddZd ejddddd.d"d#Zd ejd$d/d'd(Z d ejd$d0d)d*Z d S)1SyncRequestBuilderr rr!r"r(r)cCs||_||_dSr+)r r!)r.r r!r/r/r0r1s zSyncRequestBuilder.__init__N)countcolumnsrOptional[CountMethod]"SyncSelectRequestBuilder[_ReturnT]cGs0t|d|i\}}}}tt|j|j||||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:`SyncSelectRequestBuilder` r)rr[rr r!)r.rrmethodr%r$r&r/r/r0selects zSyncRequestBuilder.selectFTr returningupsertdefault_to_nullr&r'rrrboolr!SyncQueryRequestBuilder[_ReturnT]c Cs4t|||||d\}}}}tt|j|j||||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:`SyncQueryRequestBuilder` r)rrrr r!) r.r&rrrrrr%r$r/r/r0insert+szSyncRequestBuilder.insertrkrrignore_duplicates on_conflictrrrc Cs6t||||||d\}}} }tt|j|j|| ||S)aRun 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:`SyncQueryRequestBuilder` r)rrrr r!) r.r&rrrrrrr%r$r/r/r0rLszSyncRequestBuilder.upsertrrrGricCs0t|||d\}}}}tt|j|j||||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:`SyncFilterRequestBuilder` r)rrVrr r!)r.r&rrrr%r$r/r/r0updateqszSyncRequestBuilder.updatecCs.t||d\}}}}tt|j|j||||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:`SyncFilterRequestBuilder` r)rrVrr r!)r.rrrr%r$r&r/r/r0deleteszSyncRequestBuilder.delete)r rr!r"r(r))rr"rrr(r) r&r'rrrrrrrrr(r)r&r'rrrrrrrr"rrr(r)r&rGrrrrr(ri)rrrrr(ri) rBrCrDr1rrrepresentationrrrrr/r/r/r0r~s* %)r~N)+ __future__rr&rtypingrrrrrhttpxr r pydanticr base_request_builderr rrrrrrrrrr exceptionsrrtypesrutilsrrrrrFrMrVrZr[r~r/r/r/r0s&  4 81  Q