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

reformat

Browse Source
This commit is contained in:
Hussam.lawen
2023-09-28 20:15:18 +03:00
parent 4accddcaa7
commit bb12c75431
1 changed files with 19 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
pr_agent/tools/pr_add_docs.py
View File

@ -13,19 +13,6 @@ from pr_agent.git_providers import BitbucketProvider, get_git_provider
from pr_agent.git_providers.git_provider import get_main_pr_language from pr_agent.git_providers.git_provider import get_main_pr_language
def get_docs_for_language(language):
if language.lower() == 'java':
return "javadocs"
elif language.lower() in ['python', 'lisp', 'clojure']:
return "docstrings"
elif language.lower() in ['javascript', 'typescript']:
return "jsdocs"
elif language.lower() == 'c++':
return "doxygen"
else:
return "docs"
class PRAddDocs: class PRAddDocs:
def __init__(self, pr_url: str, cli_mode=False, args: list = None): def __init__(self, pr_url: str, cli_mode=False, args: list = None):
@ -129,12 +116,13 @@ class PRAddDocs:
relevant_line = int(d['relevant line']) # absolute position relevant_line = int(d['relevant line']) # absolute position
documentation = d['documentation'] documentation = d['documentation']
if documentation: if documentation:
new_code_snippet = self.dedent_code(relevant_file, relevant_line, documentation, add_original_line=True) new_code_snippet = self.dedent_code(relevant_file, relevant_line, documentation,
add_original_line=True)
body = f"**Suggestion:** Proposed documentation\n```suggestion\n" + new_code_snippet + "\n```" body = f"**Suggestion:** Proposed documentation\n```suggestion\n" + new_code_snippet + "\n```"
code_suggestions.append({'body': body, 'relevant_file': relevant_file, code_suggestions.append({'body': body, 'relevant_file': relevant_file,
'relevant_lines_start': relevant_line, 'relevant_lines_start': relevant_line,
'relevant_lines_end': relevant_line}) 'relevant_lines_end': relevant_line})
except Exception: except Exception:
if get_settings().config.verbosity_level >= 2: if get_settings().config.verbosity_level >= 2:
logging.info(f"Could not parse code docs: {d}") logging.info(f"Could not parse code docs: {d}")
@ -192,4 +180,18 @@ class PRAddDocs:
else: else:
data.update(data_per_chunk) data.update(data_per_chunk)
self.data = data self.data = data
return data return data
def get_docs_for_language(language):
language = language.lower()
if language == 'java':
return "javadocs"
elif language in ['python', 'lisp', 'clojure']:
return "docstrings"
elif language in ['javascript', 'typescript']:
return "jsdocs"
elif language == 'c++':
return "doxygen"
else:
return "docs"