wehub-resource-sync
94 lines
3.0 KiB
Python
94 lines
3.0 KiB
Python
"""Assigns the issue based on who created the file mentioned."""
|
|
|
|
# pylint: disable=line-too-long,too-many-arguments
|
|
import base64
|
|
import json
|
|
import os
|
|
import re
|
|
|
|
from github import Github
|
|
|
|
|
|
def get_issue_number(event_path: str) -> int:
|
|
"""Retrieves the issue number from GitHub event data."""
|
|
# Load event data
|
|
with open(event_path, encoding="utf-8") as f:
|
|
event_data = json.load(f)
|
|
|
|
# Determine the issue number based on the event
|
|
if "issue" in event_data:
|
|
return int(event_data["issue"]["number"])
|
|
|
|
raise ValueError("Unable to determine issue number from event data.")
|
|
|
|
|
|
def main() -> None:
|
|
"""Gets filename mentioned in issue, finds author username in file, assigns issue to user."""
|
|
# Get GitHub token and repository details
|
|
repo_name = os.getenv("GITHUB_REPOSITORY", "")
|
|
token = os.getenv("GITHUB_TOKEN")
|
|
issue_number = get_issue_number(os.getenv("GITHUB_EVENT_PATH", ""))
|
|
|
|
g = Github(token)
|
|
repo = g.get_repo(repo_name)
|
|
issue = repo.get_issue(number=issue_number)
|
|
|
|
# Regex to find any file with an extension
|
|
file_match = re.search(r"\b([\w-]+\.[\w]+)\b", issue.body, re.IGNORECASE)
|
|
|
|
if not file_match:
|
|
print("No file found in issue.")
|
|
return
|
|
|
|
file_name = file_match.group(1)
|
|
result = g.search_code(f"repo:{repo_name} filename:{file_name}")
|
|
|
|
if result.totalCount == 0:
|
|
result = g.search_code(f"repo:{repo_name} {file_name}")
|
|
if result.totalCount == 0:
|
|
print(f"No files found for {file_name}")
|
|
return
|
|
|
|
file_path = result[0].path
|
|
|
|
# Get the commits for the file
|
|
commits = repo.get_commits(path=file_path)
|
|
|
|
# Try to get the author of the first commit
|
|
if commits.totalCount > 0:
|
|
# The last commit in the list is the first commit
|
|
first_commit = commits[commits.totalCount - 1]
|
|
if first_commit.author:
|
|
username = first_commit.author.login
|
|
print(
|
|
f"Assigning {username} to Issue #{issue_number} for File {file_path} based on the first commit."
|
|
)
|
|
issue.add_to_assignees(username)
|
|
return
|
|
|
|
# If the file is a notebook and the first commit author wasn't found,
|
|
# check the notebook metadata as a fallback.
|
|
if file_name.endswith(".ipynb"):
|
|
print(
|
|
"Could not determine the first commit author, checking the notebook metadata."
|
|
)
|
|
file_content_encoded = repo.get_contents(file_path).content
|
|
file_content = str(base64.b64decode(file_content_encoded))[:10000]
|
|
match = re.search(
|
|
r"Author.+https://github\.com/([^/\)]+)", file_content, flags=re.DOTALL
|
|
)
|
|
|
|
if match:
|
|
username = match.group(1)
|
|
print(
|
|
f"Assigning {username} to Issue #{issue_number} for File {file_path} based on notebook metadata."
|
|
)
|
|
issue.add_to_assignees(username)
|
|
return
|
|
|
|
print(f"No User Found for {file_name}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|