o 沪g@sddlmZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z dZ dZ d.d/ddZejd0ddZejd0ddZd1ddZd2ddZd1ddZddZd3d"d#Zd4d&d'Zd5d(d)Zd*d+Zd,d-ZdS)6) annotationsN)Path)env_utilerrors)is_binary_stringz .streamlitstaticautodatabytesencodingstrreturnio.StringIO | io.BytesIOcCs<|dkr t|r dnd}n|}|rt||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)r r data_encodingrO/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/streamlit/file_util.pyget_encoded_file_data!s  rFccs|t|}t|jdkrtd|dd}|r|d7}ttjt || }|VWddS1s7wYdS)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: ""rbN) get_streamlit_file_pathosstatst_sizerErroropenpathjoinCONFIG_FOLDER_NAME)r binaryfilenamemodehandlerrrstreamlit_read@s"r'c csd}|r |d7}t|}tjtj|ddzt|| }|VWdWdS1s.wYWdStya}z dtj|g}|jtj krUt j rU| dt d|d}~ww) 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_okNzUnable to write file: zSPython is limited to files below 2GB on OSX. See https://bugs.python.org/issue24658 )rrmakedirsr dirnamerOSErrorabspatherrnoEINVALr IS_DARWINappendrrr!)r r#r%r&emsgrrrstreamlit_writeYs& &r5cCs*tjtjt}tjtj|dS)z3Get the folder where static HTML/JS/CSS files live.r)rr r,normpath__file__r!)r,rrrget_static_dirysr8main_script_pathcCst|}|jt}tj|S)z*Get the folder where app static files live)rparentAPP_STATIC_FOLDER_NAMErr r.)r9 static_dirrrrget_app_static_dirs  r=cGs,t}|dur tdt|tt|S)zyReturn the full path to a file in ~/.streamlit. This doesn't guarantee that the file (or its directory) exists. NzNo home directory.)rhome RuntimeErrorr r")filepathr>rrrrsrcGstttt|S)zReturn the full path to a filepath in ${CWD}/.streamlit. This doesn't guarantee that the file (or its directory) exists. )r rcwdr"r@rrrget_project_streamlit_file_pathsrCr@folderpath_globboolcCsJ|ds|dr|d7}n|d7}ddl}tj|d}|||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/*rN)endswithfnmatchrr r,)r@rDrIfile_dirrrrfile_is_in_folder_globs   rK directoryintcCsFd}t|D]\}}}|D]}tj||}|tj|7}qq|S)z(Return the size of a directory in bytes.r)rwalkr r!getsize)rL total_sizedirpath_ filenamesffprrrget_directory_sizesrVcsJtjdd}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)rr r..0r rrr sz&file_in_pythonpath..c3s"|] }ttj|VqdS)N)rKrr r6rYrBrr s  z%file_in_pythonpath..)renvirongetlensplitpathsepany)r@ pythonpathabsolute_pathsrrBrfile_in_pythonpaths  recGstjtjj|S)zReturn the normalized path of the joined path. Parameters ---------- *args : str The path components to join. Returns ------- str The normalized path of the joined path. )rr r6r!)argsrrrnormalize_path_joins rgcCstt|}tj|S)aReturn the full path to the main script directory. Parameters ---------- main_script : str The main script path. The path can be an absolute path or a relative path. Returns ------- str The full path to the main script directory. )rgrgetcwdr r,) main_scriptr9rrrget_main_script_directorys rj)r)r r r r r r)F)r r )r9r r r )r@r rDr r rE)rLr r rM)r@r r rE) __future__r contextlibr/rrpathlibr streamlitrrstreamlit.string_utilrr"r;rcontextmanagerr'r5r8r=rrCrKrVrergrjrrrrs2