본문 바로가기
IT 프로그래밍 관련/streamlit 대시보드

streamlit upload받은 파일을 저장하는 함수

by 지나는행인 2021. 3. 8.
728x90

streamlit에서는 파일을 업로드 받아서 화면에 출력 할 수 있다.

 

그 후에는 디렉토리를 설정하여 업로드 받은 파일을 저장할 수 있는데,

 

그것을 위한 함수 만들기이다.

 

함수는 위와 같이 , 디렉토리가 있는지 없는지 본 후에 없으면 만들고 그 경로에 저장을 하도록 되어있다.

 

save_uploaded_file('temp_files', image_file) 로 함수를 호출하면, 현재의 작업환경에서 temp_files 란 디렉토리가 있는 지 확인 후 에 있으면, 그 경로 안에 image_file변수에 담겨있는 파일명으로 저장된다.

(  image_file 은

image_file = st.file_uploader('Upload Image', type=['png', 'jpg', 'jpeg'])     ##업로드 받는 파일을 담은 변수이다. )

 

temp_files 폴더가 없다면 새롭게 만들고 그곳에 저장을 한다.

 

 

댓글