o 沪g@sdZddlmZddlZddlZddlZddlmZddlm Z m Z ddl m Z ddl mZdZd Zdd d d-ddZd.d/ddZd0ddZd1ddZd1dd Ze d!Zd2d'd(Zd)d*ZGd+d,d,e ZdS)3zA bunch of useful utilities for the watcher. These are functions that only make sense within the watcher. In particular, functions that use streamlit.config can go here to avoid a dependency cycle. ) annotationsN)Path)CallableTypeVar)Error)HASHLIB_KWARGSg?F) glob_patternallow_nonexistentpathstrr str | Noner boolreturncsx|rtjsd}ntjr!|pd}t|d}n tfddt}tj dit }| || S)a{Calculate the MD5 checksum of a given path. For a file, this means calculating the md5 of the file's contents. For a directory, we concatenate the directory's path with the names of all the files in it and calculate the md5 of that. IMPORTANT: This method calls time.sleep(), which blocks execution. So you should only use this outside the main thread. zUTF-8*cstSN)_get_file_contentr rR/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/streamlit/watcher/util.pyAsz0calc_md5_with_blocking_retries..Nr) osr existsencodeisdir_stable_dir_identifier_do_with_retriesFileNotFoundErrorhashlibmd5rupdate hexdigest)r r r contentrrrrcalc_md5_with_blocking_retries's    r#floatcs(|r tjs dStfddtS)aReturn the modification time of a path (file or directory). If allow_nonexistent is True and the path does not exist, we return 0.0 to guarantee that any file/dir later created at the path has a later modification time than the last time returned by this function for that path. If allow_nonexistent is False and no file/dir exists at the path, a FileNotFoundError is raised (by os.stat). For any path that does correspond to an existing file/dir, we return its modification time. gcs tjSr)rstatst_mtimerrrrras z(path_modification_time..)rr rrr)r r rrrpath_modification_timeMs r' file_pathbytescCs6t|d }|WdS1swYdS)Nrb)openread)r(frrrrgs $rdir_pathcCs*t|}tdd||D}d|S)NcSsg|] }|jds|jqS).)name startswith).0r-rrr osz_dirfiles..+)rsortedglobjoin)r.r p filenamesrrr _dirfilesls  r:cCs>t||}tD]}t||}||krn|}q|d|S)abWait for the files in a directory to look stable-ish before returning an id. We do this to deal with problems that would otherwise arise from many tools (e.g. git) and editors (e.g. vim) "editing" files (from the user's perspective) by doing some combination of deleting, creating, and moving various files under the hood. Because of this, we're unable to rely on FileSystemEvents that we receive from watchdog to determine when a file has been added to or removed from a directory. This is a bit of an unfortunate situation, but the approach we take here is most likely fine as: - The worst thing that can happen taking this approach is a false positive page added/removed notification, which isn't too disastrous and can just be ignored. - It is impossible (that is, I'm fairly certain that the problem is undecidable) to know whether a file created/deleted/moved event corresponds to a legitimate file creation/deletion/move or is part of some sequence of events that results in what the user sees as a file "edit". r4)r: _retry_dance)r.r dirfiles_ new_dirfilesrrrrts   rTorig_fnCallable[[], T] exceptiontype[Exception] str | Pathc CsHtD]}z|WS|y|tdkr Yqwtd|)aHelper for retrying a function. Calls `orig_fn`. If `exception` is raised, retry. To use this, just replace things like this... result = thing_to_do(file_path, a, b, c) ...with this: result = _do_with_retries( lambda: thing_to_do(file_path, a, b, c), exception: ExceptionThatWillCauseARetry, file_path, # For pretty error message. ) z!Unable to access file or folder: )r; _MAX_RETRIESMaxRetriesError)r@rBr irrrrs    rccs$ttD] }|VttqdS)aHelper for writing a retry loop. This is useful to make sure all our retry loops work the same way. For example, prior to this helper, some loops had time.sleep() *before the first try*, which just slowed things down for no reason. Usage: for i in _retry_dance(): # Do the thing you want to retry automatically. the_thing_worked = do_thing() # Don't forget to include a break/return when the thing you're trying to do # works. if the_thing_worked: break N)rangerFtimesleep_RETRY_WAIT_SECS)rHrrrr;s   r;c@s eZdZdS)rGN)__name__ __module__ __qualname__rrrrrGsrG)r r r r r rrr )F)r r r rrr$)r(r rr))r.r r r rr )r@rArBrCr rDrr?)__doc__ __future__rrrrJpathlibrtypingrrstreamlit.errorsrstreamlit.utilrrFrLr#r'rr:rr?rr;rGrrrrs,     &   # "