gpt-pr 0.2.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

{gpt_pr-0.2.0.dist-info → gpt_pr-0.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gpt-pr
-Version: 0.2.0
+Version: 0.2.1
 Summary: Automate your GitHub workflow with GPT-PR: an OpenAI powered library for streamlined PR generation.
 Home-page: http://github.com/alissonperez/gpt-pr
 Author: Alisson R. Perez
@@ -17,6 +17,7 @@ Requires-Dist: prompt-toolkit ==3.0.43 ; python_full_version >= "3.7.0"
 Requires-Dist: openai ==1.14.0 ; python_full_version >= "3.7.1"
 Requires-Dist: exceptiongroup ==1.2.0 ; python_version < "3.11"
 Requires-Dist: cached-property ==1.5.2 ; python_version < "3.8"
+Requires-Dist: typing-extensions ==4.7.1 ; python_version < "3.8"
 Requires-Dist: importlib-metadata ==6.7.0 ; python_version == "3.7"
 Requires-Dist: six ==1.16.0 ; python_version >= "2.7" and python_version not in "3.0, 3.1, 3.2"
 Requires-Dist: deprecated ==1.2.14 ; python_version >= "2.7" and python_version not in "3.0, 3.1, 3.2, 3.3"
@@ -41,7 +42,6 @@ Requires-Dist: smmap ==5.0.1 ; python_version >= "3.7"
 Requires-Dist: sniffio ==1.3.1 ; python_version >= "3.7"
 Requires-Dist: termcolor ==2.3.0 ; python_version >= "3.7"
 Requires-Dist: tqdm ==4.66.2 ; python_version >= "3.7"
-Requires-Dist: typing-extensions ==4.7.1 ; python_version >= "3.7"
 Requires-Dist: urllib3 ==2.0.7 ; python_version >= "3.7"
 Requires-Dist: zipp ==3.15.0 ; python_version >= "3.7"
 Requires-Dist: inquirerpy ==0.3.4 ; python_version >= "3.7" and python_version < "4.0"

gpt_pr-0.2.1.dist-info/RECORD

@@ -0,0 +1,13 @@
+gptpr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+gptpr/consolecolor.py,sha256=_JmBMNjIflWMlgP2VkCWu6uQLR9oHBy52uV3TRJJgF4,800
+gptpr/gh.py,sha256=aunn_8wi4pGoGkjdqyA2MZVAAxfItpnnmDCp20-i-WA,786
+gptpr/gitutil.py,sha256=NBD3iRnbFEPRU47w7c5TowwtZieDYkU4zybvv0PoOU0,5783
+gptpr/main.py,sha256=rkalqLcc1Nh5WH51w7ayEMIYNoScrRBNVYl3KLuZFdY,1270
+gptpr/prdata.py,sha256=y4VodgCiSaARPXzJHsg32-cqrR_nZqyyGSLij6ee-oo,5846
+gptpr/test_prdata.py,sha256=rSJ-yqOdw-iYdBWyqnA2SXbdrhT8KgIkRTTf9SY1S1g,474
+gptpr/version.py,sha256=HfjVOrpTnmZ-xVFCYSVmX50EXaBQeJteUHG-PD6iQs8,22
+gpt_pr-0.2.1.dist-info/METADATA,sha256=rJ-4clCl3pJh99vscyuuVQsLt4XQzqpgf4K67YYMZJY,2637
+gpt_pr-0.2.1.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+gpt_pr-0.2.1.dist-info/entry_points.txt,sha256=aXCkyNdoUHfSJXVhRKHj8m09twDfcDmY7xC66u5N3hE,43
+gpt_pr-0.2.1.dist-info/top_level.txt,sha256=DZcbzlsjh4BD8njGcvhOeCZ83U_oYWgCn0w8qx5--04,6
+gpt_pr-0.2.1.dist-info/RECORD,,

gptpr/gh.py

@@ -21,7 +21,7 @@ def create_pr(pr_data, yield_confirmation):
 
     if pr_confirmation:
         pr = repo.create_pull(title=pr_data.title, body=pr_data.body,
-                              head=pr_data.branch_info.branch, base='main')
+                              head=pr_data.branch_info.branch, base=pr_data.branch_info.base_branch)
         print("Pull request created successfully: ", pr.html_url)
     else:
         print('cancelling...')

gptpr/gitutil.py

@@ -10,6 +10,7 @@ from InquirerPy.base.control import Choice
 class BranchInfo:
     owner: str
     repo: str
+    base_branch: str
     branch: str
     commits: list
     highlight_commits: list
@@ -70,10 +71,11 @@ def get_branch_info(base_branch, yield_confirmation):
     return BranchInfo(
         owner=owner,
         repo=repo_name,
+        base_branch=base_branch,
         branch=current_branch.name,
         commits=commits,
         highlight_commits=highlight_commits,
-        diff=_get_diff_changes(repo, current_branch.name, yield_confirmation)
+        diff=_get_diff_changes(repo, base_branch, current_branch.name, yield_confirmation)
     )
 
 
@@ -85,7 +87,7 @@ def _branch_exists(repo, branch_name):
 
 
 def _get_diff_messages_against_base_branch(repo, branch, base_branch):
-    # Get commit messages that are in the current branch but not in the main branch
+    # Get commit messages that are in the current branch but not in the base branch
     commits_diff = list(repo.iter_commits(f'{base_branch}..{branch}'))
 
     return [commit.message.strip('\n') for commit in commits_diff]
@@ -147,33 +149,33 @@ def _extract_owner_and_repo(repo_url):
     return owner, '.'.join(repo_info.split('.')[:-1])
 
 
-def _get_diff_changes(repo, branch, yield_confirmation):
+def _get_diff_changes(repo, base_branch, branch, yield_confirmation):
     diff_changes = []
 
-    stats = _get_stats(repo, branch)
+    stats = _get_stats(repo, base_branch, branch)
     files_to_ignore = _get_files_to_ignore(stats, yield_confirmation)
 
     for file_change in stats:
         if file_change.file_path in files_to_ignore:
             continue
 
-        file_diff = repo.git.diff('main', branch, '--', file_change.file_path)
+        file_diff = repo.git.diff(base_branch, branch, '--', file_change.file_path)
 
         diff_changes.append(file_diff)
 
     return '\n'.join(diff_changes)
 
 
-def _get_stats(repo, branch):
+def _get_stats(repo, base_branch, branch):
     '''
-    Get the stats of the difference between the current branch and the main branch
+    Get the stats of the difference between the current branch and the base branch
     '''
 
     # returns:
     # 4       0       README.md
     # 2       0       application/aggregator/aggregator.go
     # 0       257     go.sum
-    diff_index = repo.git.diff('main', branch, '--numstat')
+    diff_index = repo.git.diff(base_branch, branch, '--numstat')
 
     files_changed = []
     for line in diff_index.split('\n'):

gptpr/prdata.py

@@ -48,7 +48,7 @@ class PrData():
             f'{cc.bold("Repository")}: {cc.yellow(self.branch_info.owner)}/{cc.yellow(self.branch_info.repo)}',
             f'{cc.bold("Title")}: {cc.yellow(self.title)}',
             f'{cc.bold("Branch name")}: {cc.yellow(self.branch_info.branch)}',
-            f'{cc.bold("Base branch")}: {cc.yellow("main")}',
+            f'{cc.bold("Base branch")}: {cc.yellow(self.branch_info.base_branch)}',
             f'{cc.bold("PR Description")}:\n{self.body}',
         ])
 

gptpr/version.py

@@ -1 +1 @@
-__version__ = "0.2.0"
+__version__ = "0.2.1"

gpt_pr-0.2.0.dist-info/RECORD

@@ -1,13 +0,0 @@
-gptpr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-gptpr/consolecolor.py,sha256=_JmBMNjIflWMlgP2VkCWu6uQLR9oHBy52uV3TRJJgF4,800
-gptpr/gh.py,sha256=y8JgWYTzYzjOWitkEK9Lshjmuv1gV7Tt4DHBrISxAJM,761
-gptpr/gitutil.py,sha256=uhNwwH3r84p5GK1j_19sZEPMyfftEiGpzuV6QZSW6qE,5667
-gptpr/main.py,sha256=rkalqLcc1Nh5WH51w7ayEMIYNoScrRBNVYl3KLuZFdY,1270
-gptpr/prdata.py,sha256=RErSbUyIb0LxCjlUS4KKHoNi2WN94NRO3Cw0mCxtHiU,5824
-gptpr/test_prdata.py,sha256=rSJ-yqOdw-iYdBWyqnA2SXbdrhT8KgIkRTTf9SY1S1g,474
-gptpr/version.py,sha256=Zn1KFblwuFHiDRdRAiRnDBRkbPttWh44jKa5zG2ov0E,22
-gpt_pr-0.2.0.dist-info/METADATA,sha256=oLLwZT78ZD4ymXo5rABD0L_vI8zbJlidiKOu4OzkaNg,2638
-gpt_pr-0.2.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-gpt_pr-0.2.0.dist-info/entry_points.txt,sha256=aXCkyNdoUHfSJXVhRKHj8m09twDfcDmY7xC66u5N3hE,43
-gpt_pr-0.2.0.dist-info/top_level.txt,sha256=DZcbzlsjh4BD8njGcvhOeCZ83U_oYWgCn0w8qx5--04,6
-gpt_pr-0.2.0.dist-info/RECORD,,