peeringdb-peeringdb/peeringdb_server/management/commands/pdb_maintenance.py

from django.core.management.base import BaseCommand
from django.conf import settings
from peeringdb_server import maintenance


class Command(BaseCommand):
    help = "Put instance in or out of maintenance mode"

    def add_arguments(self, parser):
        parser.add_argument("state", nargs="?", choices=["on", "off"])
        parser.add_argument(
            "--commit", action="store_true", help="will commit the changes"
        )

    def log(self, msg):
        if self.commit:
            self.stdout.write(msg)
        else:
            self.stdout.write("[pretend] {}".format(msg))

    def handle(self, *args, **options):
        self.commit = options.get("commit", False)
        self.state = options.get("state")

        if not settings.TUTORIAL_MODE:
            self.log(
                "Command cannot be run on environment's that are not in tutorial mode. "
                " Maintenance mode "
                " is currently implemented to the extent that it is required to facilitate"
                " an environment reset on `tutorial` type servers and probably needs more work"
                " to be useful on production."
            )
            return

        self.log("Setting maintenance mode {}".format(self.state))
        if self.state == "on":
            maintenance.on()
        else:
            maintenance.off()