U Zf3H@s$ddlmZmZmZmZddlmZddlmZddl Z ddl m Z m Z m Z mZmZdZerddlZddl mZmZmZmZmZmZmZmZdd lmZe jrejejd fZ nejejd fZ Gd d d e!Z"Gddde!Z#Gdd d e!Z$Gddde!Z%Gddde!Z&Gddde!Z'dS))absolute_importdivisionprint_functionunicode_literals) OrderedDict) StrictVersionN) doc_unwrapis_aliasis_composite_type is_list_typeis_nullable_typeF)Alias AnnotationAnnotationTypeDataTypeListNullableStruct UserDefined) AstRouteDef ApiNamespaces ApiNamespacec@s0eZdZdZddZddZddZdd Zd S) ApizL A full description of an API's namespaces, data types, and routes. cCst||_t|_d|_dSN)rversionr namespaces route_schema)selfrr0/tmp/pip-unpacked-wheel-vqb3mfeu/stone/ir/api.py__init__.s z Api.__init__cCs"||jkrt||j|<|j|S)z Only creates a namespace if it hasn't yet been defined. :param str name: Name of the namespace. :return ApiNamespace: )rrrnamerrrensure_namespace4s zApi.ensure_namespacecCsJt}t|jD]}|j|||<q||_|jD] }|q8dS)zj Alphabetizes namespaces and routes to make spec parsing order mostly irrelevant. N)rsortedrkeysvalues normalize)rZordered_namespacesZnamespace_name namespacerrrr'As z Api.normalizecCs|jdkst||_dSr)rAssertionError)rrrrradd_route_schemaPszApi.add_route_schemaN)__name__ __module__ __qualname____doc__r r#r'r*rrrrr*s  rc@seZdZdZddZdS) _ImportReasonz5 Tracks the reason a namespace was imported. cCsd|_d|_d|_d|_dS)NF)alias data_type annotationannotation_typerrrrr [sz_ImportReason.__init__Nr+r,r-r.r rrrrr/Vsr/c@seZdZdZddZddZddZdd Zd d Zd d Z ddZ d$ddZ ddZ ddZ ddZddZd%ddZddZdd Zd!d"Zd#S)&rzQ Represents a category of API endpoints and their associated data types. cCsX||_d|_g|_i|_i|_g|_i|_g|_i|_g|_ i|_ g|_ i|_ i|_ dSr)r"docroutes route_by_nameroutes_by_name data_typesdata_type_by_namealiases alias_by_name annotationsannotation_by_nameannotation_typesannotation_type_by_name_imported_namespacesr!rrrr hszApiNamespace.__init__cCsHt|tjstt|t|d}|jdkr6||_n|j|7_dS)aUAdds a docstring for this namespace. The input docstring is normalized to have no leading whitespace and no trailing whitespace except for a newline at the end. If a docstring already exists, the new normalized docstring is appended to the end of the existing one with two newlines separating them.  N) isinstancesix text_typer)typer r6)rZ docstringZnormalized_docstringrrradd_doczs   zApiNamespace.add_doccCsT|j||jdkr"||j|j<|j|jkrrIr?r")rr2rrradd_annotations zApiNamespace.add_annotationcCs|j|||j|j<dSr)r@rIrAr")rr3rrradd_annotation_types z ApiNamespace.add_annotation_typeFcCsP|j|jkstd|j|t}|r.d|_|r8d|_|rBd|_|rLd|_dS)a Keeps track of namespaces that this namespace imports. Args: namespace (Namespace): The imported namespace. imported_alias (bool): Set if this namespace references an alias in the imported namespace. imported_data_type (bool): Set if this namespace references a data type in the imported namespace. imported_annotation (bool): Set if this namespace references a annotation in the imported namespace. imported_annotation_type (bool): Set if this namespace references an annotation in the imported namespace, possibly indirectly (by referencing an annotation elsewhere that has this type). zNamespace cannot import itself.TN) r"r)rB setdefaultr/r0r1r2r3)rr(Zimported_aliasZimported_data_typeZimported_annotationZimported_annotation_typereasonrrradd_imported_namespacesz#ApiNamespace.add_imported_namespacecs4gtfddjD] }|q"S)ad Returns a list of all data types used in the namespace. Because the inheritance of data types can be modeled as a DAG, the list will be a linearization of the DAG. It's ideal to generate data types in this order so that composite types that reference other composite types are defined in the correct order. csJ|kr dS|jkrdSt|r2|jr2|j||dSr)r(r Z parent_typerIaddr1rOZlinearized_data_typesZseen_data_typesrrrrOs   z8ApiNamespace.linearize_data_types..add_data_type)setr:rNrrYrlinearize_data_typess   z!ApiNamespace.linearize_data_typescs4gtfddjD] }|q"S)z Returns a list of all aliases used in the namespace. The aliases are ordered to ensure that if they reference other aliases those aliases come earlier in the list. csF|kr dS|jkrdSt|jr.|j||dSr)r(r r1rIrWr0rQZlinearized_aliasesZ seen_aliasesrrrrQs    z1ApiNamespace.linearize_aliases..add_alias)rZr<rPrr]rlinearize_aliasess   zApiNamespace.linearize_aliasescCs0t}|jD]}|||O}q t|dddS)a Returns a list of all user-defined data types that are referenced as either an argument, result, or error of a route. If a List or Nullable data type is referenced, then the contained data type is returned assuming it's a user-defined type. cSs|jSrr"dtrrr z6ApiNamespace.get_route_io_data_types..key)rZr7!get_route_io_data_types_for_router$)rr:rLrrrget_route_io_data_typess z$ApiNamespace.get_route_io_data_typescCsZt}|j|j|jfD]>}t|s*t|r6|}|j}qt|sFt|r|}| |q|S)zV Given a route, returns a set of its argument/result/error datatypes. ) rZ arg_data_typeresult_data_typeerror_data_typer r r1r r rW)rrLr:ZdtypeZdata_list_typeZdata_user_typerrrrfs z.ApiNamespace.get_route_io_data_types_for_routecCsrg}|jD]N\}}|r"|js"q|s:|js:|js:|js:q|sR|jsR|jsR|jsRq||q|jddd|S)a Returns a list of Namespace objects. A namespace is a member of this list if it is imported by the current namespace and a data type is referenced from it. Namespaces are in ASCII order by name. Args: must_have_imported_data_type (bool): If true, result does not include namespaces that were not imported for data types. consider_annotations (bool): If false, result does not include namespaces that were only imported for annotations consider_annotation_types (bool): If false, result does not include namespaces that were only imported for annotation types. Returns: List[Namespace]: A list of imported namespaces. cSs|jSrr_nrrrrb@rcz6ApiNamespace.get_imported_namespaces..rd)rBitemsr1r0r3r2rIsort)rZmust_have_imported_data_typeZconsider_annotationsZconsider_annotation_typesZimported_namespacesZimported_namespacerUrrrget_imported_namespacess.  z$ApiNamespace.get_imported_namespacescCsJt|ddd}t}|D]}|j|kr||jqt|dddS)a Returns a list of Namespace objects. A namespace is a member of this list if it is imported by the current namespace and has a data type from it referenced as an argument, result, or error of a route. Namespaces are in ASCII order by name. cSs|jSrr_r`rrrrbLrczBApiNamespace.get_namespaces_imported_by_route_io..rdcSs|jSrr_rkrrrrbQrc)r$rgrZr(rW)rZnamespace_data_typesZreferenced_namespacesr1rrr#get_namespaces_imported_by_route_ioCs z0ApiNamespace.get_namespaces_imported_by_route_iocCsL|jjddd|jjddd|jjddd|jjddddS)zQ Alphabetizes routes to make route declaration order irrelevant. cSs|jSrr_)rLrrrrbXrcz(ApiNamespace.normalize..rdcSs|jSrr_rXrrrrbYrccSs|jSrr_r\rrrrbZrccSs|jSrr_)r2rrrrb[rcN)r7rnr:r<r>r4rrrr'SszApiNamespace.normalizecCstd|jS)NzApiNamespace({!r}))strformatr"r4rrr__repr__]szApiNamespace.__repr__N)FFFF)FFF)r+r,r-r.r rHrMrOrQrRrSrVr[r^rgrfrorpr'rsrrrrrcs0  "  & c@speZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZdS)ApiRoutez% Represents an API endpoint. cCs@||_||_||_d|_d|_d|_d|_d|_d|_d|_ dS)z :param str name: Designated name of the endpoint. :param int version: Designated version of the endpoint. :param ast_node: Raw route definition from the parser. N) r"r _ast_node deprecatedraw_docr6rhrirjattrs)rr"rZast_noderrrr gs zApiRoute.__init__cCs2||_||_t||_||_||_||_||_dS)a] Converts a forward reference definition of a route into a full definition. :param DeprecationInfo deprecated: Set if this route is deprecated. :param str doc: Description of the endpoint. :type arg_data_type: :class:`stone.data_type.DataType` :type result_data_type: :class:`stone.data_type.DataType` :type error_data_type: :class:`stone.data_type.DataType` :param dict attrs: Map of string keys to values that are either int, float, bool, str, or None. These are the route attributes assigned in the spec. N)rvrwr r6rhrirjrx)rrvr6rhrirjrxrrrset_attributes~s zApiRoute.set_attributescCs$|jdkr|jSd|j|jSdS)z Get user-friendly representation of the route. :return: Route name with version suffix. The version suffix is omitted for version 1. rz{}:{}N)rr"rrr4rrrname_with_versions zApiRoute.name_with_versioncCsd|S)Nz ApiRoute({}))rrrzr4rrrrsszApiRoute.__repr__cCspt|tstd|t|ts0td||j|jksH|j|jkrLdS|j|jksd|j|jkrhdSdSdS)Nz Expected ApiRoute for object: {}rr)rDrt TypeErrorrrr"r)rlhsrhsrrr_compares  zApiRoute._comparecCs|||dkSNrrrotherrrr__lt__szApiRoute.__lt__cCs|||dkSrrrrrr__gt__szApiRoute.__gt__cCs|||dkSrrrrrr__eq__szApiRoute.__eq__cCs|||dkSrrrrrr__le__szApiRoute.__le__cCs|||dkSrrrrrr__ge__szApiRoute.__ge__cCs|||dkSrrrrrr__ne__szApiRoute.__ne__c Cs4t|j|j|j|j|j|j|j|j|j t |j f Sr) hashr"rrurvrwr6rhrirjidrxr4rrr__hash__szApiRoute.__hash__N)r+r,r-r.r ryrzrsrrrrrrrrrrrrrtbs  rtc@seZdZdddZdS)DeprecationInfoNcCs(|dkst|tstt|||_dS)zR :param ApiRoute by: The route that replaces this deprecated one. N)rDrtr)reprby)rrrrrr szDeprecationInfo.__init__)N)r+r,r-r rrrrrsrc@seZdZdZddZdS)rJzD Represents routes of different versions for a common name. cCs i|_dS)zR :param at_version: The dict mapping a version number to a route. N)rKr4rrrr szApiRoutesByVersion.__init__Nr5rrrrrJsrJ)( __future__rrrr collectionsrZdistutils.versionrrEr:r r r r r Z_MYPYtypingrrrrrZ DataTypeListrrrZstone.frontend.astrPY3DictTextZ NamespaceDictobjectrr/rrtrrJrrrrs(  ( , i