Example #1
0
    def handle(self, *args, **options):
        verbosity = int(options.get('verbosity'))
        dry_run = options.get('dry_run')
        site_id = int(options.get('site_id') or settings.SITE_ID)

        if not Site.objects.filter(pk=site_id).exists():
            if verbosity > 0:
                self.stdout.write(self.style.HTTP_BAD_REQUEST("That site ID "
                                  "doesn't exist in the database."))
            return

        if dry_run and verbosity > 0:
            self.stdout.write(self.style.HTTP_BAD_REQUEST("This is a "
                              "dry-run, nothing new will be installed "
                              "into the database"))

        if not dry_run:
            ensure_default_for_site(model=MenuItem, site_id=site_id)
        url_iterables = [x.instance.get_urls() for x in _collect_menus()]
        all_urls = frozenset(chain(*url_iterables))

        if verbosity > 1:
            self.stdout.write(self.style.HTTP_REDIRECT("The following URLs "
                              "have been automatically discovered and will "
                              "be installed if not already in the database"))
            for possible_insert in all_urls:
                self.stdout.write(self.style.HTTP_NOT_FOUND(
                                  possible_insert.path))

        the_missing = find_missing(model=MenuItem, urls=all_urls,
                                   site_id=site_id)

        # no missing things, so fail early.
        if the_missing is None:
            if verbosity > 0:
                self.stdout.write(self.style.HTTP_REDIRECT("No URLs need "
                                  "to be added, yay!"))
                return

        actually_missing = frozenset(the_missing)

        # fail early as there is nothing to consider
        if len(actually_missing) == 0:
            if verbosity > 0:
                self.stdout.write(self.style.HTTP_REDIRECT("No URLs need "
                                  "to be added, yay!"))
            return
        else:
            # print wtf is going to happen
            if verbosity > 0:
                self.stdout.write(self.style.HTTP_REDIRECT("The following "
                                  "URLs are missing and will be "
                                  "installed."))
                for missing in actually_missing:
                    self.stdout.write(self.style.HTTP_NOT_FOUND(
                                      missing.path))
            # possibly do inserts
            if not dry_run:
                responses = add_urls(model=MenuItem, urls=actually_missing,
                                     site_id=site_id)
            else:
                responses = actually_missing

            if responses is not None:
                responses = tuple(responses)
            if verbosity > 0 and responses is not None:
                count = len(responses)
                self.stdout.write(self.style.HTTP_REDIRECT("{0} URLs have "
                                  "been added".format(count)))
            return