U ڲg(@sPdZddlmZdZddlmZddlmZGddde Z Gdd d e Z d S) aSchema processing for discovery based APIs Schemas holds an APIs discovery schemas. It can return those schema as deserialized JSON objects, or pretty print them as prototype objects that conform to the schema. For example, given the schema: schema = """{ "Foo": { "type": "object", "properties": { "etag": { "type": "string", "description": "ETag of the collection." }, "kind": { "type": "string", "description": "Type of the collection ('calendar#acl').", "default": "calendar#acl" }, "nextPageToken": { "type": "string", "description": "Token used to access the next page of this result. Omitted if no further results are available." } } } }""" s = Schemas(schema) print s.prettyPrintByName('Foo') Produces the following output: { "nextPageToken": "A String", # Token used to access the # next page of this result. Omitted if no further results are available. "kind": "A String", # Type of the collection ('calendar#acl'). "etag": "A String", # ETag of the collection. }, The constructor takes a discovery document in which to look up named schema. )absolute_importz$jcgregorio@google.com (Joe Gregorio)) OrderedDict)_helpersc@sZeZdZdZddZeddddZd d Zeddd d Z d dZ dddZ dS)SchemaszSchemas for an API.cCs|di|_i|_dS)zConstructor. Args: discovery: object, Deserialized discovery document from which we pull out the named schema. schemasN)getrpretty)selfZ discoveryr :/tmp/pip-unpacked-wheel-rfrvjkbl/googleapiclient/schema.py__init__JszSchemas.__init__NrcCsd|dkr g}||krd|S||||jkrRt|j|||d|j|j|<||j|S)aGet pretty printed object prototype from the schema name. Args: name: string, Name of schema in the discovery document. seen: list of string, Names of schema already seen. Used to handle recursive definitions. Returns: string, A string that contains a prototype object with comments that conforms to the given schema. Nz# Object with schema name: %sdent)appendr_SchemaToStructrto_str_prettyPrintByNamepop)r nameseenrr r r rVs    zSchemas._prettyPrintByNamecCs|j|gddddS)aGet pretty printed object prototype from the schema name. Args: name: string, Name of schema in the discovery document. Returns: string, A string that contains a prototype object with comments that conforms to the given schema. r)rrN)r)r rr r r prettyPrintByNamets zSchemas.prettyPrintByNamecCs"|dkr g}t|||d|jS)aoGet pretty printed object prototype of schema. Args: schema: object, Parsed JSON schema. seen: list of string, Names of schema already seen. Used to handle recursive definitions. Returns: string, A string that contains a prototype object with comments that conforms to the given schema. Nr)rrrr schemarrr r r _prettyPrintSchemas zSchemas._prettyPrintSchemacCs|j|ddddS)zGet pretty printed object prototype of schema. Args: schema: object, Parsed JSON schema. Returns: string, A string that contains a prototype object with comments that conforms to the given schema. rrNr)r)r rr r r prettyPrintSchemas zSchemas.prettyPrintSchemacCs|j||S)zGet deserialized JSON schema from the schema name. Args: name: string, Schema name. default: object, return value if name not found. )rr)r rdefaultr r r rsz Schemas.get)Nr)Nr)N) __name__ __module__ __qualname____doc__r util positionalrrrrrr r r r rGs     rc@s\eZdZdZeddddZddZdd Zd d Z d d Z ddZ ddZ ddZ dS)rz%Convert schema to a prototype object.rcCs(g|_d|_||_||_d|_||_dS)zConstructor. Args: schema: object, Parsed JSON schema. seen: list, List of names of schema already seen while parsing. Used to handle recursive definitions. dent: int, Initial indentation depth. N)valuestringrr from_cacherrr r r r s z_SchemaToStruct.__init__cCs|jd|j|dgdS)zaAdd text as a line to the output. Args: text: string, Text to output.   Nr%extendrr textr r r emitsz_SchemaToStruct.emitcCs|jd|j|gdS)ztAdd text to the output, but with no line terminator. Args: text: string, Text to output. r(Nr*r,r r r emitBeginsz_SchemaToStruct.emitBegincCsd|rPdd|jdd}|}dd|D}||}|j|d|dgn|j|dgdS) zAdd text and comment to the output with line terminator. Args: text: string, Text to output. comment: string, Python comment. r)r(r z# cSsg|] }|qSr )rstrip).0xr r r sz+_SchemaToStruct.emitEnd..z # N)r splitlinesjoinr%r+)r r-commentdividerlinesr r r emitEnds z_SchemaToStruct.emitEndcCs|jd7_dS)zIncrease indentation level.Nrr r r r indentsz_SchemaToStruct.indentcCs|jd8_dS)zDecrease indentation level.r:Nrr;r r r undentsz_SchemaToStruct.undentc Cs|d}|dkr|d|dd|d|kr~|di}tt|}|D] \}}|d|||qZn d|kr|d ||d|| d nd |kr |d }|dd}|j ||j d } | } || d || ddD]} | | qn~|dkrV|dd} |dt| |ddnH|dkr|dd} |dt| |ddn|dkr|dd} |dt| |ddn|dkr|dd} |dt| |ddn|dkr|d|ddn|dkr4|d|ddnj|dkr|d |d||d||d!|| d"n| d#||ddd|j|_|jS)$zPrototype object based on the schema, in Python code with comments. Args: schema: object, Parsed JSON schema file. Returns: Prototype object based on the schema, in Python code with comments. typeobject{ description propertiesz"%s": ZadditionalPropertiesz "a_key": z},z$ref)rrr:Nbooleanrz True or Falsez%s,r&zA Stringz"%s",integerZ42numberz3.14nullzNone,anyz"",array[itemsz],zUnknown type! %s)rr9r<rsortedrKr/ _to_str_implr=r.r'rr4r0strr5r%r&) r rstyperCZsorted_propertiesZpnameZpschemaZ schemaNamerAspartsliner%r r r rMsd                     z_SchemaToStruct._to_str_implcCs||_||jS)aPrototype object based on the schema, in Python code with comments. Args: from_cache: callable(name, seen), Callable that retrieves an object prototype for a schema with the given name. Seen is a list of schema names already seen as we recursively descend the schema definition. Returns: Prototype object based on the schema, in Python code with comments. The lines of the code will all be properly indented. )r'rMr)r r'r r r r0s z_SchemaToStruct.to_strN)r)rrr r!r"r#r r.r/r9r<r=rMrr r r r rs >rN) r! __future__r __author__ collectionsrZgoogleapiclientrr"r?rrr r r r s ,   c