xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-02 03:40:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

support more types of github ticket url / references

Browse Source
This commit is contained in:
Hussam.lawen
2024-10-13 22:41:33 +03:00
parent 49f8d86c77
commit 840e8c4d6b
1 changed files with 19 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
pr_agent/tools/ticket_pr_compliance_check.py
View File

@ -34,20 +34,26 @@ def extract_ticket_links_from_pr_description(pr_description, repo_path):
""" """
github_tickets = [] github_tickets = []
try: try:
# example link to search for: https://github.com/Codium-ai/pr-agent-pro/issues/525 # Pattern to match full GitHub issue URLs and shorthand notations like owner/repo#issue_number or https://github.com/owner/repo/issues/issue_number
pattern = r'https://github[^/]+/[^/]+/[^/]+/issues/\d+' # should support also github server (for example 'https://github.company.ai/Codium-ai/pr-agent-pro/issues/525') pattern = r'(https://github[^/]+/[^/]+/[^/]+/issues/\d+)|(\b(\w+)/(\w+)#(\d+)\b)'
# Find all matches in the text matches = re.findall(pattern, pr_description)
github_tickets = re.findall(pattern, pr_description) for match in matches:
if match[0]: # Full URL match
# Find all issues referenced like #123 and add them as https://github.com/{repo_path}/issues/{issue_number} github_tickets.append(match[0])
issue_number_pattern = r'#\d+' else: # Shorthand notation match
issue_numbers = re.findall(issue_number_pattern, pr_description) owner, repo, issue_number = match[2], match[3], match[4]
for issue_number in issue_numbers: github_tickets.append(f'https://github.com/{owner}/{repo}/issues/{issue_number}')
issue_number = issue_number[1:] # remove # if not github_tickets:
# check if issue_number is a valid number and len(issue_number) < 5 # Search for #123 format within the same repo
if issue_number.isdigit() and len(issue_number) < 5: issue_number_pattern = r'#\d+'
github_tickets.append(f'https://github.com/{repo_path}/issues/{issue_number}') issue_numbers = re.findall(issue_number_pattern, pr_description)
for issue_number in issue_numbers:
issue_number = issue_number[1:] # remove #
if issue_number.isdigit() and len(issue_number) < 5:
issue_url = f'https://github.com/{repo_path}/issues/{issue_number}'
if issue_url not in github_tickets:
github_tickets.append(issue_url)
except Exception as e: except Exception as e:
get_logger().error(f"Error extracting tickets error= {e}", get_logger().error(f"Error extracting tickets error= {e}",
artifact={"traceback": traceback.format_exc()}) artifact={"traceback": traceback.format_exc()})