def test_random_gallery_avatar(self):
"""dynamic avatar gets created"""
gallery.load_avatar_galleries()
self.assertNoAvatarIsSet()
gallery.set_random_avatar(self.user)
self.assertAvatarWasSet()
def test_random_gallery_avatar(self):
"""dynamic avatar gets created"""
gallery.load_avatar_galleries()
self.assertNoAvatarIsSet()
gallery.set_random_avatar(self.user)
self.assertAvatarWasSet()
def handle(self, *args, **options):
try:
fake_users_to_create = int(args[0])
except IndexError:
fake_users_to_create = 5
except ValueError:
self.stderr.write("\nOptional argument should be integer.")
sys.exit(1)
fake = Factory.create()
User = get_user_model()
ranks = [r for r in Rank.objects.all()]
message = 'Creating %s fake user accounts...\n'
self.stdout.write(message % fake_users_to_create)
message = '\n\nSuccessfully created %s fake user accounts in %s'
created_count = 0
start_time = time.time()
show_progress(self, created_count, fake_users_to_create)
for i in range(fake_users_to_create):
try:
kwargs = {
'rank': random.choice(ranks),
}
user = User.objects.create_user(fake.first_name(),
fake.email(),
'pass123',
set_default_avatar=False,
**kwargs)
if random.randint(0, 100) > 90:
dynamic.set_avatar(user)
else:
gallery.set_random_avatar(user)
user.avatar_hash = get_avatar_hash(user)
user.save(update_fields=['avatar_hash'])
except (ValidationError, IntegrityError):
pass
else:
created_count += 1
show_progress(self, created_count, fake_users_to_create,
start_time)
total_time = time.time() - start_time
total_humanized = time.strftime('%H:%M:%S', time.gmtime(total_time))
self.stdout.write(message % (created_count, total_humanized))
def handle(self, *args, **options):
try:
fake_users_to_create = int(args[0])
except IndexError:
fake_users_to_create = 5
except ValueError:
self.stderr.write("\nOptional argument should be integer.")
sys.exit(1)
fake = Factory.create()
User = get_user_model()
ranks = [r for r in Rank.objects.all()]
message = 'Creating %s fake user accounts...\n'
self.stdout.write(message % fake_users_to_create)
message = '\n\nSuccessfully created %s fake user accounts in %s'
created_count = 0
start_time = time.time()
show_progress(self, created_count, fake_users_to_create)
for i in range(fake_users_to_create):
try:
kwargs = {
'rank': random.choice(ranks),
}
user = User.objects.create_user(
fake.first_name(), fake.email(), 'pass123',
set_default_avatar=False, **kwargs)
if random.randint(0, 100) > 90:
dynamic.set_avatar(user)
else:
gallery.set_random_avatar(user)
user.avatar_hash = get_avatar_hash(user)
user.save(update_fields=['avatar_hash'])
except (ValidationError, IntegrityError):
pass
else:
created_count += 1
show_progress(
self, created_count, fake_users_to_create, start_time)
total_time = time.time() - start_time
total_humanized = time.strftime('%H:%M:%S', time.gmtime(total_time))
self.stdout.write(message % (created_count, total_humanized))
def handle(self, *args, **options):
items_to_create = options['users']
fake = Factory.create()
ranks = [r for r in Rank.objects.all()]
message = 'Creating %s fake user accounts...\n'
self.stdout.write(message % items_to_create)
message = '\n\nSuccessfully created %s fake user accounts in %s'
created_count = 0
start_time = time.time()
show_progress(self, created_count, items_to_create)
while created_count < items_to_create:
try:
kwargs = {
'rank': random.choice(ranks),
}
user = UserModel.objects.create_user(fake.first_name(),
fake.email(),
'pass123',
set_default_avatar=False,
**kwargs)
if random.randint(0, 100) > 90:
dynamic.set_avatar(user)
else:
gallery.set_random_avatar(user)
user.save(update_fields=['avatars'])
except (ValidationError, IntegrityError):
pass
else:
created_count += 1
show_progress(self, created_count, items_to_create, start_time)
total_time = time.time() - start_time
total_humanized = time.strftime('%H:%M:%S', time.gmtime(total_time))
self.stdout.write(message % (created_count, total_humanized))
def handle(self, *args, **options):
items_to_create = options['users']
fake = Factory.create()
ranks = [r for r in Rank.objects.all()]
message = 'Creating %s fake user accounts...\n'
self.stdout.write(message % items_to_create)
message = '\n\nSuccessfully created %s fake user accounts in %s'
created_count = 0
start_time = time.time()
show_progress(self, created_count, items_to_create)
while created_count < items_to_create:
try:
user = UserModel.objects.create_user(
fake.first_name(),
fake.email(),
'pass123',
set_default_avatar=False,
rank=random.choice(ranks),
)
if random.randint(0, 100) > 90:
dynamic.set_avatar(user)
else:
gallery.set_random_avatar(user)
user.save(update_fields=['avatars'])
except (ValidationError, IntegrityError):
pass
else:
created_count += 1
show_progress(self, created_count, items_to_create, start_time)
total_time = time.time() - start_time
total_humanized = time.strftime('%H:%M:%S', time.gmtime(total_time))
self.stdout.write(message % (created_count, total_humanized))
def test_random_gallery_avatar_no_gallery(self):
"""runtime error is raised when no gallery exists"""
with self.assertRaises(RuntimeError):
gallery.set_random_avatar(self.user)
def test_random_gallery_avatar_no_gallery(self):
"""runtime error is raised when no gallery exists"""
with self.assertRaises(RuntimeError):
gallery.set_random_avatar(self.user)
