gs7ddlmZddlmZddlmZmZmZmZm Z ddl m Z m Z ddl mZddlmZmZmZmZmZmZmZmZmZmZmZddlmZmZdd lmZdd l m!Z!m"Z"ed Z#Gd d ee#Z$Gddee#Z%Gdde%e#Z&Gddee#e$e#Z'Gddee#e%e#Z(Gddee#e$e#Z)Gddee#Z*y)) annotations)JSONDecodeError)AnyGenericOptionalTypeVarUnion)Headers QueryParams)ValidationError) APIResponseBaseFilterRequestBuilderBaseRPCRequestBuilderBaseSelectRequestBuilder CountMethodSingleAPIResponse pre_delete pre_insert pre_select pre_update pre_upsert)APIErrorgenerate_default_error_message) ReturnMethod) AsyncClientget_origin_and_cast_ReturnTc8eZdZ ddZddZy)AsyncQueryRequestBuildercX||_||_||_||_||_||_yNsessionpath http_methodheadersparamsjsonselfr$r%r&r'r(r)s U/var/www/openai/venv/lib/python3.12/site-packages/postgrest/_async/request_builder.py__init__z!AsyncQueryRequestBuilder.__init__/  &   cK|jj|j|j|j|j |j d{} |jr|jdk7r|j}|j jddk(r|S|j jdr||##v-66D||''1Z?# ||'' 6$,,:J:J8:TT"$,,*:*:8*DD#'K"8,GGJJqvvx(() * ,1668$! + >9!<= = >s\AE:D1E:AD3$E:%AD3;E:<D3E:D31E:3 E7<E!E77E:Nr$rr%strr&r?r'r r(r r)dictreturnNone)rAzAPIResponse[_ReturnT]__name__ __module__ __qualname__r-r=r/r,r r sP       $>r/r c8eZdZ ddZddZy)AsyncSingleRequestBuildercX||_||_||_||_||_||_yr"r#r*s r,r-z"AsyncSingleRequestBuilder.__init__Vr.r/cK|jj|j|j|j|j |j d{} d|jcxkrdkrnnttj|St|j 7R#t$r}t|j |d}~wt$rtt|wxYww)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. r1Ni+)r$r5r&r%r)r(r' status_coderrr9rr rrr+r:r<s r,r=z!AsyncSingleRequestBuilder.executefs,,&&    II;;LL '   >q}}++(2MMaPPqvvx((  ,1668$! + >9!<= = >sBAC4B+C43B-C4B-+C4- C16C!C11C4Nr>)rAzSingleAPIResponse[_ReturnT]rCrGr/r,rIrIUsP       >r/rIceZdZddZy)AsyncMaybeSingleRequestBuildercKd} ttj|d{}|stddddd|S7#t$r)}|jrd|jvrYd}~yYd}~Cd}~wwxYww)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)rIrr=rrVrNs r,r=z&AsyncMaybeSingleRequestBuilder.executes  /9AA$GGA1!@ r  H yy9QYYF s;A5A>AA5A A2 A-# A5-A22A5N)rAz%Optional[SingleAPIResponse[_ReturnT]])rDrErFr=rGr/r,rPrPsr/rPc0eZdZ ddZy)AsyncFilterRequestBuilderc tttj||||tttj|||||||yr")rrrr-r r*s r,r-z"AsyncFilterRequestBuilder.__init__P 4X>?HH '7F  4X>?HH '4gvt r/Nr>rDrErFr-rGr/r,rXrXsK            r/rXc0eZdZ ddZy)AsyncRPCFilterRequestBuilderc tttj||||tttj|||||||yr")rrrr-rIr*s r,r-z%AsyncRPCFilterRequestBuilder.__init__sP 4X>?HH '7F  5h?@II '4gvt r/Nr>r[rGr/r,r]r]sK            r/r]cbeZdZ ddZddZd dZif d dZd dZy) AsyncSelectRequestBuilderc tttj||||tttj|||||||yr")rrrr-r r*s r,r-z"AsyncSelectRequestBuilder.__init__rZr/cd|jd<tt|j|j|j|j |j |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+jsonr3r'r&r)r(r%r$)r'rIrr&r)r(r%r$r+s r,singlez AsyncSelectRequestBuilder.singlesQ "E X(2LL((;;LL   r/cd|jd<tt|j|j|j|j |j |jS)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.rcr3rd)r'rPrr&r)r(r%r$res r, maybe_singlez&AsyncSelectRequestBuilder.maybe_singlesN!D X-h7LL((;;LL   r/c|jd}d}|dk(rd}n|dk(rd}n|dk(rd}|jd rd |jd d nd}|jj||d |d ||_tt|j |j |j|j|j|jS)Ntypeplainplphraseph web_searchwconfig()fts.rd) r8r(addr rr'r&r)r%r$)r+columnqueryoptionstype_ type_part config_parts r, text_searchz%AsyncSelectRequestBuilder.text_searchs F# G I h I l "I6=kk(6K'++h/02QS kkoof 3{m1UG.TU '1LL((;;LL   r/cd|jd<tt|j|j|j |j|j |jS)zASpecify that the query must retrieve data as a single CSV string.r4r3r#)r'rIr?r$r%r&r(r)res r,csvzAsyncSelectRequestBuilder.csvsN!+ X(-LL((LL;;   r/Nr>)rAz#AsyncSingleRequestBuilder[_ReturnT])rAz(AsyncMaybeSingleRequestBuilder[_ReturnT])rxr?ryr?rzzdict[str, Any]rA#AsyncFilterRequestBuilder[_ReturnT])rAzAsyncSingleRequestBuilder[str])rDrErFr-rfrhr~rrGr/r,r`r`s              BD  "% 0> , .  r/r`ceZdZddZddd ddZdej ddd ddZdej dd dd  dd Zdej d  dd Z dej d  ddZ y)AsyncRequestBuilderc ||_||_yr")r$r%)r+r$r%s r,r-zAsyncRequestBuilder.__init__s  r/Ncountheadczt|||d\}}}}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:`AsyncSelectRequestBuilder` r)rr`rr$r%)r+rrcolumnsmethodr(r'r)s r,selectzAsyncRequestBuilder.selectsB)3G5t(T%(2 LL$))VWfd  r/FTr returningupsertdefault_to_nullct|||||\}}}}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:`AsyncQueryRequestBuilder` r)rr rr$r%) r+r)rrrrrr(r's r,insertzAsyncRequestBuilder.insert,sO,)3 + ) %(1 LL$))VWfd  r/rkrrignore_duplicates on_conflictrct||||||\}}} }tt|j|j|| ||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` r)rr rr$r%) r+r)rrrrrrr(r's r,rzAsyncRequestBuilder.upsertMsR2)3 /#+ ) %(1 LL$))VWfd  r/rrc~t|||\}}}}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:`AsyncFilterRequestBuilder` r)rrXrr$r%)r+r)rrrr(r's r,updatezAsyncRequestBuilder.updatersI )3 ) % )2 LL$))VWfd  r/c|t||\}}}}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:`AsyncFilterRequestBuilder` r)rrXrr$r%)r+rrrr(r'r)s r,deletezAsyncRequestBuilder.deletesF)3) %)2 LL$))VWfd  r/)r$rr%r?rArB)rr?rOptional[CountMethod]rzOptional[bool]rAz#AsyncSelectRequestBuilder[_ReturnT]) r)Union[dict, list]rrrrrboolrrrA"AsyncQueryRequestBuilder[_ReturnT])r)rrrrrrrrr?rrrAr)r)r@rrrrrAr)rrrrrAr) rDrErFr-rrrepresentationrrrrrGr/r,rrsY(,#   %   -  .(,"."="= $  %        , J(,"."="="' $# # % #  #  # # #  ,# R(,"."="=   %    -  8(,"."="=  %   -  r/rN)+ __future__rr)rtypingrrrrr httpxr r pydanticr base_request_builderrrrrrrrrrrr exceptionsrrtypesrutilsrrrr rIrPrXr]r`rrGr/r,rs" 99&$    B 4 : 5>wx05>p.> 1.>b%>x%H* 8 BD\]eDf & (#%>x%H *N 8 BD\]eDfN bK '(+K r/