summaryrefslogtreecommitdiff
path: root/tests/user_commands/management/commands/hal.py
blob: 6b9588b053c67f1d8b91540909c591acc377e55d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
from django.core.management.base import BaseCommand, CommandError


class Command(BaseCommand):
    help = "Useless command."

    def add_arguments(self, parser):
        parser.add_argument(
            "args",
            metavar="app_label",
            nargs="*",
            help="Specify the app label(s) to works on.",
        )
        parser.add_argument("--empty", action="store_true", help="Do nothing.")

    def handle(self, *app_labels, **options):
        app_labels = set(app_labels)

        if options["empty"]:
            self.stdout.write()
            self.stdout.write("Dave, I can't do that.")
            return

        if not app_labels:
            raise CommandError("I'm sorry Dave, I'm afraid I can't do that.")

        # raise an error if some --parameter is flowing from options to args
        for app_label in app_labels:
            if app_label.startswith("--"):
                raise CommandError("Sorry, Dave, I can't let you do that.")

        self.stdout.write("Dave, my mind is going. I can feel it. I can feel it.")