o SDi-j@sdddlZddlmZddlmZmZddlmZmZddlm Z ddl m Z m Z m Z mZmZddlZddlZddlZddlZddlmZddlmZdd lmZdd lmZdd lmZmZdd lmZm Z m!Z!dd l"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)ddl*m+Z+m,Z,ddl-m.Z.m/Z/ddl0m1Z1m2Z2ddl3m4Z4m5Z5ddl6m7Z7m8Z8m9Z9ddl:m;Z;e rddlededZ?edeedZ@ededZAdZBeCZDeCZEeFZGe dPiedddGdd d ZHe dPied!diGd"d#d#e e?e@fZIdd$d%e?d&e/d'ee@dfd(eIe?e@ffd)d*ZJGd+d,d,ejKjZLd%e?d(e?fd-d.ZMd/eAd(eAfd0d1ZNGd2d3d3e e?e@fZOd4ePejKjfd5d6ZQd/ed7e d8eeRedfd9eeCdfd:d;d&e/f deCd?e9d@efdAdBZT dQdCdDd4eedEeCdFeUeCdGeeCdfdHeVd(ePejKjf dIdJZWd4eedKe1d(eejKjdffdLdMZXd/ed(eVfdNdOZYdS)RN) defaultdict)CallableSequence)copydeepcopy) dataclass) TYPE_CHECKINGAnyGenericUnioncast) APIRouter)APIRoute) BaseModel) BaseRoute)TypeVar assert_never)DATACLASS_SLOTSSentinellenient_issubclass) CadwynErrorRouteAlreadyExistsError/RouteByPathConverterDoesNotApplyToAnythingError8RouteRequestBySchemaConverterDoesNotApplyToAnythingError9RouteResponseBySchemaConverterDoesNotApplyToAnythingErrorRouterGenerationErrorRouterPathParamsModifiedError) _add_request_and_response_paramsgenerate_versioned_models)Version VersionBundle)Endpoint VersionType)_AlterRequestByPathInstruction_AlterResponseByPathInstruction)EndpointDidntExistInstructionEndpointExistedInstructionEndpointHadInstruction) VersionChange) Dependant_Call.)bound_R_WR)r+default_RouteT_CADWYN_DELETED_ROUTET)frozeneqc@s"eZdZUeed<eeed<dS) _EndpointInfo endpoint_pathendpoint_methodsN)__name__ __module__ __qualname__str__annotations__ frozensetr<r<P/var/www/Datamplify/venv/lib/python3.10/site-packages/cadwyn/route_generation.pyr3>s r3r1c@s.eZdZUeeefed<eeefed<dS)GeneratedRouters endpointswebhooksN)r6r7r8dictr"r,r:r-r<r<r<r=r>Ds r>)r@routerversionsr@returncCs$|dur ttt}t|||SN)r r-r _EndpointTransformer transform)rBrCr@r<r<r=generate_versioned_routersJs rHc@seZdZdedefddZdS)VersionedAPIRouterendpointrDcCsRt|j|}|durtd|jdt|jvr!td|jd|jt|S)NzRoute not found on endpoint: "zE". Are you sure it's a route and decorators are in the correct order?z The route "z1" was already deleted. You can't delete it again.)_get_route_from_funcroutes LookupErrorr6_DELETED_ROUTE_TAGtagsrappend)selfrJrouter<r<r=only_exists_in_older_versionsVs    z0VersionedAPIRouter.only_exists_in_older_versionsN)r6r7r8r*rSr<r<r<r=rIUsrIcCst|}dd|jD|_|S)NcSsg|]}t|qSr<) copy_route.0rr<r<r= ezcopy_router..)rrL)rBr<r<r= copy_routercsrZrRcCs6t|ts t|St|}t|j|_t|j|_|SrE) isinstancerrr dependant dependencies)rR new_router<r<r=rTis   rTc seZdZdedededdffdd Zdeeeffdd Zd e d e d e fd dZ d e de e e eeeffdeedeeeffddZd e dee e e eeeffeeeeffddZdeeeeefd e fddZZS)rF parent_routerrCr@rDNcs<t||_||_||_t||_dd|jD|_dS)NcSs$g|]}t|trt|jvr|qSr<)r[rrNrOrVrRr<r<r=rX~s z1_EndpointTransformer.__init__..) super__init__r_rCparent_webhooks_routerrschema_generatorsrLroutes_that_never_existed)rQr_rCr@ __class__r<r=rbws   z_EndpointTransformer.__init__c Cst|j}t|j}i}i}|jD]>}|jt|jj||jt|jj|| ||j||||j<|||j<t|}t|}| |j |j |q|j r`t dtjjd|j t|jj D]V\}}t|tspqft|t|j}t|D]<} | j |} tt| } | jdurt| rt| jjdr| jjj} n| jj} nd} t| || | jdur| jj nd||jqqf|D] }dd|j D|_ q|D] }dd|j D|_ qt!||S)Nz.Every route you mark with @VersionedAPIRouter.z must be restored in one of the older versions. Otherwise you just need to delete it altogether. The following routes have been marked with that decorator but were never restored: __cadwyn_original_model__cS(g|]}t|tjjrt|jvs|qSr<r[fastapiroutingrrNrOr`r<r<r=rX  z2_EndpointTransformer.transform..cSrir<rjr`r<r<r=rXrm)"rZr_rcrCrdr9valueannotation_transformermigrate_router_to_version!_attach_routes_to_data_converters!_apply_endpoint_changes_to_routerrLrerrIrSr6 enumerater[rrrr\listvaluesr body_field_route_has_a_simple_body_schemahasattrtype_rh_add_data_migrations_to_routealiasr>) rQrBwebhook_routerrouterswebhook_routersversion route_index head_routecopy_of_dependant older_router older_routetemplate_older_body_modelr<r<r=rGsh              z_EndpointTransformer.transformrB head_routerrc Cs(||\}}}|jD]}g|j|jD]}|D] } ||||| qq|jD]*} | D]%} | js8q2t| j |} | rWt d|j d| j j ddd| Ddq2q.|j D]2} | D]-} | jshqbt| j |} | rtd|j d| j j dd d| Dd |j d| j j qbq^q dS) Nz"Request by body schema converter ".z9" failed to find routes with the following body schemas: cSg|]}|jqSr<r6rVmr<r<r=rXzJ_EndpointTransformer._attach_routes_to_data_converters..zV. This means that you are trying to apply this converter to non-existing endpoint(s). z&Response by response model converter "z<" failed to find routes with the following response models: cSrr<rrr<r<r=rXrz. This means that you are trying to apply this converter to non-existing endpoint(s). If this is intentional and this converter really does not apply to any endpoints, then pass check_usage=False argument to )?_extract_all_routes_identifiers_for_route_to_converter_matchingchanges#alter_response_by_path_instructionsru"alter_request_by_path_instructions _attach_routes_by_path_converter$alter_request_by_schema_instructions check_usagesetschemasrr6 transformer%alter_response_by_schema_instructionsr) rQrBrrpath_to_route_methods_mappinghead_response_modelshead_request_bodiesversion_changeby_path_convertersby_path_converterby_schema_convertersby_schema_convertermissing_modelsr<r<r=rqsj    z6_EndpointTransformer._attach_routes_to_data_convertersrrrc Cst}|jD]8}|||jvr9||j|D]!}|j|}t|tr-|jt||q|j t||qq| |q|rWt |j d|j d|jj dt|ddS)Nz "rz4" failed to find routes with the following methods: a . This means that you are trying to apply this converter to non-existing endpoint(s). Please, check whether the path and methods are correct. (hint: path must include all path variables and have a name that was used in the version that this VersionChange resides in))rmethodspathrLr[r$$_route_to_response_migration_mappingidrP#_route_to_request_migration_mappingaddr repr_namer6rrt) rQrrrrmissing_methodsmethodrrRr<r<r=rs*    z5_EndpointTransformer._attach_routes_by_path_converterc Cst}t}tdd}t|jD]F\}}t|trW|jdur+t|jtr+| |jt |rG|j durG|j j j }|durGt|trG| ||jD] }||j| |qJqdd|D} dd|D} || | fS)NcSsttSrE)rrr<r<r<r="szf_EndpointTransformer._extract_all_routes_identifiers_for_route_to_converter_matching..cSh|]}|jqSr<rh)rVmodelr<r<r= 0rzg_EndpointTransformer._extract_all_routes_identifiers_for_route_to_converter_matching..cSsh|]}t|d|qSr)getattr)rVbodyr<r<r=r1s)rrrsrLr[rresponse_modelrrrrwrv field_info annotationrr) rQrBresponse_modelsrequest_bodiesrindexrRrrrrr<r<r=rs"        zT_EndpointTransformer._extract_all_routes_identifiers_for_route_to_converter_matchingrLcCs|jD]e}|jD]^}t||j|j|jdd}t}|j}t|t rlt||j|j|jdd}|rYt} |D]} | | j O} q8t dt | d|jd|j ddd |D|D] } || j O}| jtq[d } nt|tr4|rt} |D]} | | j O} qyt dt | d|jd |j d d d |Dt||j|j|jdd}zt|Wn.ty} z"t dt |jd|jd |j dt| jddd | jD | d} ~ ww|D]S} || j O}| jtt|j| j| j | jj dd}t|dkr|j|dqt|dkr0|}t dt | j d| jd |j dt|ddd |D qd} nt|trO|D]} || j O}t|j || q.z}Endpoint "{endpoint_methods} {endpoint_path}" you tried to delete in "{version_change_name}" doesn't exist in a newer versionz" you tried to restore in "z" already existed in a newer version. If you really have two routes with the same paths and methods, please, use "endpoint(..., func_name=...)" to distinguish between them. Function names of endpoints that already existed: cSrr<rrUr<r<r=rXkrYz" has z applicable routes that could be restored. If you really have two routes with the same paths and methods, please, use "endpoint(..., func_name=...)" to distinguish between them. Function names of endpoints that can be restored: cSrr<rrUr<r<r=rX|rYra. applicable routes with the same function name and path that could be restored. This can cause problems during version generation. Specifically, Cadwyn won't be able to warn you when you deleted a route and never restored it. Please, make sure that functions for all these routes have different names: cSs"g|] }|jjd|jjqS)r)rJr7r6rUr<r<r=rXs"z}Endpoint "{endpoint_methods} {endpoint_path}" you tried to restore in "{version_change_name}" wasn't among the deleted routeszjEndpoint "{endpoint_methods} {endpoint_path}" you tried to change in "{version_change_name}" doesn't exist)r5r4version_change_name)ralter_endpoint_instructions _get_routesr4r5endpoint_func_namerrr[r%rrrtr6rOrPrNr&"_validate_no_repetitions_in_routesrlenrLremovererrJr'_apply_endpoint_had_instructionrformat)rQrLrr instructionoriginal_routes#methods_to_which_we_applied_changes)methods_we_should_have_applied_changes_todeleted_routes method_union deleted_routeoriginal_routeerrere method_diffr<r<r=rr6s                z6_EndpointTransformer._apply_endpoint_changes_to_router)r6r7r8r,r r-rbr>rGr rrqrA _RoutePath _RouteMethodr_RouteIdtyper(r r$r#rtupler rrtrrrr __classcell__r<r<rfr=rFvs. H/  ( rFrLcCsBi}|D]}t|jt|j}||vrt||||||<qdSrE)r3rr;rr)rL route_maprR route_infor<r<r=rs rrtemplate_body_fieldtemplate_body_field_name dependant_for_request_migrationsr)c Cs~|jjr|jjs!td|jjd|jjdt|jd|jd |j||||||jj|jj|jjd|j |_ |j |j_ dS)Nz#route.dependant.request_param_name=z&, route.dependant.response_param_name=z for route rz9 which should not be possible. Please, contact my author.)request_param_namebackground_tasks_param_nameresponse_param_name) r\rrrrtrr _versionedrrJcall)rRrrrrrCr<r<r=rzs*  rzrrrc Cs|jjD]c}t|j|}|turgt|||kr,td|dt|jd|jd|d |dkradd|jj D}t t d |}||krat d t|jd|jd |d t|d t|d t|||qdS)NzExpected attribute "z" of endpoint "rz" to be different in "zl", but it was the same. It means that your version change has no effect on the attribute and can be removed.rcSrr<)r{)rVpr<r<r=rrz2_apply_endpoint_had_instruction..z{(.*?)}zWhen altering the path of "z" in "z2", you have tried to change its path params from "z" to "a[". It is not allowed to change the path params of a route because the endpoint was created to handle the old path params. In fact, there is no need to change them because the change of path params is not a breaking change. If you really need to change the path params, you should create a new route with the new path params and delete the old one.) attributes__dataclass_fields__rrrrtrrr\ path_paramsrrefindallrsetattr)rrr attr_nameattroriginal_path_paramsnew_path_paramsr<r<r=rs:   rFrr4r5rrcs"dfdd|DS)N/cs`g|],}t|tjjr.|jdkr.t|jr.dus%|j j kr.t |j vkr|qS)rN) r[rkrlrrrstriprrissubsetrJr6rNrOr`rr5r4rr<r=rXs z_get_routes..)r)rLr4r5rrr<rr=rs rrJcCs.|D]}t|tjjr|j|kr|SqdSrE)r[rkrlrrJ)rLrJrRr<r<r=rK s rKcCst|jjdkS)Nr)rr\ body_params)rRr<r<r=rwsrwr<rE)Zr collectionsrcollections.abcrrrr dataclassesrtypingrr r r r fastapi.paramsrkfastapi.routingfastapi.security.base fastapi.utilsr rpydanticrstarlette.routingrtyping_extensionsrr cadwyn._utilsrrrcadwyn.exceptionsrrrrrrrcadwyn.schema_generationrrcadwyn.structurerr cadwyn.structure.commonr!r"cadwyn.structure.datar#r$cadwyn.structure.endpointsr%r&r'cadwyn.structure.versionsr(fastapi.dependencies.modelsr)r*r,r-r/rNr9rrintrr3r>rHrlrIrZrTrFrtrrrzrrboolrrKrwr<r<r<r=s      $           =    $