U Wi6@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@s4eZdZdddddddddd Zd d d d ZdS)SyncQueryRequestBuilderrstrr r Union[dict, list]Nonesessionpath http_methodheadersparamsjsonreturncCs(||_||_||_||_||_||_dSNr#r$r%r&r'r(selfr#r$r%r&r'r(r.e/home/afrizal/public_html/fastapi/venv/lib/python3.8/site-packages/postgrest/_sync/request_builder.py__init__s z SyncQueryRequestBuilder.__init__zAPIResponse[_ReturnT]r)c Cs|jj|j|j|j|j|jd}z~|jr|jdkr|j}|j ddkrN|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&HEADAccepttext/csvzapplication/vnd.pgrst.planz+jsonN)r#requestr%r$r(r'r&Z is_successtextgetr rfrom_http_request_responserr rr)r-rbodyer.r.r/execute.s4  zSyncQueryRequestBuilder.executeN__name__ __module__ __qualname__r0r=r.r.r.r/rsrc@s4eZdZdddddddddd Zd d d d ZdS)SyncSingleRequestBuilderrrr r dictr!r"cCs(||_||_||_||_||_||_dSr*r+r,r.r.r/r0Vs z!SyncSingleRequestBuilder.__init__zSingleAPIResponse[_ReturnT]r1c Cs|jj|j|j|j|j|jd}z:d|jkr8dkrLnntt  |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. r2i+N)r#r6r%r$r(r'r& status_coderrr9rr rrr-r:r<r.r.r/r=fs&  z SyncSingleRequestBuilder.executeNr>r.r.r.r/rBUsrBc@seZdZddddZdS)SyncMaybeSingleRequestBuilderz%Optional[SingleAPIResponse[_ReturnT]]r1c Csnd}ztt|}Wn<tk rR}z|jrBd|jkrBWYdSW5d}~XYnX|sjtddddd|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=rrKrFr.r.r/r=sz%SyncMaybeSingleRequestBuilder.executeN)r?r@rAr=r.r.r.r/rGsrGc@s&eZdZdddddddddd Zd S) SyncFilterRequestBuilderrrr r rCr!r"c Cs:ttt||||ttt|||||||dSr*)rrrr0rr,r.r.r/r0s  z!SyncFilterRequestBuilder.__init__Nr?r@rAr0r.r.r.r/rLsrLc@s&eZdZdddddddddd Zd S) SyncRPCFilterRequestBuilderrrr r rCr!r"c Cs:ttt||||ttt|||||||dSr*)rrrr0rBr,r.r.r/r0s  z$SyncRPCFilterRequestBuilder.__init__NrMr.r.r.r/rNsrNc@sheZdZdddddddddd Zd d d d Zdd ddZifdddddddZdd ddZdS)SyncSelectRequestBuilderrrr r rCr!r"c Cs:ttt||||ttt|||||||dSr*)rrrr0rr,r.r.r/r0s  z!SyncSelectRequestBuilder.__init__z"SyncSingleRequestBuilder[_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 zSyncSelectRequestBuilder.singlez'SyncMaybeSingleRequestBuilder[_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.rPr4rQ)r&rGrr%r(r'r$r#rRr.r.r/ maybe_singles z%SyncSelectRequestBuilder.maybe_singlezdict[str, Any]"SyncFilterRequestBuilder[_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.rQ) r8r'addrrr&r%r(r$r#)r-rVrWrXtype_Z type_partZ config_partr.r.r/ text_searchs$   z$SyncSelectRequestBuilder.text_searchzSyncSingleRequestBuilder[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(rRr.r.r/csvs zSyncSelectRequestBuilder.csvN)r?r@rAr0rSrTrfrgr.r.r.r/rOs  rOc @seZdZddddddZdddd d d d d Zdejddddd dddddddZdejdddddd ddddddddZdejddd ddd d!d"Z dejdd ddd#d$d%Z dS)&SyncRequestBuilderrrr!)r#r$r)cCs||_||_dSr*)r#r$)r-r#r$r.r.r/r0szSyncRequestBuilder.__init__N)countzOptional[CountMethod]z"SyncSelectRequestBuilder[_ReturnT])columnsrir)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` ri)rrOrr#r$)r-rirjmethodr'r&r(r.r.r/selects zSyncRequestBuilder.selectFTri returningupsertdefault_to_nullr rboolz!SyncQueryRequestBuilder[_ReturnT])r(rirnrorpr)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` rm)rrrr#r$) r-r(rirnrorprkr'r&r.r.r/insert+szSyncRequestBuilder.insertrZrirnignore_duplicates on_conflictrp)r(rirnrtrurpr)c 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` rs)rrrr#r$) r-r(rirnrtrurprkr'r&r.r.r/roLs zSyncRequestBuilder.upsertrirnrCrU)r(rirnr)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:`SyncFilterRequestBuilder` rv)rrLrr#r$)r-r(rirnrkr'r&r.r.r/updateqszSyncRequestBuilder.update)rirnr)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:`SyncFilterRequestBuilder` rv)rrLrr#r$)r-rirnrkr'r&r(r.r.r/deleteszSyncRequestBuilder.delete) r?r@rAr0rlrZrepresentationrrrorwrxr.r.r.r/rhs(%)rhN)+ __future__rr(rtypingrrrrrZhttpxr r Zpydanticr Zbase_request_builderr rrrrrrrrrr exceptionsrrtypesrutilsrrrrrBrGrLrNrOrhr.r.r.r/s&   4  81 Q