gddlmZddlZddlmZddlmZmZmZddl m Z ddl m Z er ddl mZddlmZGd d eZGd d eZGd deZGddej*ZGdde eZy)) annotationsN)abstractmethod) TYPE_CHECKING NamedTupleProtocol)util)CacheStatsProvider)Sequence)FileURLsc:eZdZUdZded<ded<ded<ded<y) UploadedFileRecz7Metadata and raw bytes for an uploaded file. Immutable.strfile_idnametypebytesdataN__name__ __module__ __qualname____doc____annotations__\/var/www/openai/venv/lib/python3.12/site-packages/streamlit/runtime/uploaded_file_manager.pyr r sA L I I Krr c0eZdZUdZded<ded<ded<y)UploadFileUrlInfoz9Information we provide for single file in get_upload_urlsrr upload_url delete_urlNrrrrrr'sC LOOrrceZdZUdZded<y) DeletedFileaRepresents a deleted file in deserialized values for st.file_uploader and st.camera_input Return this from st.file_uploader and st.camera_input deserialize (so they can be used in session_state), when widget value contains file record that is missing from the storage. DeleteFile instances filtered out before return final value to the user in script, or before sending to frontend.rrNrrrrr"r"/s&Lrr"c4eZdZdZdfd ZddZddZxZS) UploadedFilezA mutable uploaded file. This class extends BytesIO, which has copy-on-write semantics when initialized with `bytes`. ct||j|j|_|j|_|j |_t |j|_||_yN) super__init__rrrrlensize _file_urls)selfrecord file_urls __class__s rr(zUploadedFile.__init__CsN %~~ KK KK  $ #rc`t|tstS|j|jk(Sr&) isinstancer$NotImplementedr)r,others r__eq__zUploadedFile.__eq__Os%%.! !||u}},,rc,tj|Sr&)rrepr_)r,s r__repr__zUploadedFile.__repr__Tszz$r)r-r r. FileURLsProto)r3objectreturnbool)r:r)rrrrr(r4r7 __classcell__)r/s@rr$r$<s $-  rr$cTeZdZdZe ddZeddZ ddZy) UploadedFileManageraUploadedFileManager protocol, that should be implemented by the concrete uploaded file managers. It is responsible for: - retrieving files by session_id and file_id for st.file_uploader and st.camera_input - cleaning up uploaded files associated with session on session end It should be created during Runtime initialization. Optionally UploadedFileManager could be responsible for issuing URLs which will be used by frontend to upload files to. ct)aReturn a list of UploadedFileRec for a given sequence of file_ids. Parameters ---------- session_id The ID of the session that owns the files. file_ids The sequence of ids associated with files to retrieve. Returns ------- List[UploadedFileRec] A list of URL UploadedFileRec instances, each instance contains information about uploaded file. NotImplementedError)r, session_idfile_idss r get_fileszUploadedFileManager.get_filesg &"!rct)z1Remove all files associated with a given session.r@)r,rBs rremove_session_filesz(UploadedFileManager.remove_session_files|s "!rct)a!Return a list of UploadFileUrlInfo for a given sequence of file_names. Optional to implement, issuing of URLs could be done by other service. Parameters ---------- session_id The ID of the session that request URLs. file_names The sequence of file names for which URLs are requested Returns ------- List[UploadFileUrlInfo] A list of UploadFileUrlInfo instances, each instance contains information about uploaded file URLs. r@)r,rB file_namess rget_upload_urlsz#UploadedFileManager.get_upload_urlsrErN)rBrrC Sequence[str]r:zlist[UploadedFileRec])rBrr:None)rBrrIrKr:zlist[UploadFileUrlInfo])rrrrrrDrGrJrrrr>r>Xsd "")6" ""(""""+8" "rr>) __future__rioabcrtypingrrr streamlitrstreamlit.runtime.statsr collections.abcr streamlit.proto.Common_pb2r r8r rr"BytesIOr$r>rrrrVsh# 666(Dj  *  2:: 8<",h<"r