Skip to content

Package in need of a better name to automate opening pull requests 🤖

License

Notifications You must be signed in to change notification settings

astrofrog/batchpr

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

50 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

About

The aim of this package is to provide an easy way to do automated issues or pull requests to a selection of repositories and make specific changes to them. This is currently functional but could be significantly improved, so contributions are welcome!

Installation

To install the latest development version with pip:

pip install git+https://github.com/astrofrog/batchpr.git@master

To install from source:

git clone https://github.com/astrofrog/batchpr.git
cd batchpr
pip install .

Automated Pull Requests

batchpr requires the following packages:

  • git
  • pygithub
  • requests
  • termcolor

To use this, you should write a Python script in which you import and subclass the Updater class, and define the following methods and properties:

from batchpr import Updater

class MyUpdater(Updater):

    def process_repo(self):
        # This method should contain any code that you want to run inside
        # the repository to make the changes/updates. You can assume that
        # the current working directory is the repository being processed.
        # This method should return False if it was not able to make the
        # changes, and True if it was. This method should call self.add
        # to git add any files that have changed, but should not commit.

    @property
    def commit_message(self):
        # The commit message to use when making the changes

    @property
    def pull_request_title(self):
        # The title of the pull request

    @property
    def pull_request_body(self)
        # The main body/description of the pull request

    @property
    def branch_name(self):
        # The name of the branch to use

Once you have defined your updater class, you can run it with:

helper = MyUpdater(token=GITHUB_TOKEN)
helper.run('username/repo')

Where GITHUB_TOKEN is a personal access token for GitHub. If you want to customize the author of the commit, you can do this with:

helper = MyUpdater(token=GITHUB_TOKEN, author_name='Foo', author_email='foo@bar.bar')
helper.run('username/repo')

The run method can take a single repository or a list of repositories. If dry_run=True option is passed in to the Updater subclass, pull requests will not be opened but other prior steps (i.e., forking, branching, and committing the changes) are executed. If verbose=True option is passed into the Updater subclass, you will always see output of the commands, not just when a command fails.

When in the Updater class, the following methods are available:

  • self.run_command(command): should be used for running shell commands (e.g., git)
  • self.warn(message): should be used for warning messages
  • self.error(message): should be used for error messages
  • self.add(filename): should be used to add files that have changed or are new
  • self.copy(filename1, filename2): can be used to copy files

Full Example

The following shows an example of an updater that adds a few sentences from the zen of Python to the README file if present:

import os
from batchpr import Updater

DESCRIPTION = """
This is an automated update made by the ``batchpr`` tool :robot: - feel free to
close if it doesn't look good! You can report issues to @astrofrog.
"""

ADDITION = """
Beautiful is better than ugly.
Explicit is better than implicit.
Simple is better than complex.
Complex is better than complicated.
"""

class ExampleUpdater(Updater):

    def process_repo(self):

        if os.path.exists('README.md'):
            with open('README.md', 'a') as f:
                f.write(os.linesep + ADDITION)
            self.add('README.md')
            return True
        else:
            return False

    @property
    def commit_message(self):
        return "MNT: Add important text to README.rst"

    @property
    def branch_name(self):
        return 'readme-zen'

    @property
    def pull_request_title(self):
        return self.commit_message

    @property
    def pull_request_body(self):
        return DESCRIPTION.strip()

helper = ExampleUpdater(token=GITHUB_TOKEN)
helper.run('username/repo')

Automated Issues

Opening automated issues is simpler than pull requests as you do not need to create forks or modify any files. You use the same GitHub token as above.

Full Example

The following shows an example of opening a simple issue.

from batchpr import IssueUpdater

ISSUE_TITLE = 'Please fix this and that'

ISSUE_BODY = """
I found this and that with the package. Please fix them.

*This is an automated issue. If this is opened in error, please let me know!*
"""

helper = IssueUpdater(GITHUB_TOKEN, ISSUE_TITLE, ISSUE_BODY)
helper.run('username/repo')

About

Package in need of a better name to automate opening pull requests 🤖

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages