diff --git a/awx/main/management/commands/check_db.py b/awx/main/management/commands/check_db.py new file mode 100644 index 0000000000..e72b86873c --- /dev/null +++ b/awx/main/management/commands/check_db.py @@ -0,0 +1,17 @@ +# Copyright (c) 2015 Ansible, Inc. +# All Rights Reserved + +from django.core.management.base import BaseCommand +from django.db import connection + + +class Command(BaseCommand): + """Checks connection to the database, and prints out connection info if not connected""" + + def handle(self, *args, **options): + + with connection.cursor() as cursor: + cursor.execute("SELECT version()") + version = str(cursor.fetchone()[0]) + + return "Database Version: {}".format(version)