Revision 1d91a6e1350831bf6f9f0e7ff23be4cec2ce8fd2 authored by Dave Eargle on 18 March 2021, 22:15:53 UTC, committed by GitHub on 18 March 2021, 22:15:53 UTC
Fix backwards compatibility for translating ON_HEROKU=1 to ON_CLOUD=1
do_heroku_setup.py
from __future__ import generator_stop
import os
import subprocess
import shutil
def _check_heroku_cmd_found():
command = 'heroku'
if not shutil.which(command):
from psiturk.psiturk_exceptions import HerokuCmdNotFound
raise HerokuCmdNotFound()
def _check_heroku_logged_in():
if subprocess.getstatusoutput('heroku auth:whoami')[0]:
from psiturk.psiturk_exceptions import HerokuNotLoggedIn
raise HerokuNotLoggedIn()
def _check_is_git_repo():
if subprocess.getstatusoutput('git rev-parse --git-dir')[0]:
from psiturk.psiturk_exceptions import HerokuNotAGitRepo
raise HerokuNotAGitRepo()
def _check_heroku_app_set():
if subprocess.getstatusoutput('heroku config')[0]:
from psiturk.psiturk_exceptions import HerokuAppNotSet
raise HerokuAppNotSet()
def _set_heroku_config_vars():
from psiturk.psiturk_config import PsiturkConfig
CONFIG = PsiturkConfig()
CONFIG.load_config()
subprocess.call(['heroku', 'config:set', 'ON_CLOUD=true'])
print()
print('psiturk has finished setting heroku config vars.')
def _add_postgresql_db():
subprocess.call(['heroku', 'addons:create', 'heroku-postgresql:hobby-dev'])
print()
print("\n".join([
"Your psiturk heroku app will use the just-added postgresql database instead of whatever is set in your config.txt.",
"If you don't want this, then change or unset the heroku config var 'DATABASE_URL'",
"and/or remove the heroku postgresql addon."]))
HEROKU_FILES_DIR = os.path.join(os.path.dirname(__file__), "heroku_files")
def _copy_heroku_files():
"""
Heroku needs a Procfile, a requirements.txt, and a file that stores the
"""
_TARGET = os.curdir
from distutils import dir_util
print("Copying {} to {}".format(HEROKU_FILES_DIR, _TARGET))
dir_util.copy_tree(HEROKU_FILES_DIR, _TARGET)
print('Remember to commit these new files to your git repository.')
def do_heroku_setup():
_check_heroku_cmd_found()
_check_heroku_logged_in()
_check_is_git_repo()
_check_heroku_app_set()
_set_heroku_config_vars()
_copy_heroku_files()
# _add_postgresql_db()
print()
print("Heroku config done.")
print()
print("Don't forget that your database needs to be a persistent datastore!")
if __name__ == '__main__':
do_heroku_setup()
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...