o tf0@sddlmZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z ddl mZddlmZmZmZgdZGd d d ejd ZGd d d eZGdddeZGdddeZd ddZd!ddZdS)") annotationsN)Request)ResponseMountRoute)ProviderProtocol)create_file_responsecreate_resource_identifierguess_media_type)Resource FileResourceDirectoryResourceContentResourcecreate_resource_routecreate_resourcec@sJeZdZdZ ddddZedd d Zedd dZej dddZ dS)r z,A resource that can be served by a provider.NproviderrheadersNone | dict[str, str]cCs |pi|_||_tj|_dS)aCreate a new Resource. Parameters ---------- provider : ProviderProtocol The provider that will serve this resource. headers : dict[str, str], optional Additional headers to include in the response. N)r _provideruuiduuid4hex_guid)selfrrrZ/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/servir/_resources.py__init__ s zResource.__init__returnstrcCs|jS)z(The unique identifier for this resource.)rrrrrguid2sz Resource.guidcCs|jjd|jS)zThe URL for this resource.z /resources/)rurlr!r rrrr"7sz Resource.urlrequestrrcCsdS)zGet the resource. Parameters ---------- request : Request The request to handle. Returns ------- Response The response to send. Nr)rr#rrrget<sz Resource.getN)rrrr)rrr#rrr) __name__ __module__ __qualname____doc__rpropertyr!r"abcabstractmethodr$rrrrr s   r ) metaclasscs(eZdZd fdd Zdd d ZZS)r path pathlib.Pathkwargs typing.Anyc sDtjdi||sJd||_t||jd|_dS)NzPath must be a filedataidr) superris_file_pathr resolveas_posixnamerrr/r1 __class__rrrNs  zFileResource.__init__r#rrrcCs&t|j|jd}|j|j|S)Nrange)r r8rr$update)rr#responserrrr$VszFileResource.getr/r0r1r2r&)r'r(r)rr$ __classcell__rrr=rr Msr cs,eZdZdZdfdd Zdd d ZZS)rz Serve a directory as a resource.r/r0r1r2c sJtjdi||std||_t||j dd|_ dS)a^Create a new DirectoryResource. Parameters ---------- path : pathlib.Path The path to the directory to serve. kwargs : dict Additional keyword arguments to pass to the Resource constructor. Raises ------ ValueError If the path is not a directory. zPath must be a directory/r3Nr) r6ris_dir ValueErrorr8r r9r:r;lstriprr<r=rrr_s   zDirectoryResource.__init__r#rrrcCsH|jd}|j||jdd}t||jd}|j|j|S)Nr/rDr?) path_paramsr8replacerrGr rr$r@)rr# full_pathresolvedrArrrr$zs zDirectoryResource.getrBr&r'r(r)r*rr$rCrrr=rr\srcs0eZdZdZ ddfd d ZdddZZS)rz&Serve a string or bytes as a resource.Ncontent str | bytes extension None | strr1r2c sJtjdi|||_|durt|trdnd}t|d|d|_dS)aQCreate a new ContentResource. Parameters ---------- content : str | bytes The content to serve. extension : str, optional The extension to use for the resource identifier. kwargs : dict Additional keyword arguments to pass to the Resource constructor. Nz.txtz.binrNr3r)r6r_content isinstancerr r)rrNrPr1r=rrrs zContentResource.__init___rrrcCst|jt|j|jdS)N)rN media_typer)rrRr rrcopy)rrTrrrr$s zContentResource.getr%)rNrOrPrQr1r2)rTrrrrMrrr=rrs r resourcestyping.Mapping[str, Resource]rrcs*dfdd }tdtd|d d gd gd S)zCreate a route for serving resources. Parameters ---------- resources : typing.Mapping[str, Resource] A mapping of resource identifiers to resources. Returns ------- Mount A route for serving resources. r#rrrcs&|jd}|dd}||S)Nr/rDr)rIsplitr$)r#r/r!rWrrendpoints z'create_resource_route..endpointz /resourcesz /{path:path}GETHEAD)methods)r/ZroutesNr&r)rWr[rrZrrs rxpathlib.Path | strrrr1r2cKst|trt|fd|i|St|tjr#|r#t|fd|i|St|tjr7|r7t|fd|i|St dt |)aCreate a resource from a path or string. Parameters ---------- x : pathlib.Path | str The path or string to create a resource from. provider : ProviderProtocol The provider that will serve the resource. kwargs : dict Additional keyword arguments to pass to the Resource constructor. Returns ------- Resource The created resource. rzUnsupported type: ) rSrrpathlibPathr7r rEr TypeErrortype)r_rr1rrrrs r)rWrXrr)r_r`rrr1r2rr ) __future__rr,ratypingrZstarlette.requestsrZstarlette.responsesrZstarlette.routingrrZservir._protocolsrZ servir._utilr r r __all__ABCMetar r rrrrrrrrs"     0&