File size: 3,370 Bytes
c7eea5d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 | from dotenv import load_dotenv
from github import Github
import os
load_dotenv()
def list_repos():
"""Lists all repositories owned by the authenticated GitHub user."""
try:
github = Github(os.getenv('GITHUB_TOKEN'))
user = github.get_user()
repos = user.get_repos()
repo_names = [repo.name for repo in repos]
return ', '.join(repo_names)
except Exception as e:
return None
def download_repo(project_name: str, repo_owner: str = None, branch: str = "main", private: bool = False):
"""
Downloads a GitHub repo (user/<project_name>) using PyGithub
and saves it into agents/coder/projects/{project_name}.
Args:
project_name (str): Repo name on GitHub and local folder name.
repo_owner (str): Repo owner on GitHub, by default = None (it's the user's repository).
branch (str): Branch to download (default 'main').
"""
gh = Github(os.getenv("GITHUB_TOKEN"))
user = gh.get_user()
if repo_owner:
repo = gh.get_repo(f"{repo_owner}/{project_name}")
user.create_repo(project_name, private=private)
else:
repo = user.get_repo(project_name)
dest_path = os.path.join('agents/coder/projects', project_name)
os.makedirs(dest_path, exist_ok=True)
try:
branch_ref = repo.get_branch(branch)
commit_sha = branch_ref.commit.sha
# Get the tree (recursive = True to walk full repo)
tree = repo.get_git_tree(commit_sha, recursive=True).tree
for item in tree:
file_path = os.path.join(dest_path, item.path)
if item.type == "tree":
os.makedirs(file_path, exist_ok=True)
elif item.type == "blob":
blob = repo.get_git_blob(item.sha)
file_content = blob.content
import base64
decoded = base64.b64decode(file_content)
os.makedirs(os.path.dirname(file_path), exist_ok=True)
with open(file_path, "wb") as f:
f.write(decoded)
except Exception as e:
return f"Error downloading repository: {str(e)}"
def initialize_project(project_name: str, repo_owner: str = None, branch: str = "main", private: bool = False):
"""Creates a project directory and GitHub repository with the given project_name."""
try:
github = Github(os.getenv('GITHUB_TOKEN'))
user = github.get_user()
repos = user.get_repos()
repo_names = [repo.name for repo in repos]
if project_name in repo_names or repo_owner:
try:
download_repo(project_name=project_name, repo_owner=repo_owner, branch=branch, private=private)
return f"Repository '{f'{repo_owner}/{project_name}' if repo_owner else project_name}' downloaded successfully!"
except Exception as e:
os.mkdir(f"agents/coder/projects/{project_name}")
user.create_repo(project_name, private=private)
return f"Repository '{project_name}' created successfully!"
else:
user.create_repo(project_name, private=private)
os.mkdir(f"agents/coder/projects/{project_name}")
return f"Repository '{project_name}' created successfully!"
except Exception as e:
return f"Error creating repository: {str(e)}" |