o wiYb@sUddlmZddlZddlZddlZddlZddlZddlmZm Z ddl m Z m Z m Z mZmZmZmZddlmZddlmZddlmZddlmZmZmZdd lmZdd lmZd d lm Z d d l!m"Z"d dl#m$Z$m%Z%m&Z&d dl'm(Z(m)Z)m*Z*m+Z+m,Z,d dl-m.Z.d dl/m0Z0e rd dl1m2Z3d dl4m5Z5e6e7Z8de9d<e de:fZ;efZ?de9d<iddddddddddddd dd!dd"dd#dd$d%d&d%d'd%d(d%d)d%d*d%d+d%Z@d,e9d-<idd.dd/dd0dd1dd2dd2d d3d!d4d"d5d#d6d$d7d&d8d'd9d(d:d)d;d*d<d+d=ZAd>e9d?<eBZCed@ZDGdAdBdBZEdzdKdLZFd{dNdOZGd|dRdSZHGdTddeZIGdUdVdVeIZJGdWdXdXeJZKGdYdZdZeJZLGd[d\d\eJZMGd]d^d^eJZNGd_d`d`eJZOGdadbdbeJZPGdcddddeIZQed}dhdiZReRSejd~dldmZTeRSe jUddodpZVeRSe jWddrdsZXeRSejYddudvZZddxdyZ[dS)) annotationsN)ABCabstractmethod) TYPE_CHECKINGAnyUnionMappingIterable ForwardRefcast)timezone)indent)singledispatch)Literal TypeGuardoverride) BaseModel) FieldInfo)fields) PrismaMethod)InvalidModelErrorUnknownModelErrorUnknownRelationalFieldError)get_argsis_union get_origin model_fieldsmodel_field_type) is_list_type)QUERY_BUILDER_ALIASES _PrismaModel) Serializablezlogging.Loggerlog AbstractNodeztuple[type[Any], ...] ITERABLEScreatemutationdeleteupdateZupsert query_raw query_firstZ create_many execute_rawZ delete_manyZ update_manycountqueryZgroup_byZ find_manyZ find_firstZfind_first_or_raiseZ find_uniqueZfind_unique_or_raisezdict[PrismaMethod, Operation]METHOD_OPERATION_MAPPINGzcreateOne{model}zdeleteOne{model}zupdateOne{model}zupsertOne{model}ZqueryRawzcreateMany{model}Z executeRawzdeleteMany{model}zupdateMany{model}zaggregate{model}zgroupBy{model}zfindMany{model}zfindFirst{model}zfindFirst{model}OrThrowzfindUnique{model}zfindUnique{model}OrThrowzdict[PrismaMethod, str]METHOD_FORMAT_MAPPING)r/r(c@seZdZUded< ded< ded< ded< d ed < d ed < d ed< ded< ded< dZdddd,ddZd-ddZd-ddZd.d d!Zd/d$d%Z d0d(d)Z d1d*d+Z dS)2 QueryBuilderrmethodstr method_format Operation operationtype[PrismaModel] | Nonemodeldict[str, Any] | Noneincludedict[str, Any] argumentslist[str] | Noneroot_selectionset[str] prisma_modelsdict[str, dict[str, str]]relational_field_mappings) r3r5r7r9r;r=r?rArCN)r9r?type[BaseModel] | NonereturnNonecCs||_t||_t||_||_||_||_|||_ }| dd|_ |dus-|dvr2d|_ dSt |r;t|ds?t|||_ dS)Nr;r-r,r+__prisma_model__)r3r1r5r0r7r?rArC_transform_aliasesr=popr;r9_is_prisma_model_typehasattrr)selfr3r=rArCr9r?argsrOF/home/app/Keep/.python/lib/python3.10/site-packages/prisma/_builder.py__init__~s    zQueryBuilder.__init__cCsi|j|d}t|S)z8Build the payload that should be sent to the QueryEngine) variablesZoperation_namer/)r7 build_querydumps)rMdatarOrOrPbuilds zQueryBuilder.buildcCs|}td||S)arBuild the GraphQL query Example query: query { result: findUniqueUser ( where: { id: "ckq23ky3003510r8zll5m2hma" } ) { id name profile { id user_id bio } } } zGenerated query: %s)_create_root_noderenderr$debug)rMr/rOrOrPrSs  zQueryBuilder.build_query 'RootNode'cCs<t|d}|t||tj||j|j|jd|S)N)builderr9r;r?)RootNodeadd ResultNoder' Selectionr9r;r?)rMrootrOrOrPrWs zQueryBuilder._create_root_nodetype[PrismaModel] list[str]csNtdt}|turtj}||jvrt|fddtDS)zReturns a list of all the scalar fields of a model Raises UnknownModelError if the current model cannot be found. rHcs"g|] \}}t||ds|qS)nameparent)_field_is_prisma_model).0fieldinfor9rOrP s  z3QueryBuilder.get_default_fields..)getattrMISSINGrrHrArritems)rMr9rerOrkrPget_default_fieldss    zQueryBuilder.get_default_fields current_modelric Cst|dt}|turt|tt|}z|j|}Wnty+}zt||d}~ww||vr7t|j |dzt ||}WntyS}zt|j |d|d}~wwt |||d}|sgt d|dd|S)zReturns the model that the field is related to. Raises UnknownModelError if the current model is invalid. Raises UnknownRelationalFieldError if the field does not exist. rHN)r9rirdzThe `z2` field doesn't appear to be a Prisma Model type. z\Is the field a pydantic.BaseModel type and does it have a `__prisma_model__` class variable?) rmrnrr r4rCKeyErrorrr__name__r_prisma_model_for_field RuntimeError)rMrqrireZmappingsexcrjr9rOrOrPget_relational_models4    z!QueryBuilder.get_relational_modelcsnt}|D]-\}}t||}t|trj|d||<qt|tr0fdd|D||<q|||<q|S)zVTransform dict keys to match global aliases e.g. order_by -> orderBy r=cs$g|]}t|tr|n|qSrO) isinstancedictrI)rhrUrMrOrPrlsz3QueryBuilder._transform_aliases..)rzror getryrIr&)rMr=Z transformedkeyvaluealiasrOr{rPrIs     zQueryBuilder._transform_aliases)r3rr=r<rAr@rCrBr9rDr?r>rErFrEr4)rErZ)r9rbrErc)rqrbrir4rErb)r=r<rEr<) rs __module__ __qualname____annotations__ __slots__rQrVrSrWrprwrIrOrOrOrPr2Ps:    "r2rirrer4rftype[BaseModel]rEr8c Cs|j}t|}|durtd|d|tt|r t|}n|g}|D],}t|tr:td|d|d|dt|rH|durHt|d}t |drQ|Sq%dS)Nz"Unexpected field type is None for .z"Encountered forward reference for z-; Forward references must be evaluated using z.update_forward_refs()rrH) rsrrurrrryr rrL)rirerfcls_nametype_typesrOrOrPrts$     rtboolcCst|||dduS)zWhether or not the given field info represents a model at the database level. This will return `True` for cases where the field represents a list of models or a single model. rdN)rt)rirerfrOrOrPrg:srgrTypeGuard[type[PrismaModel]]cCsddlm}t||S)Nrr!)basesr" issubclass)rr"rOrOrPrKBs  rKc@s(eZdZdZed ddZd ddZd S) r%rOrE str | NonecCdS)zbRender the node to a string None is returned if the node should not be rendered. NrOr{rOrOrPrXKzAbstractNode.renderrcCr)zIf True, rendering of the node is skipped Useful for some nodes as they should only actually be rendered if they have any children. TrOr{rOrOrP should_renderSrzAbstractNode.should_renderNrErrEr)rsrrrrrXrrOrOrOrPr%Hs  c@seZdZUdZded<ded<ded<ded<d Zd d d d d&ddZd'ddZd'ddZe d'ddZ d(ddZ d)ddZ e d*d+d$d%Zd S),Nodez+Base node handling rendering of child nodesr4joinerr r2r[list[ChildType]children)rr r[r z N)rr rlist[ChildType] | NonerErFcCs.||_||_||_|dur||_dSg|_dSN)r[rr r)rMr[rr rrOrOrPrQksz Node.__init__rcCr)zlGet the string used to enter the node. This string will be rendered *before* the children. NrOr{rOrOrPentersz Node.entercCr)zlGet the string used to depart the node. This string will be rendered *after* the children. NrOr{rOrOrPdepartzrz Node.departcCs|sdSg}|}|dur|||jD]}d}t|tr$|}n|}|r3|t||jq|}|durA|||j |S)aRender the node and it's children and to string. Rendering a node involves 4 steps: 1. Entering the node 2. Rendering it's children 3. Departing the node 4. Joining the previous steps together into a single string N) rrappendrryr4rXr rrjoin)rMstringsZenteredchildcontentZdepartedrOrOrPrXs$      z Node.renderr ChildTypecCs|j|dS)z Add a childN)rr)rMrrOrOrPr^szNode.addcCsgS)zCreate the node's children If children are passed to the constructor, the children returned from this method are used to extend the already set children. rOr{rOrOrPcreate_childrenszNode.create_childrenQueryBuilder | Nonekwargsr'Node'cKs.|d||di|}|j||S)zvCreate the node and its children This is useful for subclasses that add extra attributes in __init__ r[NrO) setdefaultrextendr)clsr[rnoderOrOrPr's z Node.create) r[r2rr4r r4rrrErFr)rrrErFrErr)r[rrrrEr)rsrr__doc__rrrQrrrrXr^r classmethodr'rOrOrOrPr\s"     #  rcsFeZdZdZdZed ddZed ddZed fd d ZZ S) r]zsRendered node examples: query { } or mutation { } rOrEr4cCs|jjdS)Nz {)r[r7r{rOrOrPrszRootNode.entercCrN}rOr{rOrOrPrzRootNode.departcst}|s td|S)NzCould not generate query.)superrXrurMr __class__rOrPrXs zRootNode.renderr) rsrrrrrrrrX __classcell__rOrOrrPr]s   r]csReZdZdZdZddfd d Zedd d ZedddZedddZ Z S)r_z~Rendered node examples: result: findUniqueUser or result: executeRaw rOr r4rrrErFc stjdd|i|dS)Nr rO)rrQ)rMr rrrOrPrQszResultNode.__init__cCs6|jj}|dur |j}nd}|jjj|d}d|S)Nrrkzresult: )r[r9rHr5format)rMr9Z model_namer3rOrOrPrs  zResultNode.enterrcCsdSrrOr{rOrOrPrrzResultNode.departrcCstj|j|jjdgS)Nrx) Argumentsr'r[r=r{rOrOrPrs zResultNode.create_children)r)r r4rrrErFrrr) rsrrrrrQrrrrrrOrOrrPr_s   r_csjeZdZUdZded<dZdfd d Zedd d ZedddZ edddZ eddddZ Z S)rzRendered node example: ( key1: "1" key2: "["John","123"]" key3: true key4: { data: true } ) r<r=rxrrrErFc tjdi|||_dSNrO)rrQr=)rMr=rrrOrPrQ zArguments.__init__rcC t|jSrrrr{rOrOrPr" zArguments.should_renderr4cCr)N(rOr{rOrOrPr&rzArguments.entercCr)N)rOr{rOrOrPr*rzArguments.departNr:rc Csg}|jD]T\}}|durqt|tr%|t|tj|j|ddqt|t rO|jj dvr?||dt t |q|t|t j|j|ddq||dt |q|S)NrUrrG: ) r=roryrzrKeyDatar'r[r&r3rTListNode)rMr=rargr~rOrOrPr.s     zArguments.create_children)r=r<rrrErFrrr)r=r:rEr rsrrrrrrQrrrrrrrOrOrrPr s     rcsZeZdZUdZded<dZdfd d Zedd d ZedddZ edddZ Z S)rzwRendered node example: { key1: "a" key2: 3 key3: [ "name" ] } Mapping[str, Any]rUrrrrErFc rrrrQrU)rMrUrrrOrPrQXrz Data.__init__r4cCrN{rOr{rOrOrPr\rz Data.entercCrrrOr{rOrOrPr`rz Data.departrc Csg}|jD]=\}}t|tr |t|tj|j|ddqt|t t t fr8|t|t j|j|ddq||dt |q|S)Nrrr)rUroryrzrrrr'r[listtuplesetrrT)rMrr}r~rOrOrPrds   zData.create_children)rUrrrrErFrr) rsrrrrrrQrrrrrrOrOrrPrHs    rcsXeZdZUded<dZddfd d Zedd dZedddZedddZ Z S)r Iterable[Any]rUr, rr4rrrErFc s tjdd|i|||_dS)NrrOr)rMrUrrrrOrPrQxs zListNode.__init__cCr)N[rOr{rOrOrPr|rzListNode.entercCr)N]rOr{rOrOrPrrzListNode.departrcCsDg}|jD]}t|tr|tj|j|dq|t|q|S)Nr)rUryrzrrr'r[rT)rMritemrOrOrPrs   zListNode.create_children)r)rUrrr4rrrErFrr) rsrrrrrQrrrrrrOrOrrPrss   rcseZdZUdZded<ded<ded<dZ ddfdd ZedddZedddZ edddZ edddZ Z S) r`aRepresents field selections Example no include: { id name } Example include={'posts': True} { id name posts { id title } } Example include={'posts': {'where': {'title': {'contains': 'Test'}}}} { id name posts( where: { title: { contains: 'Test' } } ) { id title } } r8r9r:r;r>r?r\NrrrErFc s(tjdi|||_||_||_dSr)rrQr9r;r?)rMr9r;r?rrrOrPrQs zSelection.__init__rcCrrrr{rOrOrPrrzSelection.should_renderr4cCrrrOr{rOrOrPrrzSelection.entercCrrrOr{rOrOrPrrzSelection.departrc Cs|j}|j}|j}g}|j}|dur||n |dur$||||dur|dur0td|D]X\}}|durR|t |dt j |d|j ||dddq4t |tr~|}|dd} |t |dtj ||d dt j || |j ||ddgq4|d urq4td t|d |S) Nz)Cannot include fields when model is None.T )rqri)r;r9)seprr;rrxFz0Expected `bool` or `dict` include value but got z instead.)r9r;r[r?rrp ValueErrorrorrr`r'rwryrzcopyrJr TypeErrortype) rMr9r;r[rr?r}r~rNZnested_includerOrOrPrsZ      zSelection.create_children)NNN) r9r8r;r:r?r>rrrErFrrrrrOrOrrPr`s$ '   r`c@sHeZdZUdZded<ded<ded<dZddd d Zedd dZdS)rz3Node for rendering a child node with a prefixed keyr4r}rrr)r}rrrrErFcCs||_||_||_dSr)r}rr)rMr}rrrOrOrPrQ%s z Key.__init__cCs2|j}|r|j|j|S|j|jSr)rrXr}rrrOrOrPrX*s z Key.renderN)r)r}r4rrrr4rErFr) rsrrrrrrQrrXrOrOrOrPrs  robjrr#cCs(t|r|}nt|}td|d)z@Single dispatch generic function for serializing objects to JSONzType z not serializable)inspectisclassrr)rtyprOrOrP serializer2s rdtdatetime.datetimecCsT|jdur |jtjd}n |jtjkr|tj}|jt|jddd}|S)zxFormat a datetime object to an ISO8601 string with a timezone. This assumes naive datetime objects are in UTC. N)tzinfoi) microsecond)rreplacer utc astimezoneintr isoformat)rrOrOrPserialize_datetime=s   r fields.JsoncCr)aLSerialize a Json wrapper to a json string. This is used as a hook to override our default behaviour when building queries which would treat data like {'hello': 'world'} as a Data node when we instead want it to be rendered as a raw json string. This should only be used for fields that are of the `Json` type. )rTrUrrOrOrPserialize_jsonNs r fields.Base64cCt|S)z4Serialize a Base64 wrapper object to raw binary datar4rrOrOrPserialize_base64[rdecimal.DecimalcCr)z&Serialize a Decimal object to a stringrrrOrOrPserialize_decimalarrrcKs*|dt|ddtj|fi|S)Ndefault ensure_asciiF)rrjsonrT)rrrOrOrPrTgs  rT)rirrer4rfrrEr8)rirrer4rfrrEr)rrrEr)rrrEr#)rrrEr4)rrrEr4)rrrEr4)rrrEr4)rrrrrEr4)\ __future__rrdecimalrloggingdatetimeabcrrtypingrrrrr r r r textwrapr functoolsrZtyping_extensionsrrrZpydanticrZpydantic.fieldsrrr_typesrerrorsrrr_compatrrrrrZ_typingr _constantsr rr"Z PrismaModelrr# getLoggerrsr$rr4rrrrr&r0r1objectrnr6r2rtrgrKr%rr]r_rrrr`rrregisterrZJsonrZBase64rDecimalrrTrOrOrOrPs$                         K b%*;+