U WiC7@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) AsyncClientget_origin_and_cast_ReturnTc@s4eZdZdddddddddd Zd d d d ZdS)AsyncQueryRequestBuilderrstrr r dictNonesessionpath http_methodheadersparamsjsonreturncCs(||_||_||_||_||_||_dSNr#r$r%r&r'r(selfr#r$r%r&r'r(r.f/home/afrizal/public_html/fastapi/venv/lib/python3.8/site-packages/postgrest/_async/request_builder.py__init__s z!AsyncQueryRequestBuilder.__init__zAPIResponse[_ReturnT]r)c s|jj|j|j|j|j|jdIdH}z~|jr|jdkr|j}|j ddkrT|WS|j drd|j dkrd|j dkr|WSt t  |WSt |WnNtk r}zt ||W5d}~XYn tk rt t|YnXdS)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&NHEADAccepttext/csvzapplication/vnd.pgrst.planz+json)r#requestr%r$r(r'r&Z is_successtextgetr rfrom_http_request_responserr rr)r-rbodyer.r.r/execute.s4   z AsyncQueryRequestBuilder.executeN__name__ __module__ __qualname__r0r=r.r.r.r/rsrc@s4eZdZdddddddddd Zd d d d ZdS)AsyncSingleRequestBuilderrrr r r r!r"cCs(||_||_||_||_||_||_dSr*r+r,r.r.r/r0Vs z"AsyncSingleRequestBuilder.__init__zSingleAPIResponse[_ReturnT]r1c s|jj|j|j|j|j|jdIdH}z:d|jkr>dkrRnntt  |WSt |WnNt k r}zt ||W5d}~XYn t k rt t|YnXdS)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. r2Ni+)r#r6r%r$r(r'r& status_coderrr9rr rrr-r:r<r.r.r/r=fs&   z!AsyncSingleRequestBuilder.executeNr>r.r.r.r/rBUsrBc@seZdZddddZdS)AsyncMaybeSingleRequestBuilderz%Optional[SingleAPIResponse[_ReturnT]]r1c std}ztt|IdH}Wn<tk rX}z|jrHd|jkrHWYdSW5d}~XYnX|sptddddd|S)NzThe result contains 0 rowszMissing responseZ204z"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)rBrr=rrJrEr.r.r/r=sz&AsyncMaybeSingleRequestBuilder.executeN)r?r@rAr=r.r.r.r/rFsrFc@s&eZdZdddddddddd Zd S) AsyncFilterRequestBuilderrrr r r r!r"c Cs:ttt||||ttt|||||||dSr*)rrrr0rr,r.r.r/r0s  z"AsyncFilterRequestBuilder.__init__Nr?r@rAr0r.r.r.r/rKsrKc@s&eZdZdddddddddd Zd S) AsyncRPCFilterRequestBuilderrrr r r r!r"c Cs:ttt||||ttt|||||||dSr*)rrrr0rBr,r.r.r/r0s  z%AsyncRPCFilterRequestBuilder.__init__NrLr.r.r.r/rMsrMc@sheZdZdddddddddd Zd d d d Zdd ddZifdddddddZdd ddZdS)AsyncSelectRequestBuilderrrr r r r!r"c Cs:ttt||||ttt|||||||dSr*)rrrr0rr,r.r.r/r0s  z"AsyncSelectRequestBuilder.__init__z#AsyncSingleRequestBuilder[_ReturnT]r1cCs.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+jsonr4r&r%r(r'r$r#)r&rBrr%r(r'r$r#r-r.r.r/singles z AsyncSelectRequestBuilder.singlez(AsyncMaybeSingleRequestBuilder[_ReturnT]cCs.d|jd<tt|j|j|j|j|j|jdS)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.rOr4rP)r&rFrr%r(r'r$r#rQr.r.r/ maybe_singles z&AsyncSelectRequestBuilder.maybe_singlezdict[str, Any]#AsyncFilterRequestBuilder[_ReturnT])columnqueryoptionsr)cCs|d}d}|dkrd}n|dkr*d}n |dkr6d}|d rRd |d d nd}|j||d |d ||_tt|j|j|j|j|j|j dS)NtypeplainplphrasephZ web_searchwconfig()Zfts.rP) r8r'addrrr&r%r(r$r#)r-rUrVrWtype_Z type_partZ config_partr.r.r/ text_searchs$   z%AsyncSelectRequestBuilder.text_searchzAsyncSingleRequestBuilder[str]cCs.d|jd<tt|j|j|j|j|j|jdS)zASpecify that the query must retrieve data as a single CSV string.r5r4r+)r&rBrr#r$r%r'r(rQr.r.r/csvs zAsyncSelectRequestBuilder.csvN)r?r@rAr0rRrSrerfr.r.r.r/rNs  rNc @seZdZddddddZdddd d d d d Zdejddddd dddddddZdejdddddd ddddddddZdejddd ddd d!d"Z dejdd ddd#d$d%Z dS)&AsyncRequestBuilderrrr!)r#r$r)cCs||_||_dSr*)r#r$)r-r#r$r.r.r/r0szAsyncRequestBuilder.__init__N)countzOptional[CountMethod]z#AsyncSelectRequestBuilder[_ReturnT])columnsrhr)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:`AsyncSelectRequestBuilder` rh)rrNrr#r$)r-rhrimethodr'r&r(r.r.r/selects zAsyncRequestBuilder.selectFTrh returningupsertdefault_to_nullzUnion[dict, list]rboolz"AsyncQueryRequestBuilder[_ReturnT])r(rhrmrnror)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:`AsyncQueryRequestBuilder` rl)rrrr#r$) r-r(rhrmrnrorjr'r&r.r.r/insert+szAsyncRequestBuilder.insertrYrhrmignore_duplicates on_conflictro)r(rhrmrsrtror)c Cs6t||||||d\}}} }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` rr)rrrr#r$) r-r(rhrmrsrtrorjr'r&r.r.r/rnLs zAsyncRequestBuilder.upsertrhrmr rT)r(rhrmr)cCs0t|||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:`AsyncFilterRequestBuilder` ru)rrKrr#r$)r-r(rhrmrjr'r&r.r.r/updateqszAsyncRequestBuilder.update)rhrmr)cCs.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:`AsyncFilterRequestBuilder` ru)rrKrr#r$)r-rhrmrjr'r&r(r.r.r/deleteszAsyncRequestBuilder.delete) r?r@rAr0rkrZrepresentationrqrnrvrwr.r.r.r/rgs(%)rgN)+ __future__rr(rtypingrrrrrZhttpxr r Zpydanticr Zbase_request_builderr rrrrrrrrrr exceptionsrrtypesrutilsrrrrrBrFrKrMrNrgr.r.r.r/s&   4  81 Q