refactor: use modern faker calls

https://github.com/joke2k/faker/pull/1991
This commit is contained in:
Éloi Rivard 2024-02-16 09:15:37 +01:00
parent 30921f8249
commit 9af9a303ba
No known key found for this signature in database
GPG key ID: 7EDA204EA57DD184

View file

@ -8,15 +8,14 @@ from faker.config import AVAILABLE_LOCALES
def faker_generator(locales=None): def faker_generator(locales=None):
locales = locales or list(set(available_language_codes()) & set(AVAILABLE_LOCALES)) locales = locales or list(set(available_language_codes()) & set(AVAILABLE_LOCALES))
return [faker.Faker(locale) for locale in locales] return faker.Faker(locales)
def fake_users(nb=1): def fake_users(nb=1):
fakes = faker_generator()
users = list() users = list()
for _ in range(nb): for _ in range(nb):
try: try:
fake = random.choice(fakes) fake = faker_generator()
name = fake.unique.name() name = fake.unique.name()
user = models.User( user = models.User(
formatted_name=name, formatted_name=name,
@ -45,10 +44,10 @@ def fake_users(nb=1):
def fake_groups(nb=1, nb_users_max=1): def fake_groups(nb=1, nb_users_max=1):
fake = faker_generator(["en_US"])[0]
users = models.User.query() users = models.User.query()
groups = list() groups = list()
for _ in range(nb): for _ in range(nb):
fake = faker_generator(["en_US"])
try: try:
group = models.Group( group = models.Group(
display_name=fake.unique.word(), display_name=fake.unique.word(),