Update app.py

app.py

@@ -1,83 +1,81 @@
-from flask import Flask, render_template, request, send_file
+import os
+import tempfile
+from flask import Flask, render_template, request, send_file, flash
 import requests
+from git import Repo
+import shutil
 import zipfile
-from io import BytesIO
-from urllib.parse import urlparse
+import re
 
-# 创建 Flask 应用实例
 app = Flask(__name__)
+app.secret_key = 'your_secret_key'  # 更改为更安全的密钥
 
-def is_directory(url):
-    """检查 URL 是否指向 GitHub 文件夹"""
-    pathname = urlparse(url).path.split("/")
-    return "tree" in pathname and len(pathname) > 5
+def download_github_files(repo_url, file_paths):
+    """从GitHub仓库下载指定路径的文件，并打包成zip。"""
+    try:
+        temp_dir = tempfile.mkdtemp()
+        repo_name = repo_url.split('/')[-1].replace(".git", "")
+        local_repo_path = os.path.join(temp_dir, repo_name)
+        
+        # 克隆仓库
+        Repo.clone_from(repo_url, local_repo_path)
 
-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
+        zip_file_path = os.path.join(temp_dir, f"{repo_name}_files.zip")
+        with zipfile.ZipFile(zip_file_path, 'w', zipfile.ZIP_DEFLATED) as zf:
+            for file_path in file_paths:
+                abs_file_path = os.path.join(local_repo_path, file_path)
+                if os.path.exists(abs_file_path) and os.path.isfile(abs_file_path):
+                     zf.write(abs_file_path, os.path.basename(abs_file_path))
+                elif os.path.exists(abs_file_path) and os.path.isdir(abs_file_path):
+                    for root, dirs, files in os.walk(abs_file_path):
+                        for file in files:
+                            file_abs_path = os.path.join(root, file)
+                            zf.write(file_abs_path, os.path.relpath(file_abs_path, local_repo_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()
+        shutil.rmtree(local_repo_path) # 删除克隆的仓库
+        if os.path.exists(zip_file_path) and os.path.getsize(zip_file_path) > 0:
+            return zip_file_path, f"{repo_name}_files.zip"
+        else:
+            return None, None
+    except Exception as e:
+        print(f"Error during download: {e}")
+        return None, None
 
-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
+def parse_github_url(url):
+    """解析 GitHub URL，提取仓库 URL 和文件路径。"""
+    match = re.match(r'https://github\.com/([^/]+)/([^/]+)/tree/([^/]+)(.*)', url)
+    if match:
+        owner, repo, branch, path = match.groups()
+        repo_url = f'https://github.com/{owner}/{repo}.git'
+        file_paths = [path.lstrip('/')] if path else []
+        return repo_url, file_paths
+    else:
+        return None, None
 
-@app.route("/", methods=["GET", "POST"])
-def index():
-    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)
+@app.route('/', methods=['GET', 'POST'])
+def index():
+    if request.method == 'POST':
+        repo_path = request.form.get('repo_path')
+        if not repo_path:
+            flash('请提供 GitHub 链接', 'error')
+            return render_template('index.html')
 
-            # 获取文件夹内容
-            contents = fetch_folder_contents(owner, repo, branch, folder_path)
+        repo_url, file_paths = parse_github_url(repo_path)
+        if not repo_url:
+             flash('无效的 GitHub 链接', 'error')
+             return render_template('index.html')
 
-            # 下载所有文件
-            files = {}
-            for item in contents:
-                if item["type"] == "file":
-                    file_content = download_file(item["download_url"])
-                    files[item["name"]] = file_content
+        zip_path, zip_filename = download_github_files(repo_url, file_paths)
+        if zip_path:
+            return send_file(zip_path, as_attachment=True, download_name=zip_filename)
+        else:
+            flash('下载失败，请检查链接和文件路径。', 'error')
+            return render_template('index.html')
 
-            # 打包成 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')
 
-    return render_template("index.html")
 
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=5000)
+if __name__ == '__main__':
+    app.run(debug=True, host="0.0.0.0")