!{SrSSKJr SSKJr SSKJr SSKJrJrJrJ r J r J r SSK J r SSKJr SSKJr SSKJr SS KJr SFS \S \S \S\ \S\ \S\ \S\ \S\ \S\ \S\\\44SjjrSFS \S \S \S\ \S\ \S\ \S\ \S\ \S\ \S\\\44SjjrSGSSSSS\S\ \S\ \S\4 SjjrSrSrSHS jrS!\4S"jrSISSSSS \S \S \S\ \S\ \S\ \S\ \S#\ \\\4S$\ \ \S\ \S\ \\\4\44S%jjrSIS \S\ \S\ \S\ \S\ \S#\ \\\4S$\ \ \S\ \4S&jjr S \S'\S(\S\ \S\\\44 S)jr!S*r"S+r#SGS,jr$S-r%S.r&S/r'S0r(S1r)S2\"S3S4\ \4S \ \4/54S5jr*S6r+S7r,S8\S9\S:\S;\S\\\44 S<jr-S=\ R\S\R^R`S\\\44S>jr1S=\ R\S\R^R`S\S\SS4 S?jr2S=\ R\S\R^R`SS4S@jr3S=\ R\S\R^R`SS4SAjr4SB\ R\S\R^RjSC\SS4SDjr6SB\ R\S\R^RjSS4SEjr7g )Jz,The BigQuery CLI reservation client library.)absolute_import)division)print_function)AnyDict NamedTupleOptionalSetTuple) discovery)utils)bq_error) bq_id_utilsN api_versionslotsignore_idle_slotstarget_job_concurrencymulti_region_auxiliaryautoscale_max_slots max_slots scaling_modereservation_group_namereturnc 20n XS'X*S'UbXZS'UbXJS'Ub 0U S'XjSS'UbX:S'[R"Xv5(a[R"S5eUbUbUcUb[R"S 5eUbXzS'UbXS 'U bXS 'U $) aYReturn the request body for CreateReservation. Arguments: api_version: The api version to make the request against. slots: Number of slots allocated to this reservation subtree. ignore_idle_slots: Specifies whether queries should ignore idle slots from other reservations. edition: The edition for this reservation. target_job_concurrency: Job concurrency target. multi_region_auxiliary: Whether this reservation is for the auxiliary region. autoscale_max_slots: Number of slots to be scaled when needed. max_slots: The overall max slots for the reservation. scaling_mode: The scaling mode for the reservation. reservation_group_name: The reservation group name which the reservation belongs to. Returns: Reservation object that was created. Raises: bq_error.BigqueryError: if requirements for parameters are not met. slot_capacityrr concurrency autoscalereditionAmax_slots and autoscale_max_slots cannot be set at the same time.z8max_slots and scaling_mode must be set at the same time.rreservation_group)frontend_utilsValidateAtMostOneSelectedr BigqueryError) rrrrrrrrrr reservations )platform/bq/clients/client_reservation.pyGetBodyForCreateReservationr&sF+!&o%6!"',B()'!7 $!K ,? [) $ --iMM  K  4L4  B ( ".''=#$ c  [UUUUUUUU U U 5 n SUR<SUR<3n UR5R 5R 5R XURS9R5$)aCreate a reservation with the given reservation reference. Arguments: client: The client used to make the request. api_version: The api version to make the request against. reference: Reservation to create. slots: Number of slots allocated to this reservation subtree. ignore_idle_slots: Specifies whether queries should ignore idle slots from other reservations. edition: The edition for this reservation. target_job_concurrency: Job concurrency target. multi_region_auxiliary: Whether this reservation is for the auxiliary region. autoscale_max_slots: Number of slots to be scaled when needed. max_slots: The overall max slots for the reservation. scaling_mode: The scaling mode for the reservation. reservation_group_name: The reservation group name which the reservation belongs to. Returns: Reservation object that was created. Raises: bq_error.BigqueryError: if autoscale_max_slots is used with other version. projects/ /locations/)parentbody reservationId) r& projectIdlocationprojects locations reservationscreater-execute)clientr referencerrrrrrrrrr$r+s r%CreateReservationr7\sP,   + &  ooy{|~ v9P9Pwyr'r5.r6 page_size page_tokenr cnSUR<SUR<3nUbMUR5R5R 5R UUUSU-S9R 5$UR5R5R 5R XRUS9R 5$)aList reservations in the project and location for the given reference. Arguments: client: The client used to make the request. reference: Reservation reference containing project and location. page_size: Number of results to show. page_token: Token to retrieve the next page of results. reservation_group: When specified, only list reservations in the given reservation group. Returns: Reservation object that was created. r)r*zreservation_group=%s)r+pageSize pageTokenfilterr+r;r<)r.r/r0r1r2listr4)r5r6r8r9r r+s r%ListReservationsr@s* &"    ),==    ooy{|~ t6tDwy r'cSUR<SUR<S3nUR5R5R US9R 5nU$)a List BI reservations in the project and location for the given reference. Arguments: client: The client used to make the request. reference: Reservation reference containing project and location. Returns: List of BI reservations in the given project/location. r)r*z/biReservationname)r.r/r0r1getBiReservationr4)r5r6r+responses r%ListBiReservationsrFsU &  oo!!#44&4AIIK  /r'cUR5R5R5RUR 5S9R 5$)zGets a reservation with the given reservation reference. Arguments: client: The client used to make the request. reference: Reservation to get. Returns: Reservation object corresponding to the given id. rB)r0r1r2getpathr4r5r6s r%GetReservationrKs; ooy{|~ s  s!wy r'cUR5R5R5RUR 5S9R 5 g)zDeletes a reservation with the given reservation reference. Arguments: client: The client used to make the request. reference: Reservation to delete. rBN)r0r1r2deleterIr4rJs r%DeleteReservationrNsA //,,.55 >> 6 GIr'reservation_sizec"UR5RS5(aUSSR5(aUSSnOeUR5RS5(aUSSR5(aUSSnO#UR5(aUnO [S5e[ U5S-S-S-n0nSnX$S 'US - nUR 5R 5RUR5XTS 9R5$) aUpdates a BI reservation with the given reservation reference. Arguments: client: The client used to make the request. reference: Reservation to update. reservation_size: size of reservation in GBs. It may only contain digits, optionally followed by 'G', 'g', 'GB, 'gb', 'gB', or 'Gb'. Returns: Reservation object that was updated. Raises: ValueError: if reservation_size is malformed. GBNGzInvalid reservation size. The unit for BI reservations is GB. The specified reservation size may only contain digits, optionally followed by G, g, GB, gb, gB, or Gb.isizezsize,rC updateMaskr,) upperendswithisdigit ValueErrorintr0r1updateBiReservationrIr4)r5r6rOreservation_digitsbi_reservation update_masks r%UpdateBiReservationrbs ''-- 3B  ' ' ) ))#2.'',,1A#21F1N1N1P1P)#2.!!) , --+,t3d:TA.++ + ooy{~~Kwy r' labels_to_setlabel_keys_to_removec h0n Sn Ub X!K$a!#k+.A+{+++k\!k%)B':%%0:k( k(9K%# H% %  !# &$#))+ U#(k(C ,::  K #6#B005HA5M  N ( <K".?"K''=#$''K  !!r'c [UUUUUUUUUU U U 5 upUR5R5R5R UR 5XS9R 5$)aBUpdates a reservation with the given reservation reference. Arguments: client: The client used to make the request. api_version: The api version to make the request against. reference: Reservation to update. slots: Number of slots allocated to this reservation subtree. ignore_idle_slots: Specifies whether queries should ignore idle slots from other reservations. target_job_concurrency: Job concurrency target. autoscale_max_slots: Number of slots to be scaled when needed. max_slots: The overall max slots for the reservation. scaling_mode: The scaling mode for the reservation. reservation_group_name: The reservation group name that the reservation belongs to. Returns: Reservation object that was updated. Raises: bq_error.BigqueryError: if autoscale_max_slots is used with other version. rW)rmr0r1r2patchrIr4)r5rr6rrrrrrrcrdrr$ras r%UpdateReservationrpsqJ;   + ooy{|~ u).."{uMwy r'plan renewal_plancj0nX7S'XGS'XWS'UbXgS'UbX'S'SUR<SUR<3nSn UR(aURS :wa URn UR5R 5R 5R UUU S 9n U R5$) aCreate a capacity commitment. Arguments: client: The client used to make the request. reference: Project to create a capacity commitment within. edition: The edition for this capacity commitment. slots: Number of slots in this commitment. plan: Commitment plan for this capacity commitment. renewal_plan: Renewal plan for this capacity commitment. multi_region_auxiliary: Whether this commitment is for the auxiliary region. Returns: Capacity commitment object that was created. slot_countrqrrNrrr)r* )r+r,capacityCommitmentId)r.r/rvr0r1capacityCommitmentsr3r4) r5r6rrrqrrrcapacity_commitmentr+capacity_commitment_idrequests r%CreateCapacityCommitmentr{s.&+l# $f(4n%'4J01 %, " & ## (F(F#(M&;; ooy{ v"5    r'cSUR<SUR<3nUR5R5R 5R XBUS9R 5$)a=Lists capacity commitments for given project and location. Arguments: client: The client used to make the request. reference: Reference to the project and location. page_size: Number of results to show. page_token: Token to retrieve the next page of results. Returns: list of CapacityCommitments objects. r)r*r>)r.r/r0r1rwr?r4)r5r6r8r9r+s r%ListCapacityCommitmentsr}sS &  ooy{ t6tDwy r'cUR5R5R5RUR 5S9R 5$)zGets a capacity commitment with the given capacity commitment reference. Arguments: client: The client used to make the request. reference: Capacity commitment to get. Returns: Capacity commitment object corresponding to the given id. rB)r0r1rwrHrIr4rJs r%GetCapacityCommitmentr+s= ooy{ s  s!wy r'cUR5R5R5RUR 5US9R 5 g)zDeletes a capacity commitment with the given capacity commitment reference. Arguments: client: The client used to make the request. reference: Capacity commitment to delete. force: Force delete capacity commitment, ignoring commitment end time. )rCforceN)r0r1rwrMrIr4)r5r6rs r%DeleteCapacityCommitmentr>sC //335<< >> 5= GIr'cnUcUc[R"S5e0n/nUbX$S'URS5 UbX4S'URS5 UR5R 5R 5R UR5SRU5US9R5$)aUpdates a capacity commitment with the given reference. Arguments: client: The client used to make the request. reference: Capacity commitment to update. plan: Commitment plan for this capacity commitment. renewal_plan: Renewal plan for this capacity commitment. Returns: Capacity commitment object that was updated. Raises: bq_error.BigqueryError: if capacity commitment cannot be updated. z$Please specify fields to be updated.rqrr,rW) rr#appendr0r1rwrorIjoinr4)r5r6rqrrrxras r%UpdateCapacityCommitmentrKs \l*  !G HH+ "&v*6'~& ooy{ u~~XXk*"  wy r'c@Uc[R"S5eSU0nUR5R5R 5R UR 5US9R5nSU;dSU;a[R"S5eUSUS/$)aSplits a capacity commitment with the given reference into two. Arguments: client: The client used to make the request. reference: Capacity commitment to split. slots: Number of slots in the first capacity commitment after the split. Returns: List of capacity commitment objects after the split. Raises: bq_error.BigqueryError: if capacity commitment cannot be updated. z#Please specify slots for the split. slotCountrCr,firstsecondzinternal error)rr#r0r1rwsplitrIr4)r5r6rr,rEs r%SplitCapacityCommitmentrrs$ ]  !F GG u $ ooy{ u).."u.wy   H 8  !1 22 7 Xh/ 00r'cpU(d[R"S5eU(d[R"S5eUb[U5S:a[R"S5eSU<SU<3nSU0nUR5R 5R 5R XES9R5$) auMerges capacity commitments into one. Arguments: client: The client used to make the request. project_id: The project ID of the resources to update. location: Capacity commitments location. capacity_commitment_ids: List of capacity commitment ids. Returns: Merged capacity commitment. Raises: bq_error.BigqueryError: if capacity commitment cannot be merged. zproject id must be specified.zlocation must be specified.z2at least 2 capacity commitments must be specified.r)r*capacityCommitmentIdsr+r,)rr#lenr0r1rwmerger4)r5 project_idr/capacity_commitment_idsr+r,s r%MergeCapacityCommitmentsrs"   !@ AA   !> ??$,C(Dq(H  < )I*4X >& !#: ;$ ooy{ uFu&wy r'c0nU(d[R"S5eX&S'U(aX6S'U(d[R"S5eU(d[R"S5eUR5<SU<3US'UR5R 5R 5R 5RUR5US9R5$) a~Creates a reservation assignment for a given project/folder/organization. Arguments: client: The client used to make the request. reference: Reference to the project reservation is assigned. Location must be the same location as the reservation. job_type: Type of jobs for this assignment. priority: Default job priority for this assignment. assignee_type: Type of assignees for the reservation assignment. assignee_id: Project/folder/organization ID, to which the reservation is assigned. Returns: ReservationAssignment object that was created. Raises: bq_error.BigqueryError: if assignment cannot be created. job_type not specified.job_typepriorityassignee_type not specified.assignee_id not specified.s/assigneer) rr#lowerr0r1r2 assignmentsr3rIr4)r5r6rr assignee_type assignee_idreservation_assignments r%CreateReservationAssignmentrs*   !: ;;'/$ )1:&   !? @@   != >>($  ooy{|~{} vY^^%,BvCwy r'cUR5R5R5R5R UR 5S9R 5 g)zDeletes given reservation assignment. Arguments: client: The client used to make the request. reference: Reference to the reservation assignment. rBN)r0r1r2rrMrIr4rJs r%DeleteReservationAssignmentrsK //,,.::<CC >> D GIr' id_fallbacksIDSrc[R"UUUSS9nSUR50nUR5R 5R 5R 5RUR5US9R5$)z=Moves given reservation assignment under another reservation.F)r identifierdefault_locationcheck_reservation_project destinationIdr) bq_client_utilsGetReservationReferencerIr0r1r2rmover4)r5rr6destination_reservation_idr!destination_reservation_referencer,s r%MoveReservationAssignmentrsy'6&M&M+' % '# <AAC D$ ooy{|~{} t!t-wy r'c0nSnUbU(dSnX#S'US- nUR5R5R5R5R UR 5UUS9R 5$)aGUpdates reservation assignment. Arguments: client: The client used to make the request. reference: Reference to the reservation assignment. priority: Default job priority for this assignment. Returns: Reservation assignment object that was updated. Raises: bq_error.BigqueryError: if assignment cannot be updated. rUJOB_PRIORITY_UNSPECIFIEDrz priority,rW)r0r1r2rrorIr4)r5r6rrras r%UpdateReservationAssignmentrs}+  +h)1:&;K ooy{|~{} u~~ %  wy r'cUR5R5R5R5R UR 5X#S9R 5$)aILists reservation assignments for given project and location. Arguments: client: The client used to make the request. reference: Reservation reference for the parent. page_size: Number of results to show. page_token: Token to retrieve the next page of results. Returns: ReservationAssignment object that was created. r>)r0r1r2rr?rIr4)r5r6r8r9s r%ListReservationAssignmentsr6sD ooy{|~{} t9>>#itNwy r'r/rrrcU(d[R"S5eU(d[R"S5eU(d[R"S5eU(d[R"S5eUR5<SU<3nSU-nSU-nUR5R 5R XvS9R 5nS U;aUS Hn U S U:XdMU s $ [R"S 5e) aSearches reservations assignments for given assignee. Arguments: client: The client used to make the request. location: location of interest. job_type: type of job to be queried. assignee_type: Type of assignees for the reservation assignment. assignee_id: Project/folder/organization ID, to which the reservation is assigned. Returns: ReservationAssignment object if it exists. Raises: bq_error.BigqueryError: If required parameters are not passed in or reservation assignment not found. zlocation not specified.rrrrz assignee=%szprojects/-/locations/%s)r+queryrjobTypez Reservation assignment not found)rr#rr0r1searchAllAssignmentsr4) r5r/rrrrrr+rE assignments r%SearchAllReservationAssignmentsrNs(   !: ;;   !: ;;   !? @@   != >>',,. <( ( "% $x /& ooy{67wy  h}- I ( *. ABBr'reservation_group_clientcSUR<SUR<3n0nUR5R5R 5R UUUR S9R5$)aACreates a reservation group with the given reservation group reference. Arguments: reservation_group_client: The client used to make the request. reference: Reservation group to create. Returns: Reservation group object that was created. Raises: bq_error.BigqueryError: if reservation group cannot be created. r)r*)r+r,reservationGroupId)r.r/r0r1reservationGroupsr3rr4)rr6r+r s r%CreateReservationGroupr}sm$ &'')y{ v &99 wy r'cSUR<SUR<3nUR5R5R 5R XBUS9R 5$)a~Lists reservation groups in the project and location for the given reference. Arguments: reservation_group_client: The client used to make the request. reference: Reservation group reference containing project and location. page_size: Number of results to show. page_token: Token to retrieve the next page of results. Returns: Reservation group object that was created. r)r*r>)r.r/r0r1rr?r4)rr6r8r9r+s r%ListReservationGroupsrsU$ & '')y{ t6tDwy r'cUR5R5R5RUR 5S9R 5$)zGets a reservation group with the given reservation group reference. Arguments: reservation_group_client: The client used to make the request. reference: Reservation group to get. Returns: Reservation group object corresponding to the given id. rB)r0r1rrHrIr4rr6s r%GetReservationGrouprs?'')y{ s  s!wy r'cUR5R5R5RUR 5S9R 5 g)zDeletes a reservation group with the given reservation group reference. Arguments: reservation_group_client: The client used to make the request. reference: Reservation group to delete. rBN)r0r1rrMrIr4rs r%DeleteReservationGrouprsD##%//1CCELL >> M GIr' apiclientpolicyc[R"U[RRSS9 SU0nUR 5R 5R 5RX1R5S9R5$)a~Sets IAM policy for the given reservation resource. Arguments: apiclient: the apiclient used to make the request. reference: the ReservationReference for the reservation resource. policy: The policy string in JSON format. Returns: The updated IAM policy attached to the given reservation resource. Raises: BigqueryTypeError: if reference is not a ReservationReference. SetReservationIAMPolicymethodr)r,resource) r typecheckApiClientHelperReservationReferencer0r1r2 setIamPolicyrIr4)rr6rrzs r%rrsl$!!66 & v 'y{|~|>>+;|<wy r'c[R"U[RRSS9 UR 5R 5R 5RUR5S9R5$)aJGets IAM policy for the given reservation resource. Arguments: apiclient: the apiclient used to make the request. reference: the ReservationReference for the reservation resource. Returns: The IAM policy attached to the given reservation resource. Raises: BigqueryTypeError: if reference is not a ReservationReference. GetReservationIAMPolicyr)r) rrrrr0r1r2 getIamPolicyrIr4)rr6s r%rrsb !!66 & y{|~|Y^^-|.wy r')NNNN)N)r6.)NNNNNN)8__doc__ __future__rrrtypingrrrr r r googleapiclientr clientsr rfrontendr!rrstrr]boolr&r7r@rFrKrNrbrmrpr{r}rrrrrrrrrrrResourcerReservationGroupReferencerrrrrrrr'r%rs2&%>>%,,*.#"&,0GG GG %SM G %TN G"#G}G3-G%SMG #s(^Gf*.#"&,0@@  @  @%SM@%TN@"#@}@3-@%SM@ #s(^@P(, + +++ +  } +  +\(&  ,S,p*.#"&.2/3,0l" l"l"l"  l"  l" %SM l""#l"}l"3-l"DcN+l"#3s8,l"%SMl" 4S>3 l"l*.#"&.2/3,099 %SM 9"#9}93-9DcN+9#3s8,9%SM9x0  0  0  0%TN0 #s(^0f2& $N1B!H,^  8C= ) HSM * @!H0,C,C%(,C9<,CKN,C #s(^,C^ '00 **DD  #s(^ F'00**DD   <'00**DD , '00 **DD   !!**??  B!!**?? r'