You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
3.4 KiB
64 lines
3.4 KiB
8 months ago
|
import sys
|
||
|
|
import argparse
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
def update_fork():
|
||
|
|
# Sync your fork's main branch with the original repository's main branch
|
||
|
|
print("Updating fork...")
|
||
|
|
subprocess.run(['git', 'fetch', 'upstream'], check=True) # Fetch the branches and their respective commits from the upstream repository
|
||
|
|
subprocess.run(['git', 'checkout', 'main'], check=True) # Switch to your local main branch
|
||
|
|
subprocess.run(['git', 'merge', 'upstream/main'], check=True) # Merge changes from upstream/main into your local main branch
|
||
|
|
subprocess.run(['git', 'push', 'origin', 'main'], check=True) # Push the updated main branch to your fork on GitHub
|
||
|
|
print("Fork updated successfully.")
|
||
|
|
|
||
|
|
def push_changes(branch_name, commit_message):
|
||
|
|
# Push your local changes to your fork on GitHub
|
||
|
|
print("Pushing changes to fork...")
|
||
|
|
subprocess.run(['git', 'checkout', branch_name], check=True) # Switch to the branch where your changes are
|
||
|
|
subprocess.run(['git', 'add', '.'], check=True) # Stage all changes for commit
|
||
|
|
subprocess.run(['git', 'commit', '-m', commit_message], check=True) # Commit the staged changes with a custom message
|
||
|
|
subprocess.run(['git', 'push', 'origin', branch_name], check=True) # Push the commit to the same branch in your fork
|
||
|
|
print("Changes pushed successfully.")
|
||
|
|
|
||
|
|
def create_pull_request(pr_title, pr_file, branch_name):
|
||
|
|
# Create a pull request on GitHub using the GitHub CLI
|
||
|
|
print("Creating pull request...")
|
||
|
|
with open(pr_file, 'r') as file:
|
||
|
|
pr_body = file.read() # Read the PR description from a markdown file
|
||
|
|
subprocess.run(['gh', 'pr', 'create',
|
||
|
|
'--base', 'main',
|
||
|
|
'--head', f'{branch_name}',
|
||
|
|
'--title', pr_title,
|
||
|
|
'--body', pr_body], check=True) # Create a pull request with the specified title and markdown body
|
||
|
|
print("Pull request created successfully.")
|
||
|
|
|
||
|
|
def main():
|
||
|
|
parser = argparse.ArgumentParser(description="Automate your GitHub workflow")
|
||
|
|
subparsers = parser.add_subparsers(dest='command', help='Available commands')
|
||
|
|
|
||
|
|
# Subparser for updating fork
|
||
|
|
parser_update = subparsers.add_parser('update-fork', help="Update fork with the latest from the original repository")
|
||
|
|
|
||
|
|
# Subparser for pushing changes
|
||
|
|
parser_push = subparsers.add_parser('push-changes', help="Push local changes to the fork")
|
||
|
|
parser_push.add_argument('--branch-name', required=True, help="The name of the branch you are working on")
|
||
|
|
parser_push.add_argument('--commit-message', required=True, help="The commit message for your changes")
|
||
|
|
|
||
|
|
# Subparser for creating a pull request
|
||
|
|
parser_pr = subparsers.add_parser('create-pr', help="Create a pull request to the original repository")
|
||
|
|
parser_pr.add_argument('--branch-name', required=True, help="The name of the branch the pull request is from")
|
||
|
|
parser_pr.add_argument('--pr-title', required=True, help="The title of your pull request")
|
||
|
|
parser_pr.add_argument('--pr-file', required=True, help="The markdown file path for your pull request description")
|
||
|
|
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
if args.command == 'update-fork':
|
||
|
|
update_fork()
|
||
|
|
elif args.command == 'push-changes':
|
||
|
|
push_changes(args.branch_name, args.commit_message)
|
||
|
|
elif args.command == 'create-pr':
|
||
|
|
create_pull_request(args.branch_name, args.pr_title, args.pr_file)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|