:vjRUdZddlZddlZddlZddlmZddlmZddlm Z ddl m Z m Z ddl mZmZGd d e j ZGd d eZGd deZGddeZGddeZddZeezZeed<y)z resolvers.py --------------- Provides a common interface to load assets referenced by name like MTL files, texture images, etc. Assets can be from ZIP archives, web assets, or a local file path. N)Path) TypeAlias)urlparse)cachingutil)HttpSessionLikeMappingceZdZdZej dZej dZej defdZ ej defdZ ej dZ d efd Z d efd Z d ed efd Zy)Resolverz' The base class for resolvers. ctd)NzUse a resolver subclass!NotImplementedError)selfargskwargss b/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/trimesh/resolvers.py__init__zResolver.__init__!"<==ctNrrkeys rgetz Resolver.get s !##rnamectdNz`write` not implemented!rrrdatas rwritezResolver.write$rr namespacectd)Nz`namespaced` not implemented!rrr"s r namespacedzResolver.namespaced(s!"ABBrctd)Nz`keys` not implemented!rrs rkeysz Resolver.keys,s!";<> $$ >#>> CCCC ==s&s&"""rr cjeZdZdZddedefdZdZdeddfdZd ede fd Z d efd Z d ed ee zfd Z y)FilePathResolverz> Resolve files from a source path on the file system. sourceallow_anywherec6tjjtjjt |}t ||_||_tjj|r||_ n)tjj|}|d|_ tjj|jstd|jd||_ tjj||_y)uZ Resolve files based on a source path. Parameters ------------ source : str File path where mesh was loaded from allow_anywhere : bool If True allow assets to reference paths outside the resolver root, i.e. `../textures/thing.png` — the pre-5.0 behavior. rzpath `z `not a directory!N)ospath expanduserabspathr8r9r>cleanisdirparentsplit ValueError file_pathbasename file_name)rr=r>rDrGs rrzFilePathResolver.__init__?s""277??3v;#?@">2 77== DKGGMM%(E(DKww}}T[[)vdkk]2DEF F))&1rc#K|j}tj|jD]N\}}}|j|r|t |d}|D]$}tj j ||&Pyw)z List all files available to be loaded. Yields ----------- name : str Name of a file which can be accessed. N)rFr@walk startswithlenrAjoin)rrFrA_namesrs rr(zFilePathResolver.keysasi ggdkk2ND!Uv&CKM*ggll4.. 3sA>Br"r/cttjj|j||j S)aI Return a resolver which changes the root of the resolver by an added namespace. Parameters ------------- namespace : str Probably a subdirectory Returns -------------- resolver : FilePathResolver Resolver with root directory changed. )r>)r<r@rArPrFr>r$s rr%zFilePathResolver.namespacedss/ GGLLi 0ATAT  rrct|jj}||jz j}|js&|j |st d|d|ddz|S)a Resolve an asset name to an absolute path under the resolver root. Parameters ------------ name : str Name of an asset relative to the resolver root. Returns ------------ path : pathlib.Path Absolute resolved path. Raises ------------ ValueError If the path escapes the resolver root and `allow_anywhere` was not set. 'z' escapes resolver root 'u ' — pass z9`FilePathResolver(path, allow_anywhere=True)` to allow it)rrFresolvestripr>is_relative_torH)rrrFrAs rabsolutezFilePathResolver.absolutesz*dkk"**,%..0""4+>+>v+FD626(+FMN  rc|j|jjdtjj |df}|D]L} |j |}|js&t|d5}|jcdddcS|j |t|#t $rYvwxYw#1swYxYw)a Get an asset, restricted to the resolver root. Parameters ------------- name : str Name of the asset. Must resolve inside the resolver root. Returns ------------ data : bytes Loaded data from asset. /rbN) rWlstripr@rArGrYrHexistsopenreadFileNotFoundError)rr candidates candidaterAfs rrzFilePathResolver.gets JJL JJL   $ GGMM$  # $I }}Y/{{}$%668&% $ d%%  &%sB>C > C  C  C r ct|j|d5}tj||dddy#1swYyxYw)a Write an asset to a file path, restricted to the resolver root. Parameters ----------- name : str Name of the file to write. Must resolve inside the resolver root. data : str or bytes Data to write to the file. wb)file_objstuffN)r`rYr write_encoded)rrr res rr!zFilePathResolver.writes5$--%t ,    6- , ,s =AN)F)r2r3r4r5r8r9rr(r%rrYrbytesr!r:rrr<r<:sk 2s 2D 2D/$ C ,> &ST<&&@ 7# 7S5[ 7rr<cpeZdZdZddedzdedzfdZdZdeddfd Zd ede fd Z deddfd Z de fd Z y) ZipResolverz- Resolve files inside a ZIP archive. Narchiver"c||_t|tr(|jj ddz|_yd|_y)a{ Resolve files inside a ZIP archive as loaded by trimesh.util.decompress Parameters ------------- archive : dict Contains resources as file object namespace : None or str If passed will only show keys that start with this value and this substring must be removed for any get calls. r[N)rn isinstancer8rWrstripr")rrnr"s rrzZipResolver.__init__s: i %&__.55c:S@DN!DNrc|ja|j}t|}|jjDcgc]&}|j |rt||kDr||d(c}S|jjScc}w)z Get the available keys in the current archive. Returns ----------- keys : iterable Keys in the current archive. N)r"rOrnr(rN)rr"lengthks rr(zZipResolver.keyss >> %I^F **,,A<< *s1v&' ,  ||  "" s+Brr/cH|ji|_||j|<y)z Store a value in the current archive. Parameters ----------- key : hashable Key to store data under. value : str, bytes, file-like Value to store. N)rnr,s rr!zZipResolver.writes# << DL! SrrcJ|yt|dr|jd}|j}||vr#t||jD] }||vs|}n||}t |t tfr|S|jd|j}|jd|S)z Get an asset from the ZIP archive. Parameters ------------- name : str Name of the asset Returns ------------- data : bytes Loaded data from asset Ndecodeutf-8r) hasattrrwrn nearby_namesr"rprkr8seekra)rrrnoptionobjr s rrzZipResolver.gets <  4 ";;w'D,, w &tT^^<W$"D =dm cE3< (J  xxz   rc0t|j|S) Return a "sub-resolver" with a root namespace. Parameters ------------- namespace : str The root of the key to clip off, i.e. if this resolver has key `a/b/c` you can get 'a/b/c' with resolver.namespaced('a/b').get('c') Returns ----------- resolver : Resolver Namespaced resolver. )rnr")rmrnr$s rr%zZipResolver.namespaced@s 4<<9EErc@tj|jS)z Export the contents of the current archive as a ZIP file. Returns ------------ compressed : bytes Compressed data in ZIP format. )rcompressrnr's rexportzZipResolver.exportRs}}T\\**r)NN) r2r3r4r5dictr8rr(r!rkrr%rr:rrrmrmsp"t "sTz"(#, " " ",,,\FCFMF$ + +rrmcheZdZdZ ddededzdefdZdedefd Z defd Z d eddfd Z d Z dZ y) WebResolverz+ Resolve assets from a remote URL. Nurlsessiontimeoutc.t|dr|jd}t|}|jdvrt d|jd|ddl}|j dtd ||_||_ |d n&t|jjd d}|d k(r d |d|_ n'|dk(r d |d|_ n|dk(r t di|_ |jjdDcgc]}t|dkDs|}}t|dk(rd} n-d |dvrdj!|dd} ndj!|} ||_dj!d|jdz|j$j'd| j'dfDdz|_d|j(t|jdzdvsJ|j(j+dsJ|jdd|_ycc}w)a Resolve assets from a base URL. Parameters -------------- url : str Location where a mesh was stored or directory where mesh was stored. session : HttpSessionLike or None Optional HTTP session used for fetches. Accepts `httpx.Client` or `requests.Session`. timeout : float Per-request timeout in seconds. rwrx)httphttpszscheme z not in ('http', 'https')Nrz`WebResolver` without a `session` is deprecated and will require one in a future release. pass an `httpx.Client` or `requests.Session`.)category stacklevelhttpx.T)follow_redirectsrrequests)allow_redirectsraiohttpzx`aiohttp` sessions are async-only and bound to their creation loop: pass an `httpx.Client` or `requests.Session` insteadr[r\c3<K|]}t|dkDr|yw)rN)rO).0is r z'WebResolver.__init__..s#ZAq6A:Zsz:/z//)ryrwrschemerHwarningswarnDeprecationWarningrrtyper3rGrequest_kwargsrArOrPrnetlocrWbase_urlendswithrK) rrrrparsedrlibraryrrGrAs rrzWebResolver.__init__ds*( 3 !**W%C# == 1 1wv}}&77PQR R ?  MMB,    %_'$w-2J2J2P2PQT2UVW2X g 7;"PD   "6:w"OD   !O  #%D #KK--c2A2qc!fqj2A u:?D E"I 88E#2J'D88E?D HH --$. 0C0CC0H$**UX/Z     4==V]]);a)?)ABBBB}}%%c***3+?Bs ;HHrr/ctddl}|j}|jxs|}|j|j|zfi|j }|j dk\r@|jdr|dd}|j|j|zfi|j }|j|jS)z Get a resource from the remote site. Parameters ------------- name : str Asset name, i.e. 'quadknot.obj.mtl' rNi,./r) rrWrrrr status_coderNraise_for_statuscontent)rrrclientresponses rrzWebResolver.gets zz|&6::dmmd2Jd6I6IJ   3 &t$ABx!vzz$--$"6N$:M:MNH !!#rcddl}|jxs|j|jfi|j}|j |j S)z Fetch the data at the full URL this resolver was instantiated with, i.e. `https://stuff.com/hi.glb` this will return the response. Returns -------- content The value at `self.url` rN)rrrrrrr)rrrs rget_basezWebResolver.get_basesK /DLL)E..txxO4;N;NO!!#rr"c`t|j|z|j|jS)a Return a namespaced version of current resolver. Parameters ------------- namespace : str URL fragment Returns ----------- resolver : WebResolver With sub-url: `https://example.com/{namespace}` rrr)rrrrr$s rr%zWebResolver.namespaceds+ )LLLL  rctd)Nz`WebResolver` is read-only!rr,s rr!zWebResolver.write s!"?@@rctd)Nz`WebResolver` can't list keysrr's rr(zWebResolver.keyss!"ABBr)N>@)r2r3r4r5r8r floatrrkrrr%r!r(r:rrrr_sx+/ a, a,!4'a, a,F   < % $ C M *ACrrc ~eZdZ ddededzdedzdedzdedzdef dZd Zd Ze d e fd Z d Z dZ y)GithubResolverNrepobranchcommitsaverrc|d|d|d|_n|d|d|d|_n tdt|j|||_|t j ||_yd|_y)a Get files from a remote Github repository by downloading a zip file with the entire branch or a specific commit. Parameters ------------- repo In the format of `owner/repo`. branch The remote branch you want to get files from. commit The full commit hash: pass either this OR branch. save A path if you want to save results locally. session : HttpSessionLike or None Optional HTTP session used for fetches. Accepts `httpx.Client` or `requests.Session`. timeout : float Per-request timeout in seconds. Nzhttps://github.com/z /archive/z.zipz/archive/refs/heads/z$`commit` or `branch` must be passed!r)rrHrresolverr DiskCachecache)rrrrrrrs rrzGithubResolver.__init__s~>  ,TF)F84HDH  ,TF2FvhdSDHCD D$'7S   **40DJDJrc6|jjS)z List the available files in the repository. Returns ---------- keys : iterable Keys available to the resolved. )zippedr(r's rr(zGithubResolver.keysDs{{!!rctdrrrs rr!zGithubResolver.writeOs!"<==rr/c t|dr |jS|jj|j|j j }ttjtj|djDcic]\}}|jddd|c}}|_|jScc}}w)zj - opened zip file - locally saved zip file - retrieve zip file and saved _zipzip) file_typer[r) ryrrrrrrrmr decompresswrap_as_streamitemsrG)rrawrtvs rrzGithubResolver.zippedRs 4 99 jjnnTXXt}}'='=> !OO'',%' DAqQ"A%   yy s C c8|jj|Sr)rrrs rrzGithubResolver.getls{{s##rc8|jj|S)r)rr%r$s rr%zGithubResolver.namespacedos {{%%i00r)NNNNr)r2r3r4r8r rrr(r!propertyrmrrr%r:rrrrs"!*...d .d . Dj . !4' ..` "> 2$1rrc# Kd dd fd fd fdddg}|d }t}|D]'}||}||vr|j|||z)tj|d D][\}}|||}||vr|j|||z|||}||vrD|j|||z]d |vr||j d j d d |j d  }|j|j d dj d j d d j|y y y w)z Try to find nearby variants of a specified name. Parameters ------------ name : str Initial name. Yields ----------- nearby : str Name that is a lightly permutated version of the initial name. cD|j|r|t|dS|Sr)rNrO)prefixitems rtrimznearby_names..trims$ ??6 "F & & rc|Srr:xs rznearby_names..s!rc"|jSr)rWrs rrznearby_names..s !'')rcd|S)Nrr:rrs rrznearby_names.. $tQ-rcd|S)Nz.\r:rs rrznearby_names..s $ua.rcd|S)N\r:rs rrznearby_names..rrcFtjj|dS)Nr\)r@rArGrs rrznearby_names..s"''--"2&rc&|jddS)Nz%20 )replacers rrznearby_names..s!))E3'rNrrz..r[r\) setadd itertools combinationsrWrGcountextendrP) rr"cleanershitrecurrentabrWrs @rrzrzso"  &'H  %C D' c>  '!! &&x31AdG* c>  '!!AdG* c>  '!!4 t| -$**3/0C4::d3C2CD TZZ%b)//4::3?@hhuo .|sEE ResolverLiker)r5r6rr@pathlibrtypingr urllib.parserrrrtypedr r ABCr r<rmrrrzr__annotations__r:rrrs  "+ "txx "FW7xW7tH+(H+VqC(qChl1Xl1^@H#W, i,r