from django.core.management.base import BaseCommand
import django.db
import sys


class Command(BaseCommand):
    """Management command for healthchecks"""
    def handle(self, *args, **options):
        """Ensure DB and other things are alive"""
        print("Checking if database is alive")
        try:
            django.db.connection.cursor()
            print("Database is alive!")
        except Exception as e:
            print("There was a problem connecting to the database", e)
            sys.exit(1)