예제 #1
0
 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
예제 #2
0
 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