Skip to main content

Locust: Track changes to Python code across git refs

Project description

locust

"It's git diff --stat on steroids!" - @scottmilliken

What is Locust?

Locust helps you reason about your code base as it evolves over time.

Locust provides a semantic layer on top of git diff. It emits metadata describing AST-level changes to your code base between git revisions.

This metadata is useful to both humans and computers. For example:

  1. (Humans) Locust can generate much more humane summaries of changes than the standard git diff.

  2. (Computers) Bugout.dev uses Locust metadata to learn high level abstractions about code.

Installation

Locust requires Python3 (specifically, it was written in Python3.8).

Install from PyPI

pip install -U setuptools
pip install bugout-locust

Install from source

Clone this repository and run from the project root:

python setup.py install

Docker

You can also use the Locust docker image:

docker pull bugout/locust

Usage

CLI

Locust is a command line tool, and you can invoke it as:

locust 22dd7fd6adf392bb29d13d10f10e7dbb1d97bfce c9813bd5871a9919551ccd917712135c40367c5c --format yaml

This produces the following output:

locust:
  - file: locust/ci_helpers/github.py
    changes:
      - name: generate_argument_parser
        type: function
        line: 11
        changed_lines: 9
        total_lines: 9
        children: []
      - name: helper_push
        type: function
        line: 22
        changed_lines: 13
        total_lines: 13
        children: []
      - name: helper_pr
        type: function
        line: 37
        changed_lines: 14
        total_lines: 14
        children: []
      - name: main
        type: function
        line: 53
        changed_lines: 29
        total_lines: 29
        children: []
refs:
  initial: 22dd7fd
  terminal: c9813bd

Language plugins

To use Locust to process a code base containing Python (>3.5) and Javascript, use the Javascript plugin (written in Node 14).

If you are running Locust from the root of this project, you would do this as follows:

locust -r <path to repo> <initial revision> <terminal revision> --plugins "node js/out/index.js"

A Locust language plugin is simply a program that you can invoke from the shell (like node js/out/index.js) which takes two arguments:

  • -i - an input file containing a locust.git.RunResponse object

  • -o - path to an output file into which it writes a list of tuples consisting of locust.git.PatchInfo objects and their corresponding list of locust.parse.RawDefinition objects.

The Javascript plugin provides a rubric for how to build your own plugin.

You can add custom plugins to a Locust invocation like this:

locust -r <path to repo> <initial revision> <terminal revision> \
  --plugins "node js/out/index.js" "<custom plugin invocation 1>" "<custom plugin invocation 2>"

CI/CD

Locust is easy to use in CI/CD pipelines:

Bugout App integration

You can use Locust with our Bugout GitHub Bot. Locust extension will work after you push ./github/workflows/locust.yml file in main branch:

name: Locust summary
on: [pull_request_target]
jobs:
  build:
    runs-on: ubuntu-20.04
    steps:
      - name: PR head repo
        id: head_repo_name
        run: |
          HEAD_REPO_NAME=$(jq -r '.pull_request.head.repo.full_name' "$GITHUB_EVENT_PATH")
          echo "PR head repo: $HEAD_REPO_NAME"
          echo "::set-output name=repo::$HEAD_REPO_NAME"
      - name: Checkout git repo
        uses: actions/checkout@v2
        with:
          repository: ${{ steps.head_repo_name.outputs.repo }}
          fetch-depth: 0
      - name: Install python
        uses: actions/setup-python@v2
        with:
          python-version: "3.8"
      - name: Install dependencies
        run: |
          python -m pip install --upgrade pip setuptools
          pip install bugout-locust
      - name: Generate and send Locust summary
        env:
          BUGOUT_SECRET: ${{ secrets.BUGOUT_SECRET }}
        run: |
          locust.github publish

BUGOUT_SECRET should be setted up in repository/organization secrets. Value you can take from Bugout Account token page. Also BUGOUT_API_URL: ${{ secrets.BUGOUT_API_URL }} could be specified if you want to setup your personal server for processing locust summaries.

Docker

To run Locust using docker:

docker run -v $ABSOLUTE_PATH_TO_GIT_REPO:/usr/src/app bugout/locust -r /usr/src/app \
    $INITIAL_REVISION \
    $TERMINAL_REVISION \
    --format yaml

Output formats

Locust can produce output in many formats. The currently supported formats are:

  1. JSON (--format json)

  2. YAML (--format yaml)

  3. HTML (--format html)

  4. GitHub-flavored HTML, meant to be used with GitHub styles (--format html-github)

Contributing

Running tests

Make sure to clone simiotics/locust-test-cases repo to your machine.

Run:

git -c <path to locust-test-cases repo> fetch origin

Then, from the root of this repo:

LOCUST_TESTCASES_DIR=<path to locust-test-cases repo> ./test.sh

Similar projects

Kythe

Kythe is a Google open source project. It grew out of the need to semantically link different parts of Google's code base.

The goals of Locust are different from those of Kythe. Locust is specifically about generating metadata describing changes to code.

Kythe on GitHub: https://github.com/kythe/kythe

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

bugout-locust-0.3.2.tar.gz (21.0 kB view hashes)

Uploaded Source

Built Distribution

bugout_locust-0.3.2-py3-none-any.whl (26.1 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page