9SrSSKJr SSKJr SSKJr SSKrSSKJr SSKrSSK J r SSK J r SSK r SS KJr "S S \ 5r"S S \ 5r"SS\ 5r"SS\ 5r"SS\5r"SS\ R,"\R.\55rSr"SS\5r"SS\5r"SS\5r"SS\5rSrg) aProvides helper methods for dealing with Cloud Spanner Writes API. The main reasons for adding the util functions for Writes API are as below: - API expects column values to be extra_types.JsonValue, apitool cannot handle it by default. - for different data types the API expects different formats, for example: for INT64, API expects a string value; for FLOAT64, it expects a number. As the values user input are strings by default, the type conversion is necessary. )absolute_import)division)unicode_literalsN) OrderedDict) extra_types)Error)zipc\rSrSrSrSrg)BadColumnNameError'zDRaised when a column name entered by user is not found in the table.N__name__ __module__ __qualname____firstlineno____doc____static_attributes__r 4lib/googlecloudsdk/command_lib/spanner/write_util.pyr r 'sLrr c\rSrSrSrSrg)BadTableNameError+zFRaised when a table name entered by user is not found in the database.r Nrr rrrr+sNrrc\rSrSrSrSrg)InvalidKeysError/zARaised when the number of keys user input does not match the DDL.r Nrr rrrr/sIrrc\rSrSrSrSrg)InvalidArrayInputError3zGRaised when the user tries to input a list as a value in the data flag.r Nrr rrrr3sOrrc\rSrSrSr\R "S\R\R-5r Sr Sr \ S5r SrSrg ) _TableColumn7zA wrapper that stores the column information. Attributes: name: String, the name of the table column. col_type: _ScalarColumnType or _ArrayColumnType. a # A column definition has a name and a type, with some additional # properties. # Some examples: # Foo INT64 NOT NULL # Bar STRING(1024) # Baz ARRAY [`]?(?P\w+)[`]?\s+ (?P[\w<>]+) # We don't care about "NOT NULL", and the length number after STRING # or BYTES (e.g.STRING(MAX), BYTES(1024)). cXlX lgNnamecol_type)selfr&r's r__init___TableColumn.__init__Ls IMrctURUR:H=(a URUR:H$r$r%r(others r__eq___TableColumn.__eq__Ps' 99 " Ft}}'FFrcURRU5nU(d[SRU55eUR S5n[ R UR S55n[X45$)a|Constructs an instance of _TableColumn from a column_def DDL statement. Args: column_ddl: string, the parsed string contains the column name and type information. Example: SingerId INT64 NOT NULL. Returns: A _TableColumn object. Raises: ValueError: invalid DDL, this error shouldn't happen in theory, as the API is expected to return valid DDL statement strings. zInvalid DDL: [{}].r&type)_COLUMN_DDL_PATTERNsearch ValueErrorformatgroup _ColumnTypeFromDdlr!)cls column_ddl column_match column_namer's rr8_TableColumn.FromDdlSsj**11*=L  +22:> ??$$V,K""<#5#5f#=>H  ..rc8URRU5$)aConvert the user input values to JSON value or JSON array value. Args: value: String or string list, the user input values of the column. Returns: extra_types.JsonArray or extra_types.JsonValue, the json value of a single column in the format that API accepts. )r' GetJsonValuer(values r GetJsonValues_TableColumn.GetJsonValuesjs == % %e ,,r)r'r&N)rrrrrrecompileDOTALLVERBOSEr2r)r. classmethodr8rBrr rrr!r!7sS    YY # %G//, -rr!cV\rSrSrSrSrSr\S5r\ RS5r Sr g) r7waRA wrapper that stores the column type information. A column type can be one of the scalar types such as integers, as well as array. An array type is an ordered list of zero or more elements of scalar type. Attributes: scalar_type: String, the type of the column or the element of the column (if the column is an array). ) BOOLBYTESDATEFLOAT64INT64STRING TIMESTAMPNUMERICJSON TOKENLISTFLOAT32cXlgr$ scalar_type)r(rXs rr)_ColumnType.__init__s"rcSnURH nX1;dM Un O U(d[SRU55eURS5(a [ U5$[ U5$)aConstructs a _ColumnType object from a partial DDL statement. Args: column_type_ddl: string, the parsed string only contains the column type information. Example: INT64 NOT NULL, ARRAY or BYTES(200). Returns: A _ArrayColumnType or a _ScalarColumnType object. Raises: ValueError: invalid DDL, this error shouldn't happen in theory, as the API is expected to return valid DDL statement strings. Nz$Invalid DDL: unrecognized type [{}].ARRAY) _SCALAR_TYPESr4r5 startswith_ArrayColumnType_ScalarColumnType)r9column_type_ddl scalar_match data_types rr8_ColumnType.FromDdlsrL&&  %  '   0 7 7 H JJ!!'** l ++ | ,,rc[5er$)NotImplementedErrorr@s rr?_ColumnType.GetJsonValues  rrWN) rrrrrr\r)rHr8abcabstractmethodr?rr rrr7r7wsB K-#--:  rr7c2US:Xa[R"SS9$US:Xa'UR5S:Hn[R"US9$US;a7US;a[R"US 9$[R"[U5S 9$[R"US 9$) aConvert the user input scalar value to JSON value. Args: scalar_type: String, the scalar type of the column, e.g INT64, DATE. scalar_value: String, the value of the column that user inputs. Returns: An API accepts JSON value of a column or an element of an array column. NULLT)is_nullrKTRUE) boolean_value)rNrU)NaNInfinityz -Infinity) string_value) double_value)r JsonValueupperfloat)rX scalar_value bool_values rConvertJsonValueForScalarTypesrwsV   ..f##%/J  z ::,,77  " " ==  " "l0C DD  l ;;rc4^\rSrSrU4SjrSrSrSrU=r$)r_c,>[[U] U5 gr$)superr_r)r(rX __class__s rr)_ScalarColumnType.__init__s T+K8rcbURUR:H=(a [U[5$r$)rX isinstancer_r,s rr._ScalarColumnType.__eq__s-   u00 0 "Z  6""rc.[URU5$r$)rwrXr@s rr?_ScalarColumnType.GetJsonValues )$*:*:E BBrr rrrrr)r.r?r __classcell__r}s@rr_r_s9"CCrr_c4^\rSrSrU4SjrSrSrSrU=r$)r^c,>[[U] U5 gr$)r{r^r)r|s rr)_ArrayColumnType.__init__s D*;7rcbURUR:H=(a [U[5$r$)rXrr^r,s rr._ArrayColumnType.__eq__s-   u00 0 !Z 6!!rc [R"[R"UVs/sHn[URU5PM snS9S9$s snf)N)entries) array_value)rrr JsonArrayrwrX)r(valuesvs rr?_ArrayColumnType.GetJsonValuesM  ))IO3 IOA *4+;+;Q ?3    3 s A r rrs@rr^r^s8!  rr^c\rSrSrSrSrSrg)ColumnJsonDataaContainer for the column name and value to be written in a table. Attributes: col_name: String, the name of the column to be written. col_value: extra_types.JsonArray(array column) or extra_types.JsonValue(scalar column), the value to be written. cXlX lgr$col_name col_value)r(rrs rr)ColumnJsonData.__init__s MNrrN)rrrrrr)rr rrrrs rrc\rSrSrSr\R "S\R\R-5r S Sjr Sr Sr \ S5rS rS rS rg) TableaIContainer for the properties of a table in Cloud Spanner database. Attributes: name: String, the name of table. _columns: OrderedDict, with keys are the column names and values are the _TableColumn objects. _primary_keys: String list, the names of the primary key columns in the order defined in the DDL statement a # Every table starts with "CREATE TABLE" followed by name and column # definitions, in a big set of parenthesis. # For example: # CREATE TABLE Foos ( # Bar INT64 NOT NULL, # Baz INT64 NOT NULL, # Qux STRING(MAX), # ) CREATE\s+TABLE\s+ (?P[\w\.]+)\s+\(\s+ (?P.*)\)\s+ # Then, it has "PRIMARY KEY" and a list of primary keys, in parens: # PRIMARY KEY ( Bar, Qux ) PRIMARY\s+KEY\s*\( (?P.*)\) # It may have extra instructions on the end (e.g. INTERLEAVE) to # tell Spanner how to store the data, but we don't really care. Nc<XlX lU=(d /Ulgr$)r&_columns _primary_keys)r( table_namerrs rr)Table.__init__sIM&,"Drc/n[R"U5HAup4URU5nURU5nUR [ X655 MC U$)aAGet the column names and values to be written from data input. Args: data_dict: Dictionary where keys are the column names and values are user input data value, which is parsed from --data argument in the command. Returns: List of ColumnJsonData, which includes the column names and values to be written. )six iteritems_FindColumnByNamerBappendr)r( data_dict column_listrr col_in_tablecol_json_values r GetJsonDataTable.GetJsonDatasZK"}}Y7++H5l#11).D.DS.IJ.I779.IJ :,= >>58 >EE  /2 4 55 Ks2Ec[5n[R"UR5Hup#URX'M U$)z[Maps the column name to the column type. Returns: OrderedDict of column names to types. )rrrrr')r( col_to_typer&rs rGetColumnTypesTable.GetColumnTypess6 -K dmm4  //k5 rcURU$![aM SR[URR 555n[ SR X55ef=f)zFind the _TableColumn object with the given column name. Args: col_name: String, the name of the column. Returns: _TableColumn. Raises: BadColumnNameError: the column name is invalid. rz6Column name [{}] is invalid. Valid column names: [{}].)rKeyErrorrlistkeysr r5)r(rvalid_column_namess rrTable._FindColumnByNamesd- ]]8 $$ -99T$--*<*<*>%?@  B I I , ---s AA()rrr&r$)rrrrrrDrErFrGrr)rrrHr8rrrr rrrrs]zz $ 99rzz !'#*- (<0505d -rrcUR5n[R"U5HMup4X#n[U[5(dM [U[ 5SLdM5[ SRX455e U$)aChecks array input is valid. Args: table: Table, the table which data is being modified. data: OrderedDict, the data entered by the user. Returns: data (OrderedDict) the validated data. Raises: InvalidArrayInputError: if the input contains an array which is invalid. FzgColumn name [{}] has an invalid array input: {}. `--flags-file` should be used to specify array values.)rrrrr^rrr5)tabledatarrrAr's rValidateArrayInputrsu$$&+}}T*mf"H("$$(25$(?5(H " 44:F64I KK + +r)r __future__rrrrg collectionsrrDapitools.base.pyrgooglecloudsdk.core.exceptionsrr six.movesr r rrrobjectr!with_metaclassABCMetar7rwr_r^rrrr rrrs '' # (0 MMOOJuJPUP=-6=-@4 #$$S[[&94 n<: C C  {  V f-Ff-Rr