Skip to content

Add a GitHub workflow to comment the SQL generated by a migration in PRs #21

Add a GitHub workflow to comment the SQL generated by a migration in PRs

Add a GitHub workflow to comment the SQL generated by a migration in PRs #21

# Generate migration as SQL if necessary
# https://github.com/actions/github-script#comment-on-an-issue
name: "Migration as SQL"
on:
workflow_dispatch:
pull_request:
paths:
- alembic/versions/**
jobs:
comment:
runs-on: ubuntu-latest
permissions:
pull-requests: write
steps:
- name: "Checkout repository"
uses: actions/checkout@692973e3d937129bcbf40652eb9f2f61becf3332 # v4.1.7
with:
# needed to diff with the main branch later
fetch-depth: 0
- name: "Setup PDM"
uses: pdm-project/setup-pdm@568ddd69406b30de1774ec0044b73ae06e716aa4 # v4
with:
python-version: "3.12"
cache: true
- name: "Install dependencies"
run: pdm install --dev
- name: "Generate SQL"
id: "sql"
run: |
#!/bin/bash
set -euo pipefail
# we need to find the down revision of the first migration made in this PR,
# which should be the head of main. to do this, we are going to import the
# migration file which will set globals that we can easily read
#
# see https://docs.python.org/3/library/importlib.html#importing-a-source-file-directly
git diff --name-only origin/main -- alembic/versions | xargs pdm run python -c '
import importlib.util
import sys
spec = importlib.util.spec_from_file_location("rev", sys.argv[1])
module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(module)
print(f"{module.revision},{module.down_revision}")
' | pdm run python -c '
import sys
down_revs = set()
revs = set()
lines = (line.split(",") for line in sys.stdin)
for revision, down_revision in lines:
down_revs.add(down_revision)
revs.add(revision)
head ,= down_revs - revs
print(head)
' > phr
previous_head_revision=$(cat phr)
# `alembic upgrade --sql` outputs logs to stderr and the sql to stdout, so
# make temp files to get both parts
pdm run alembic upgrade --sql $previous_head_revision:head > sql 2> summary
EOF=$(dd if=/dev/urandom bs=15 count=1 status=none | base64)
echo "MIGRATION<<$EOF" >> $GITHUB_OUTPUT
echo "$(cat summary sql)" >> $GITHUB_OUTPUT
echo "$EOF" >> $GITHUB_OUTPUT
- name: "Comment on PR"
uses: actions/github-script@60a0d83039c74a4aee543508d2ffcb1c3799cdea # 7.0.1
with:
script: |
github.rest.issues.createComment({
issue_number: context.issue.number,
owner: context.repo.owner,
repo: context.repo.repo,
body: "test"
})