github.py

# Author: Robert Wojciechowski <robertwoj@microsoft.com>
# License: MIT

import subprocess
import datetime
import json
from enum import Enum
from typing import List

from pr import PR
from repo_type import RepoType

def determine_user():
    return subprocess.check_output(["gh", "api", "user", "--jq", ".login"]).decode().strip()

def list_prs(month_start, month_end) -> List[PR]:
    search_range = f"created:{month_start}..{month_end}"
    try:
        user = determine_user()
        result = subprocess.check_output([
            "gh", "pr", "list",
            "--author", user,
            "--search", search_range,
            "--state", "all",
            "--json", "number,title,createdAt,closedAt,updatedAt"
        ]).decode()
        array = json.loads(result)
        result = []
        for item in array:
            number = item["number"]
            pr = PR(
                repo_type=RepoType.GITHUB,
                title=item["title"],
                created_at=item["createdAt"].split("T")[0],
                url=f"https://github.com/Azure/azure-osconfig/pull/{number}",
            )
            result.append(pr)

        return result
    except subprocess.CalledProcessError as e:
        print(f"Error fetching PRs: {e}")
        return []