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

improve tweaks

Browse Source
This commit is contained in:
mrT23
2024-02-26 12:05:29 +02:00
parent 116dd75a14
commit 8594c93186
1 changed files with 16 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
pr_agent/tools/pr_code_suggestions.py
View File

@ -338,11 +338,13 @@ class PRCodeSuggestions:
for ext in extensions: for ext in extensions:
extension_to_language[ext] = language extension_to_language[ext] = language
pr_body = "## PR Code Suggestions\n\n"
pr_body += "<table>" pr_body += "<table>"
header = f"Suggestions" header = f"Suggestions"
delta = 75 delta = 76
header += "&nbsp; " * delta header += "&nbsp; " * delta
pr_body += f"""<thead><tr><th></th><th>{header}</th></tr></thead>""" pr_body += f"""<thead><tr><td>Category</td><td align=left>{header}</td></tr></thead>"""
pr_body += """<tbody>""" pr_body += """<tbody>"""
suggestions_labels = dict() suggestions_labels = dict()
# add all suggestions related to each label # add all suggestions related to each label
@ -353,11 +355,13 @@ class PRCodeSuggestions:
suggestions_labels[label].append(suggestion) suggestions_labels[label].append(suggestion)
for label, suggestions in suggestions_labels.items(): for label, suggestions in suggestions_labels.items():
pr_body += f"""<tr><td><strong>{label}</strong></td>""" num_suggestions=len(suggestions)
pr_body += f"""<td>""" # pr_body += f"""<tr><td><strong>{label}</strong></td>"""
pr_body += f"""<tr><td rowspan={num_suggestions}><strong>{label}</strong></td>\n"""
# pr_body += f"""<td>"""
# pr_body += f"""<details><summary>{len(suggestions)} suggestions</summary>""" # pr_body += f"""<details><summary>{len(suggestions)} suggestions</summary>"""
pr_body += f"""<table>""" # pr_body += f"""<table>"""
for suggestion in suggestions: for i, suggestion in enumerate(suggestions):
relevant_file = suggestion['relevant_file'].strip() relevant_file = suggestion['relevant_file'].strip()
relevant_lines_start = int(suggestion['relevant_lines_start']) relevant_lines_start = int(suggestion['relevant_lines_start'])
@ -385,17 +389,17 @@ class PRCodeSuggestions:
example_code = "" example_code = ""
example_code += f"```diff\n{patch}\n```\n" example_code += f"```diff\n{patch}\n```\n"
if i==0:
pr_body += f"""<tr><td>""" pr_body += f"""<td>\n\n"""
else:
pr_body += f"""<tr><td>\n\n"""
suggestion_summary = suggestion['one_sentence_summary'].strip() suggestion_summary = suggestion['one_sentence_summary'].strip()
if '`' in suggestion_summary: if '`' in suggestion_summary:
suggestion_summary = replace_code_tags(suggestion_summary) suggestion_summary = replace_code_tags(suggestion_summary)
suggestion_summary = suggestion_summary + max((77-len(suggestion_summary)), 0)*"&nbsp;" # suggestion_summary = suggestion_summary + max((77-len(suggestion_summary)), 0)*"&nbsp;"
pr_body += f"""\n\n<details><summary>{suggestion_summary}</summary>\n\n___\n\n""" pr_body += f"""\n\n<details><summary>{suggestion_summary}</summary>\n\n___\n\n"""
pr_body += f""" pr_body += f"""
**{suggestion_content}** **{suggestion_content}**
[{relevant_file} {range_str}]({code_snippet_link}) [{relevant_file} {range_str}]({code_snippet_link})
@ -403,9 +407,9 @@ class PRCodeSuggestions:
{example_code} {example_code}
""" """
pr_body += f"</details>" pr_body += f"</details>"
pr_body += f"</td></tr>" pr_body += f"</td></tr>"
pr_body += """</table>"""
# pr_body += "</details>" # pr_body += "</details>"
pr_body += """</td></tr>""" pr_body += """</td></tr>"""
pr_body += """</tr></tbody></table>""" pr_body += """</tr></tbody></table>"""