gpt-pr 0.2.0__tar.gz → 0.2.1__tar.gz

{gpt-pr-0.2.0 → gpt-pr-0.2.1}/PKG-INFO

@@ -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

{gpt-pr-0.2.0 → gpt-pr-0.2.1}/gpt_pr.egg-info/PKG-INFO

@@ -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

{gpt-pr-0.2.0 → gpt-pr-0.2.1}/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...')

{gpt-pr-0.2.0 → gpt-pr-0.2.1}/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'):

{gpt-pr-0.2.0 → gpt-pr-0.2.1}/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}',
         ])
 

gpt-pr-0.2.1/gptpr/version.py

@@ -0,0 +1 @@
+__version__ = "0.2.1"

{gpt-pr-0.2.0 → gpt-pr-0.2.1}/requirements.txt

@@ -33,7 +33,7 @@ smmap==5.0.1; python_version >= '3.7'
 sniffio==1.3.1; python_version >= '3.7'
 termcolor==2.3.0; python_version >= '3.7'
 tqdm==4.66.2; python_version >= '3.7'
-typing-extensions==4.7.1; python_version >= '3.7'
+typing-extensions==4.7.1; python_version < '3.8'
 urllib3==2.0.7; python_version >= '3.7'
 wcwidth==0.2.13
 wrapt==1.16.0; python_version >= '3.6'

gpt-pr-0.2.0/gptpr/version.py

@@ -1 +0,0 @@
-__version__ = "0.2.0"

{gpt-pr-0.2.0 → gpt-pr-0.2.1}/gpt_pr.egg-info/requires.txt

@@ -16,6 +16,7 @@ exceptiongroup==1.2.0
 
 [:python_version < "3.8"]
 cached-property==1.5.2
+typing-extensions==4.7.1
 
 [:python_version == "3.7"]
 importlib-metadata==6.7.0
@@ -52,7 +53,6 @@ smmap==5.0.1
 sniffio==1.3.1
 termcolor==2.3.0
 tqdm==4.66.2
-typing-extensions==4.7.1
 urllib3==2.0.7
 zipp==3.15.0