U ARd@sddlZddlZddlZddlZddlZddlmZddlmZm Z ddl m Z dZ dZ d"dd Zejd#d d Zejd$d dZddZeedddZddZedddZddZedddZeedddZedd d!ZdS)%N)Path)env_utilutil)is_binary_stringz .streamlitstaticautocCs8|dkrt|rd}nd}|r.t||St|S)a<Coerce bytes to a BytesIO or a StringIO. Parameters ---------- data : bytes encoding : str Returns ------- BytesIO or StringIO If the file's data is in a well-known textual format (or if the encoding parameter is set), return a StringIO. Otherwise, return BytesIO. rNzutf-8)rioStringIOdecodeBytesIO)dataencodingr7/tmp/pip-unpacked-wheel-b9et7o5g/streamlit/file_util.pyget_encoded_file_data srFc cs`t|}t|jdkr&td|d}|r6|d7}ttjt || }|VW5QRXdS)a$Opens a context to read this file relative to the streamlit path. For example: with streamlit_read('foo.txt') as foo: ... opens the file `.streamlit/foo.txt` path - the path to write to (within the streamlit directory) binary - set to True for binary IO rzRead zero byte file: "%s"rbN) get_streamlit_file_pathosstatst_sizerErroropenpathjoinCONFIG_FOLDER_NAME)rbinaryfilenamemodehandlerrrstreamlit_read>sr c csd}|r|d7}t|}tjtj|ddz t|| }|VW5QRXWn`tk r}zBdtj|g}|jtj krt j r| dt d|W5d}~XYnXdS) aOpens a file for writing within the streamlit path, and ensuring that the path exists. For example: with streamlit_write('foo/bar.txt') as bar: ... opens the file .streamlit/foo/bar.txt for writing, creating any necessary directories along the way. path - the path to write to (within the streamlit directory) binary - set to True for binary IO wrT)exist_okzUnable to write file: %szSPython is limited to files below 2GB on OSX. See https://bugs.python.org/issue24658 N)rrmakedirsrdirnamerOSErrorabspatherrnoEINVALrZ IS_DARWINappendrrr)rrrremsgrrrstreamlit_writeWs r-cCs*tjtjt}tjtj|dS)z3Get the folder where static HTML/JS/CSS files live.rrrr%normpath__file__rr%rrrget_static_dirwsr2)main_script_pathreturncCst|}|jt}tj|S)z*Get the folder where app static files live)rparentAPP_STATIC_FOLDER_NAMErrr')r3Z static_dirrrrget_app_static_dir}s r7cCs*tjtjt}tjtj|dS)z(Get the folder where static assets live.z static/assetsr.r1rrrget_assets_dirsr8)r4cGs0tjd}|dkrtdtjj|tf|S)zyReturn the full path to a file in ~/.streamlit. This doesn't guarantee that the file (or its directory) exists. ~NzNo home directory.)rr expanduser RuntimeErrorrr)filepathhomerrrrs rcGstjjttf|S)zReturn the full path to a filepath in ${CWD}/.streamlit. This doesn't guarantee that the file (or its directory) exists. )rrrgetcwdrr<rrrget_project_streamlit_file_pathsr@cCsB|ds&|dr|d7}n|d7}tj|d}t||S)zTest whether a file is in some folder with globbing support. Parameters ---------- filepath : str A file path. folderpath_glob: str A path to a folder that may include globbing. */z/*)endswithrrr%fnmatch)r<Zfolderpath_globZfile_dirrrrfile_is_in_folder_globs   rE) directoryr4cCsFd}t|D]2\}}}|D]"}tj||}|tj|7}qq|S)z(Return the size of a directory in bytes.r)rwalkrrgetsize)rFZ total_sizedirpath_ filenamesffprrrget_directory_sizes rNcsJtjdd}t|dkrdSdd|tjD}tfdd|DS) aETest whether a filepath is in the same folder of a path specified in the PYTHONPATH env variable. Parameters ---------- filepath : str An absolute file path. Returns ------- boolean True if contained in PYTHONPATH, False otherwise. False if PYTHONPATH is not defined or empty. PYTHONPATHrFcSsg|]}tj|qSr)rrr'.0rrrr sz&file_in_pythonpath..c3s |]}ttj|VqdS)N)rErrr/rQr?rr sz%file_in_pythonpath..)renvirongetlensplitpathsepany)r< pythonpathZabsolute_pathsrr?rfile_in_pythonpaths  r\)r)F)F) contextlibr(rDrrpathlibrZ streamlitrrZstreamlit.string_utilrrr6rcontextmanagerr r-r2strr7r8rr@boolrEintrNr\rrrrs,