def handle(self, url, *args, **options): verbosity = int(options.get('verbosity', 1)) traceback = options.get('traceback', False) if verbosity > 1: self.stdout.write('Importing {0}'.format(url)) try: Cookie.objects.import_from_url(url) self.stdout.write('Imported {0}'.format(url)) except ValueError as exc: ce = CommandError(str(exc).format(url)) if traceback: tb = sys.exc_info()[2] raise ce.with_traceback(tb) else: raise ce
def handle(self, username, **options): traceback = options.get('traceback', False) try: user = BakeryUser.objects.get(username=username) user.is_staff = True user.is_superuser = True user.save(update_fields=['is_staff', 'is_superuser']) self.stdout.write('Updated {0} to superuser status'.format(username)) except (BakeryUser.DoesNotExist, ValueError) as exc: ce = CommandError(exc) if traceback: tb = sys.exc_info()[2] raise ce.with_traceback(tb) else: raise ce