main_v1.py

from fastapi import FastAPI, HTTPException
from git import Repo
import os
import shutil
from pydantic import BaseModel

app = FastAPI()


# 定义请求体的模型
class GitRepo(BaseModel):
    git_url: str


@app.get("/")
async def get_main():
    return {"message": "Welcome to Code Reader!"}


@app.post("/get-repo-content/")
async def print_repo_url(repo: GitRepo):
    print(repo.git_url)
    git_url = repo.git_url
    base_dir = f"./{git_url.split('/')[-1]}"
    try:
        # 从URL中解析仓库名称
        repo_name = git_url.split("/")[-1]
        if repo_name.endswith(".git"):
            repo_name = repo_name[:-4]

        repo_dir = os.path.join(base_dir, repo_name)
        print(repo_dir)
        # 如果目录存在，先删除
        if os.path.exists(repo_dir):
            shutil.rmtree(repo_dir)

        # 克隆仓库
        Repo.clone_from(git_url, repo_dir)

        content = read_all_files(repo_dir)

        # 清理克隆的仓库目录
        shutil.rmtree(repo_dir)

        return {"content": content[:50000]}

    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


def read_all_files(directory):
    all_text = ""
    for root, dirs, files in os.walk(directory):
        for file in files:
            file_path = os.path.join(root, file)
            if os.path.isfile(file_path) and file_path.endswith(".py"):
                try:
                    with open(file_path, "r", encoding="utf-8") as file:
                        filename = file_path.split("/")[-1]
                        all_text += f"File: {filename}\n\n" + file.read() + "\n\n"
                except UnicodeDecodeError:
                    # 忽略无法读取的文件（如二进制文件）
                    pass
    return all_text


if __name__ == "__main__":
    import uvicorn

    uvicorn.run("main:app", host="0.0.0.0", port=80)