darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-01 23:55:58 -05:00
Code Issues Packages Projects Releases Wiki Activity

reformat

Browse Source
This commit is contained in:
Kenneth Kehl
2023-08-29 14:54:30 -07:00
parent 19dcd7a48b
commit 1ecb747c6d
588 changed files with 34100 additions and 23589 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
scripts/check_if_new_migration.py
View File

@@ -6,40 +6,48 @@ 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')]
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', '')
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'))
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']
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 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
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 ''
return ""
def run():
if get_current_db_version() == get_latest_db_migration_to_apply():
print('no')
print("no")
else:
print('yes')
print("yes")
if __name__ == "__main__":