Update app.py

app.py

@@ -7,47 +7,77 @@ from urllib.parse import urlparse
 # 创建 Flask 应用实例
 app = Flask(__name__)
 
-@app.route('/', methods=['GET', 'POST'])
+def is_directory(url):
+    """检查 URL 是否指向 GitHub 文件夹"""
+    pathname = urlparse(url).path.split("/")
+    return "tree" in pathname and len(pathname) > 5
+
+def get_repo_info_from_url(url):
+    """从 GitHub URL 中提取仓库信息"""
+    pathname = urlparse(url).path.split("/")
+    owner = pathname[1]
+    repo = pathname[2]
+    branch = pathname[4]
+    folder_path = "/".join(pathname[5:])
+    return owner, repo, branch, folder_path
+
+def fetch_folder_contents(owner, repo, branch, folder_path):
+    """获取文件夹内容"""
+    api_url = f"https://api.github.com/repos/{owner}/{repo}/contents/{folder_path}?ref={branch}"
+    response = requests.get(api_url)
+    if response.status_code != 200:
+        raise Exception(f"Unable to fetch folder contents: {response.status_code}")
+    return response.json()
+
+def download_file(file_url):
+    """下载单个文件"""
+    response = requests.get(file_url)
+    if response.status_code != 200:
+        raise Exception(f"Unable to download file: {response.status_code}")
+    return response.content
+
+def create_zip(files):
+    """将文件打包成 ZIP"""
+    zip_buffer = BytesIO()
+    with zipfile.ZipFile(zip_buffer, "a", zipfile.ZIP_DEFLATED, False) as zip_file:
+        for file_name, file_content in files.items():
+            zip_file.writestr(file_name, file_content)
+    zip_buffer.seek(0)
+    return zip_buffer
+
+@app.route("/", methods=["GET", "POST"])
 def index():
-    if request.method == 'POST':
-        github_url = request.form['github_url']
-        
-        # Parse the GitHub URL to extract owner, repo, branch, and folder path
-        parsed_url = urlparse(github_url)
-        path_parts = parsed_url.path.strip('/').split('/')
-        
-        if len(path_parts) < 5 or path_parts[2] != 'tree':
-            return "Error: Invalid GitHub URL format. Please provide a valid link to a folder."
-
-        owner = path_parts[0]
-        repo = path_parts[1]
-        branch = path_parts[3]
-        folder_path = '/'.join(path_parts[4:])
-
-        # Construct the API URL to get the contents of the folder
-        api_url = f"https://api.github.com/repos/{owner}/{repo}/contents/{folder_path}?ref={branch}"
-        
-        response = requests.get(api_url)
-        if response.status_code != 200:
-            return "Error: Unable to fetch the folder contents."
-
-        files = response.json()
-        if not isinstance(files, list):
-            return "Error: The provided path is not a folder."
-
-        # Create a zip file in memory
-        zip_buffer = BytesIO()
-        with zipfile.ZipFile(zip_buffer, 'a', zipfile.ZIP_DEFLATED, False) as zip_file:
-            for file in files:
-                if file['type'] == 'file':
-                    file_content = requests.get(file['download_url']).content
-                    zip_file.writestr(file['name'], file_content)
-
-        # Prepare the zip file for download
-        zip_buffer.seek(0)
-        return send_file(zip_buffer, download_name=f"{repo}_{folder_path.replace('/', '_')}.zip", as_attachment=True)
-
-    return render_template('index.html')
+    if request.method == "POST":
+        github_url = request.form["github_url"]
+
+        if not is_directory(github_url):
+            return "Error: Please provide a valid GitHub folder URL."
+
+        try:
+            # 提取仓库信息
+            owner, repo, branch, folder_path = get_repo_info_from_url(github_url)
+
+            # 获取文件夹内容
+            contents = fetch_folder_contents(owner, repo, branch, folder_path)
+
+            # 下载所有文件
+            files = {}
+            for item in contents:
+                if item["type"] == "file":
+                    file_content = download_file(item["download_url"])
+                    files[item["name"]] = file_content
+
+            # 打包成 ZIP
+            zip_buffer = create_zip(files)
+            return send_file(
+                zip_buffer,
+                download_name=f"{repo}_{folder_path.replace('/', '_')}.zip",
+                as_attachment=True,
+            )
+        except Exception as e:
+            return f"Error: {str(e)}"
+
+    return render_template("index.html")
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=5000)