bookwyrm/bw-dev

181 lines
4.5 KiB
Text
Raw Normal View History

#!/bin/bash
# exit on errors
set -e
# import our ENV variables
# catch exits and give a friendly error message
function showerr {
echo "Failed to load configuration! You may need to update your .env and quote values with special characters in them."
}
trap showerr EXIT
source .env
trap - EXIT
function clean {
docker-compose stop
docker-compose rm -f
}
function runweb {
docker-compose run --rm web "$@"
}
function execdb {
docker-compose exec db $@
}
function execweb {
docker-compose exec web "$@"
}
function initdb {
execweb python manage.py migrate
execweb python manage.py initdb
}
2021-03-08 03:43:36 +00:00
function makeitblack {
2021-03-08 18:10:30 +00:00
docker-compose run --rm web black celerywyrm bookwyrm
2021-03-08 03:43:36 +00:00
}
2021-06-08 16:19:52 +00:00
function awscommand {
# expose env vars
2021-06-07 19:40:37 +00:00
export AWS_ACCESS_KEY_ID=${AWS_ACCESS_KEY_ID}
export AWS_SECRET_ACCESS_KEY=${AWS_SECRET_ACCESS_KEY}
export AWS_DEFAULT_REGION=${AWS_S3_REGION_NAME}
2021-06-08 16:19:52 +00:00
# first arg is mountpoint, second is the whole aws command
docker run --rm -it -v $1\
2021-06-07 19:40:37 +00:00
-e AWS_ACCESS_KEY_ID -e AWS_SECRET_ACCESS_KEY -e AWS_DEFAULT_REGION\
2021-06-08 16:19:52 +00:00
amazon/aws-cli $2
2021-06-07 21:20:27 +00:00
}
CMD=$1
if [ -n "$CMD" ]; then
shift
fi
# show commands as they're executed
set -x
case "$CMD" in
up)
docker-compose up --build "$@"
;;
run)
docker-compose run --rm --service-ports web
;;
initdb)
initdb
;;
resetdb)
clean
# Start just the DB so no one else is using it
docker-compose up --build -d db
execdb dropdb -U ${POSTGRES_USER} ${POSTGRES_DB}
execdb createdb -U ${POSTGRES_USER} ${POSTGRES_DB}
# Now start up web so we can run the migrations
docker-compose up --build -d web
initdb
clean
;;
makemigrations)
runweb python manage.py makemigrations "$@"
;;
migrate)
runweb python manage.py migrate "$@"
;;
bash)
runweb bash
;;
shell)
runweb python manage.py shell
;;
dbshell)
execdb psql -U ${POSTGRES_USER} ${POSTGRES_DB}
;;
restart_celery)
docker-compose restart celery_worker
;;
test)
runweb coverage run --source='.' --omit="*/test*,celerywyrm*,bookwyrm/migrations/*" manage.py test "$@"
;;
pytest)
2021-05-28 00:29:24 +00:00
execweb pytest --no-cov-on-fail "$@"
;;
collectstatic)
runweb python manage.py collectstatic --no-input
;;
makemessages)
2021-06-06 20:58:41 +00:00
runweb django-admin makemessages --no-wrap --ignore=venv --all $@
;;
compilemessages)
runweb django-admin compilemessages --ignore venv $@
;;
build)
docker-compose build
;;
clean)
clean
;;
2021-03-08 03:43:36 +00:00
black)
makeitblack
;;
2021-04-02 18:20:56 +00:00
populate_streams)
2021-08-08 00:38:07 +00:00
runweb python manage.py populate_streams $@
2021-03-28 18:46:07 +00:00
;;
populate_suggestions)
runweb python manage.py populate_suggestions
;;
2021-05-26 11:52:10 +00:00
generate_preview_images)
runweb python manage.py generate_preview_images $@
2021-05-26 11:52:10 +00:00
;;
2021-06-07 19:40:37 +00:00
copy_media_to_s3)
2021-06-08 16:19:52 +00:00
awscommand "bookwyrm_media_volume:/images"\
"s3 cp /images s3://${AWS_STORAGE_BUCKET_NAME}/images\
--endpoint-url ${AWS_S3_ENDPOINT_URL}\
--recursive --acl public-read"
2021-06-07 19:40:37 +00:00
;;
2021-06-07 21:20:27 +00:00
set_cors_to_s3)
2021-06-08 16:19:52 +00:00
awscommand "$(pwd):/bw"\
"s3api put-bucket-cors\
--bucket ${AWS_STORAGE_BUCKET_NAME}\
--endpoint-url ${AWS_S3_ENDPOINT_URL}\
--cors-configuration file:///bw/$@"
2021-06-07 21:20:27 +00:00
;;
runweb)
runweb "$@"
;;
rundb)
rundb "$@"
;;
*)
set +x # No need to echo echo
2021-06-19 15:18:13 +00:00
echo "Unrecognised command. Try:"
echo " up [container]"
echo " run"
echo " initdb"
echo " resetdb"
echo " makemigrations [migration]"
echo " migrate [migration]"
echo " bash"
echo " shell"
echo " dbshell"
echo " restart_celery"
echo " test [path]"
echo " pytest [path]"
echo " collectstatic"
echo " makemessages [locale]"
echo " compilemessages [locale]"
echo " build"
echo " clean"
echo " black"
2021-08-08 00:38:07 +00:00
echo " populate_streams [--stream=<stream name>]"
2021-08-03 00:55:30 +00:00
echo " populate_suggestions"
echo " generate_preview_images [--all]"
echo " copy_media_to_s3"
echo " set_cors_to_s3 [cors file]"
echo " runweb [command]"
echo " rundb [command]"
;;
esac