- d Z ddlmZ ddlmZ ddlmZ ddlZddlZddlZddlZddl Z ddl
m Z ddl m
Z
ddl mZ dd lmZ ddlZd
Zd Zd Zd
Z ed ez
ZdZdZdZdZd Z d Z d Z d Z d Z G d de! Z"ddZ#y) zFUtilities for handling YAML schemas for gcloud export/import commands. )absolute_import)division)unicode_literalsN)log)resource_projector) yaml_printer)filesz6A gcloud export/import command YAML validation schema.P ( z
description: zz Optional.z Output only.z Required.c t | t z z
}t j |t t
| }t
| dk7 rdj | S |d }|t
| t z z
}|dk r|dz |dz z t z S |S )a; Returns description: |- text wrapped so it won't exceed _WIDTH at depth.
The YAML representer doesn't seem to take the length of the current tag
into account when deciding whether to inline strings or use |-. In this case
the tag is always "description: ". This function detects when YAML would fail
and adds temporary marker lines to produce the desired output. The marker
lines are removed prior to final output.
Args:
depth: The nested dict depth.
text: The text string to wrap.
Returns:
Text wrapped so it won't exceed _WIDTH at depth.
r )
_WIDTH_INDENTtextwrapwrapmax_MINWRAPlenjoin_DESCRIPTION_INDENT_YAML_WORKAROUND)depthtextwidthlineslinenudges 2lib/googlecloudsdk/command_lib/util/apis/export.py_WrapDescriptionr" 1 s EGO
$%
--c(E 2
3%Z1_ 99U q$
3t9 22
3%
QY $; $'7 77
+ c t j | j }t j d| ryt j d| ryt j d| ry|dk( ry|dk( ry|dk( ry |d
k( ry |S ) z6Returns the JSON schema normalized type name for name.z.?int64integerz.?int32z^int\d*$floatnumberdoubleboolbooleanbytesstring)six text_typelowerrematch)namess r! _NormalizeTypeNamer4 O sy mmD !!XXj! XXj! XXk1 '\ (] &[ '\
(r# c | j di j di j di j di j dd }|dk( S )a? Returns whether the field represents a google.protobuf.Struct message.
google.protobuf.Struct is the following message:
message Struct {
// Unordered map of dynamically typed values.
map fields = 1;
}
In apitools, this corresponds to a message with an additionalProperties
field containing a list of AdditionalProperty messages each of which holds a
key/value pair, where the value is a extra_types.JsonValue message.
Args:
field: A message spec field dict.
Returns:
True iff the field is a google.protobuf.Struct.
fieldsadditionalPropertiesvaluetype JsonValue)get)fieldmaybe_map_value_types r! _IsProtobufStructFieldr? c sc $ C
"&&)c' C2 "s b "
,,r# c g }t j | D ]7 \ }}|d }|dk7 s|j t s'|j | 9 |S )zReturns the list of required field names in fields.
Args:
fields: A message spec fields dict.
Returns:
The list of required field names in fields.
descriptionr7 )r- iteritems
startswith _REQUIREDappend)r6 requiredr2 r8 rA s r! _GetRequiredFieldsrG ~ sU (]]6*kdE
&K %%+*@*@*Kood +
/r# c < t | }|rt | | d< yy)z Adds required fields to spec.rF N)rG sorted)specr6 rF s r! _AddRequiredFieldsrK s#
'(
h'D r# c 6 e Zd ZdZd dZd Zd Zd Zd Zd Z y)
ExportSchemasGeneratorz>Recursively generates export JSON schemas for nested messages.Nc > || _ || _ t | _ y N)_api
_directoryset
_generated)selfapi directorys r! __init__z ExportSchemasGenerator.__init__ s DI DOeDOr# c |dz S )z4Returns the schema file name given the message name.z.yaml )rT message_names r! _GetSchemaFileNamez)ExportSchemasGenerator._GetSchemaFileName s
' !!r# c | j | }| j r*t j j | j | }|S )z9Returns the schema file path name given the message name.)r[ rQ ospathr )rT rZ file_paths r! _GetSchemaFilePathz)ExportSchemasGenerator._GetSchemaFilePath s7 '' 5I '',,t :i r# c t j }|j d t j dt j | j | t j dj t d|j }| j | }t j dj | t! j" | 5 }|j | ddd y# 1 sw Y yxY w)z;Writes the schema in spec to the _GetSchemaFilePath() file.z4$schema: "http://json-schema.org/draft-06/schema#"
yaml)r2 projectoroutz
*{}
r z Generating JSON schema [{}].N)ioStringIOwriter YamlPrinterr IdentityProjectorPrintr0 subformatr getvaluer` r infor
FileWriter)rT rZ rJ tmpcontentr_ ws r! _WriteSchemaz#ExportSchemasGenerator._WriteSchema s
++-CIIFG
$668 tffZ&&'78$ OG'' 5IHH
+
2
29
=> ) $ggg
% $ $s !C<<Dc |dz
}t t j | D ]H \ }}|d j }|j t
r!|t
t
d j }n5|j t r |t
t d j }|j t rt j }|||< t || |d< |j d r6d|d< t j |j di } | |d< | }|dz
}t |j d t }
|j dd } |j d i } |
ru|d
k( r6||= t j }
| j |||
| |
s|
||< t! | rd |d< | j# | |d < | j% | | | | j& v r| j# | |d < t) | } | d
k( rL|j d }d|d< t t j | D cg c] \ }}| c}} |d
< D| |d< K yc c}}w )a Adds message fields to the YAML spec.
Args:
depth: The nested dict depth.
parent: The parent spec (nested ordered dict to add fields to) of spec.
spec: The nested ordered dict to add fields to.
fields: A message spec fields dict to add to spec.
r rA Nrepeatedarrayr9 itemsr* r6 r7 objectz$refenumchoicesr, )rI r- rB striprC _OPTIONALr rD _OUTPUT_ONLY collections OrderedDictr" r<
isinstancestr
_AddFieldsr? r[ GeneraterS r4 )rT r parentrJ r6 r2 r8 rA drw is_message_field type_name subfields
propertiesry n_s r! r z!ExportSchemasGenerator._AddFields s>
QJE cmmF 34 e-(..0k *!#i./288: ! !) ,!#i./288: -
!
!
#ad4j)%=a
: & '' '2(>?'
$EIIf$5s;))FI.i))Hb)i ) )4j"..0*
//%J
:
%F4L
#E
* !F) --i8!F)
-- 9
- '--i8!F)(3)
&
99Y'D AfI cmmD.A B.Ada.A BCAfI!AfIm 5h Cs I1 c || j v ry| j j | t j }dj | j
j | j
j | |d< t |d< d|d< t || d|d< t j }||d < dd
i}t j }||d < d|d< d |d< d|d< t j }||d < t j | |d
< t j | |d< t j | |d< t j | |d< t j | |d< t j }||d< d|d< d|d< ||d< | j d||| | j || y)aq Recursively generates export/import YAML schemas for message_spec.
The message and nested messages are generated in separate schema files in
the destination directory. Pre-existing files are silently overwritten.
Args:
message_name: The API message name for message_spec.
message_spec: An arg_utils.GetRecursiveMessageSpec() message spec.
Nz{} {} {} export schematitlerA rx r9 Fr7 r r, COMMENTz-User specified info ignored by gcloud import.z template-idregiondateversionUNKNOWNrv z+Unknown API fields that cannot be imported.rw r ) rS addr~ r rl rP r2 r _SPEC_DESCRIPTIONrK r rs ) rT rZ message_specrJ r type_stringcommentcomment_propertiesunknowns r! r z ExportSchemasGenerator.Generate s t& OO % " " $D,33 ))<9DM+D DLt\*#(D ((*J#D 8$K %%'G #Jy GFO LGM&+G "#$002 .GL(3(?(? (L}%#.#:#:;#Gx (3(?(? (L}%!,!8!8!Ev $/$;$;K$Hy! %%'G #Jy GFO JGM"GGOOAtZ 6lD)r# rO )
__name__
__module__ __qualname____doc__rW r[ r` rs r r rY r# r! rM rM s% F
" @"D6*r# rM c < t | | j || y)a Recursively generates export/import YAML schemas for message_spec in api.
The message and nested messages are generated in separate schema files in the
current directory. Pre-existing files are silently overwritten.
Args:
api: An API registry object.
message_name: The API message name for message_spec.
message_spec: An arg_utils.GetRecursiveMessageSpec() message spec.
directory: The path name of the directory to place the generated schemas,
None for the current directory.
N)rM r )rU rZ r rV s r! GenerateExportSchemasr 8 s i(11, Mr# rO )$r
__future__r r r r~ re r] r0 r googlecloudsdk.corer googlecloudsdk.core.resourcer r googlecloudsdk.core.utilr r- r r r r r r r r| r} rD r" r4 r? rG rK rx rM r rY r# r! r s M & ' # ; 5 *
M
/*W4 ' < (-6.(Y*V Y*x
Nr#