获取streamlit中上传文件的原名

Posted 2023-03-29

【中文标题】获取streamlit中上传文件的原名

【英文标题】：Get the original name of uploaded files in streamlit

【发布时间】：2020-12-06 01:29:04

【问题描述】：

我正在使用 streamlit 制作一个基本的可视化应用程序来比较两个数据集，为此我使用了以下由来自 streamlit 画廊的 Marc Skov 制作的示例：

from typing import Dict

import streamlit as st


@st.cache(allow_output_mutation=True)
def get_static_store() -> Dict:
    """This dictionary is initialized once and can be used to store the files uploaded"""
    return 


def main():
    """Run this function to run the app"""
    static_store = get_static_store()

    st.info(__doc__)
    result = st.file_uploader("Upload", type="py")
    if result:
        # Process you file here
        value = result.getvalue()

        # And add it to the static_store if not already in
        if not value in static_store.values():
            static_store[result] = value
    else:
        static_store.clear()  # Hack to clear list if the user clears the cache and reloads the page
        st.info("Upload one or more `.py` files.")

    if st.button("Clear file list"):
        static_store.clear()
    if st.checkbox("Show file list?", True):
        st.write(list(static_store.keys()))
    if st.checkbox("Show content of files?"):
        for value in static_store.values():
            st.code(value)


main()

这确实有效，但比较数据集却无法显示它们的名称很奇怪。 代码确实明确表示无法使用此方法获取文件名。但这是 8 个月前的一个例子，我想知道现在是否还有其他方法可以做到这一点。

【参考方案1】：

在对9 July 的提交中，对file_uploader() 进行了轻微修改。它现在返回一个包含：

name 键包含上传的文件名 数据键包含 BytesIO 或 StringIO 对象

所以您应该能够使用result.name 获取文件名，使用result.data 获取数据。