Upload 4 files

main.py

@@ -0,0 +1,56 @@
+from typing import List
+import os
+import streamlit as st
+from streamlit.runtime.uploaded_file_manager import UploadedFile
+
+from zip import zip_files, generate_password
+
+DARA_PATH = "data"
+os.makedirs(DARA_PATH, exist_ok=True)
+
+def upload_files(files: List[UploadedFile]):
+    for file in files:
+        with open(file.name, "wb") as f:
+            f.write(file.read())
+
+import base64
+def get_binary_file_downloader_html(file_path, text="Download Link"):
+    # ダウンロードリンクを作成する関数
+    with open(file_path, 'rb') as f:
+        data = f.read()
+    b64 = base64.b64encode(data).decode()
+    href = f'<a href="data:application/octet-stream;base64,{b64}" download="{file_path}">{text}</a>'
+    return href
+
+def main():
+    st.title("File Zipper")
+
+    # ファイルのアップロード
+    files = st.file_uploader("Upload Files", accept_multiple_files=True)
+
+    if files:
+        os.chdir(DARA_PATH)
+        upload_files(files)
+
+        # zipファイルを作成するためのフォームを表示する
+        form = st.form(key='zip_form')
+        zip_file_name = form.text_input("Zip file name",
+                                        value=f"{os.path.splitext(files[0].name)[0]}.zip")
+        submit_button = form.form_submit_button("Create Zip File")
+
+        if submit_button:
+            # zipファイルを作成する
+            password = generate_password()
+            zip_files(src=[file.name for file in files],
+                      dst=zip_file_name,
+                      password=password)
+            # ダウンロードリンクとパスワードの表示
+            st.markdown(get_binary_file_downloader_html(zip_file_name,
+                                                        text=f"Download {zip_file_name}"),
+                        unsafe_allow_html=True)
+            st.markdown(f"password: :orange[{password}]")
+
+        os.chdir("../")
+
+if __name__ == '__main__':
+    main()

requirements.txt

@@ -0,0 +1 @@
+pyminizip==0.2.6

zip.py

@@ -0,0 +1,31 @@
+from typing import List, Union
+import os
+import shutil
+import zipfile
+import pyminizip
+
+import secrets
+import string
+
+def zip_files(src: Union[str, List[str]], dst: str, password: str) -> bool:
+    try:
+        # 型のチェック
+        if isinstance(src, str):
+            src_list = [src]
+        elif isinstance(src, list) and all(isinstance(src_i, str) for src_i in src):
+            src_list = src
+        else:
+            assert False, "Invalid src: [str, List[str]]"
+        pyminizip.compress_multiple([os.path.basename(src_i) for src_i in src_list],
+                                    [os.path.dirname(src_i) for src_i in src_list],
+                                    dst, password, 4)
+        return True
+
+    except Exception as e:
+        print(f"Error: {e}")
+        return False
+
+def generate_password(length: int = 8) -> str:
+    alphabet = string.ascii_letters + string.digits
+    password = ''.join(secrets.choice(alphabet) for _ in range(length))
+    return password