a h @s@ddlZddlmZddlmZddlmZGdddeZdS) N)AbstractFileSystem)infer_storage_options) MemoryFilecseZdZdZdZdZdZdfdd Zedd Z d d Z d d Z e ddZ eddZdddZdddZdddZZS) GistFileSystema9 Interface to files in a single GitHub Gist. Provides read-only access to a gist's files. Gists do not contain subdirectories, so file listing is straightforward. Parameters ---------- gist_id : str The ID of the gist you want to access (the long hex value from the URL). filenames : list[str] (optional) If provided, only make a file system representing these files, and do not fetch the list of all files for this gist. sha : str (optional) If provided, fetch a particular revision of the gist. If omitted, the latest revision is used. username : str (optional) GitHub username for authentication (required if token is given). token : str (optional) GitHub personal access token (required if username is given). timeout : (float, float) or float, optional Connect and read timeouts for requests (default 60s each). kwargs : dict Stored on `self.request_kw` and passed to `requests.get` when fetching Gist metadata or reading ("opening") a file. Zgistz&https://api.github.com/gists/{gist_id}z,https://api.github.com/gists/{gist_id}/{sha}Nc srt||_||_||_|du|duAr<|s4|r/file.txt gist://username:TOKEN@/file.txt rpasswordrhostr r9r:rNrrrr )rrsplitlen)r9sor6 path_partsrrr_get_kwargs_from_urlss      z$GistFileSystem._get_kwargs_from_urlsr Fc sh|p ddkr"|jd}n(|jd}fdd|D}|sJt|rR|Stdd|DSdS)a List files in the gist. Gists are single-level, so any 'path' is basically the filename, or empty for all files. Parameters ---------- path : str, optional The filename to list. If empty, returns all files in the gist. detail : bool, default False If True, return a list of dicts; if False, return a list of filenames. r csg|]}|dkr|qSr0r.0fr9rr z%GistFileSystem.ls..css|]}|dVqdS)r0NrrGrrr rLz$GistFileSystem.ls..N)r=rr&sorted)rr9Zdetailrresults all_filesrrJrlss   zGistFileSystem.lsrbc  s|dkrtd|fdd|jdD}|s>t|d}|d}|sbtdtj|fd |ji|j}|jd krt| t d |j S) z3 Read a single file from the gist. rRz'GitHub Gist FS is read-only (no write).csg|]}|dkr|qSrFrrGrJrrrKrLz(GistFileSystem._open..r rr/zNo raw_url for file: rrN) NotImplementedErrorr=rr&r$r#rrr%r'rcontent) rr9mode block_sizermatchesr8r/r*rrJr_opens   zGistFileSystem._openraisec Ks|j||d}i}|D]}z<||d}|||<Wdn1sJ0YWqty} z,|dkrt| n|dkr~n| ||<WYd} ~ qd} ~ 00q|S)z Return {path: contents} for the given file or files. If 'recursive' is True, and path is empty, returns all files in the gist. ) recursiverRNrYZomit)Z expand_pathopenreadr&) rr9rZZon_errorrpathsr6prIerrrcats. zGistFileSystem.cat)NNNNN)r F)rRN)FrY)__name__ __module__ __qualname____doc__protocolr"r r propertyrr+r classmethodr= staticmethodrErQrXr` __classcell__rrrrrs*      r)r#specrutilsrZmemoryrrrrrrs