mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-10 15:22:24 -05:00
b01ec05aaf
normally we check the app's status page to see if migrations need running. However, if the _status endpoint doesn't respond with 200, we don't necessarily want to abort the deploy - we may be trying to deploy a code fix that fixes that status endpoint for example. We don't know whether to run the migrations or not, so err on the side of caution by re-running the migration. The migration itself might be the fix that gets the app working after all. had to do a little song and dance because sometimes the response won't be populated before an exception is thrown
47 lines
1.8 KiB
Python
47 lines
1.8 KiB
Python
import os
|
|
from os.path import dirname, abspath
|
|
import requests
|
|
import sys
|
|
|
|
|
|
def get_latest_db_migration_to_apply():
|
|
project_dir = dirname(dirname(abspath(__file__))) # Get the main project directory
|
|
migrations_dir = '{}/migrations/versions/'.format(project_dir)
|
|
migration_files = [migration_file for migration_file in os.listdir(migrations_dir) if migration_file.endswith('py')]
|
|
# sometimes there's a trailing underscore, if script was created with `flask db migrate --rev-id=...`
|
|
latest_file = sorted(migration_files, reverse=True)[0].replace('_.py', '').replace('.py', '')
|
|
return latest_file
|
|
|
|
|
|
def get_current_db_version():
|
|
api_status_url = '{}/_status'.format(os.getenv('API_HOST_NAME'))
|
|
|
|
try:
|
|
response = requests.get(api_status_url)
|
|
response.raise_for_status()
|
|
current_db_version = response.json()['db_version']
|
|
return current_db_version
|
|
except requests.exceptions.ConnectionError:
|
|
print(f'Could not make web request to {api_status_url}', file=sys.stderr)
|
|
return ''
|
|
except Exception: # we expect these to be either either a http status code error, or a json decoding error
|
|
print(
|
|
f'Could not read status endpoint!\n\ncode {response.status_code}\nresponse "{response.text}"',
|
|
file=sys.stderr
|
|
)
|
|
# if we can't make a request to the API, the API is probably down. By returning a blank string (which won't
|
|
# match the filename of the latest migration), we force the migration to run, as the code change to fix the api
|
|
# might involve a migration file.
|
|
return ''
|
|
|
|
|
|
def run():
|
|
if get_current_db_version() == get_latest_db_migration_to_apply():
|
|
print('no')
|
|
else:
|
|
print('yes')
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run()
|