def ban_user(self): new_ban = Ban(banned_value=self.user.username, user_message=self.cleaned_data['user_message'], staff_message=self.cleaned_data['staff_message'], valid_until=self.cleaned_data['valid_until']) new_ban.save() Ban.objects.invalidate_cache()
def handle(self, *args, **options): try: fake_bans_to_create = int(args[0]) except IndexError: fake_bans_to_create = 5 except ValueError: self.stderr.write("\nOptional argument should be integer.") sys.exit(1) fake = Factory.create() message = 'Creating %s fake bans...\n' self.stdout.write(message % fake_bans_to_create) message = '\n\nSuccessfully created %s fake bans' created_count = 0 show_progress(self, created_count, fake_bans_to_create) for i in range(fake_bans_to_create): ban = Ban(check_type=random.randint(Ban.USERNAME, Ban.IP)) ban.banned_value = create_fake_test(fake, ban.check_type) if random.randint(0, 10) == 0: ban.user_message = fake.sentence() if random.randint(0, 10) == 0: ban.staff_message = fake.sentence() if random.randint(0, 1): # Lets make ban temporary ban_length = timedelta(days=random.randint(0, 300)) if random.randint(0, 1): ban.valid_until = timezone.now().date() - ban_length else: ban.valid_until = timezone.now().date() + ban_length ban.save() created_count += 1 show_progress(self, created_count, fake_bans_to_create) self.stdout.write(message % created_count)
def handle(self, *args, **options): try: fake_bans_to_create = int(args[0]) except IndexError: fake_bans_to_create = 5 except ValueError: self.stderr.write("\nOptional argument should be integer.") sys.exit(1) fake = Factory.create() message = 'Creating %s fake bans...\n' self.stdout.write(message % fake_bans_to_create) message = '\n\nSuccessfully created %s fake bans' created_count = 0 show_progress(self, created_count, fake_bans_to_create) for i in xrange(fake_bans_to_create): ban = Ban(check_type=random.randint(BAN_USERNAME, BAN_IP)) ban.banned_value = create_fake_test(fake, ban.check_type) if random.randint(0, 10) == 0: ban.user_message = fake.sentence() if random.randint(0, 10) == 0: ban.staff_message = fake.sentence() if random.randint(0, 1): # Lets make ban temporary ban_length = timedelta(days=random.randint(0, 300)) if random.randint(0, 1): ban.valid_until = timezone.now().date() - ban_length else: ban.valid_until = timezone.now().date() + ban_length ban.save() created_count += 1 show_progress(self, created_count, fake_bans_to_create) self.stdout.write(message % created_count)