election/tools/generate-rolls.sh
Tony Breeds 15ee429ee4 Tool to generate electorate rolls
This is a shell wrapper around 'owners.py' from the system-config repo

Change-Id: I35b089ed9c1f5d8e05d7fc361e002117d145b95e
2016-09-21 10:28:58 +10:00

124 lines
3.2 KiB
Bash

#!/usr/bin/env bash
CGIT_URL=http://git.openstack.org/cgit/openstack-infra/system-config/plain/tools/owners.py
OWNERS=$(basename "$CGIT_URL")
WITH_STABLE=1
while [ $# -gt 0 ] ; do
case "$1" in
--tag)
TAG=$2
shift 1
;;
--before)
BEFORE=$2
shift 1
;;
--after)
AFTER=$2
shift 1
;;
--without-stable)
WITH_STABLE=0
;;
*)
;;
esac
shift 1
done
if [ -z "$TAG" ] ; then
echo $(basename "$0"): You must supply a --tag >&2
exit 1
fi
if [ -z "$AFTER" ] ; then
echo $(basename "$0"): You must supply a --after >&2
exit 1
fi
if [ -z "$BEFORE" ] ; then
echo $(basename "$0"): You must supply a --before >&2
exit 1
fi
ROLLS_DIR="./rolls/${TAG}"
echo "Run information:"
echo " - TAG :" $TAG
echo " - ROLLS :" $ROLLS_DIR
echo " - STABLE :" $WITH_STABLE
echo " - AFTER :" $AFTER
echo " - BEFORE :" $BEFORE
rm -rf "${ROLLS_DIR}"
mkdir -p "${ROLLS_DIR}"
cd $(dirname "${ROLLS_DIR}")
# Building the venv isn't really needed everytime
if [ ! -d venv ] ; then
echo "Building tools venv"
(
# owners.py isn't python3 safe as string.maketrans is now
# str.maketrans
virtualenv -p python2 venv
./venv/bin/pip install -U pip setuptools wheel
./venv/bin/pip install pyyaml requests ndg-httpsclient
) >/dev/null 2>&1
if [ $? -ne 0 ] ; then
echo Creating Virtualenv failed >&2
exit 1
fi
else
echo "Using existing venv"
fi
# Always get the newest script in case a bug has been fixed.
rm -f "./${OWNERS}" >/dev/null 2>&1
echo "Grabbing script from: ${CGIT_URL}"
wget "${CGIT_URL}" -O "./${OWNERS}" -o /dev/null
if [ ! -e "./${OWNERS}" ] ; then
echo Failed to download ${CGIT_URL} >&2
exit 1
fi
echo "Starting roll generation @" $(TZ=UTC date)
# Exclude the system / bot accounts
# OpenStack Release Bot:
## curl https://review.openstack.org/accounts/22816
# OpenStack Proposal Bot
## curl https://review.openstack.org/accounts/11131
venv/bin/python "./${OWNERS}" -a "${AFTER}" -b "${BEFORE}" \
-i 11131 -i 22816 \
-o "${TAG}" -r "${TAG}"
if [ "$WITH_STABLE" -eq 1 ] ; then
TMP_DIR=$(mktemp -d election.XXXXXXX)
echo "Starting (Stable) roll generation @" $(TZ=UTC date)
venv/bin/python "./${OWNERS}" -a "${AFTER}" -b "${BEFORE}" \
-i 11131 -i 22816 \
-o "${TMP_DIR}" -r "${TAG}" \
-n -s 'branch:^stable/.*'
find
cp "${TMP_DIR}/_electorate.txt" "./${TAG}/stable_branch_maintenance.txt"
cp "${TMP_DIR}/_all_owners.yaml" "./${TAG}/stable_branch_maintenance.yaml"
echo "Finished (Stable) roll generation @" $(TZ=UTC date)
rm -rf "${TMP_DIR}"
else
echo Dropping stable_branch_maintenance data
rm "./${TAG}/stable_branch_maintenance.txt" \
"./${TAG}/stable_branch_maintenance.yaml"
fi
echo "Finished roll generation @" $(TZ=UTC date)
echo "Electoral rolls SHA256sum"
case "$(uname -s)" in
Darwin)
shasum -a 256 ./${TAG}/*.txt | tee "${TAG}.shasums"
;;
*)
sha256sum ./${TAG}/*.txt | tee "${TAG}.shasums"
;;
esac