canaille-globuzma/canaille/core/populate.py

import random

import faker
import ldap
from canaille.core.models import Group
from canaille.core.models import User
from canaille.utils.i18n import available_language_codes
from faker.config import AVAILABLE_LOCALES


def faker_generator(locales=None):
    locales = locales or list(set(available_language_codes()) & set(AVAILABLE_LOCALES))
    return [faker.Faker(locale) for locale in locales]


def fake_users(nb=1):
    fakes = faker_generator()
    users = list()
    for _ in range(nb):
        try:
            fake = random.choice(fakes)
            name = fake.unique.name()
            user = User(
                formatted_name=name,
                given_name=name.split(" ")[0],
                family_name=name.split(" ")[1],
                user_name=fake.unique.user_name(),
                email=fake.unique.email(),
                phone_number=fake.unique.ssn(),
                profile_url=fake.unique.uri(),
                address=fake.unique.address(),
                street=fake.street_name(),
                postal_code=fake.postcode(),
                locality=fake.city(),
                region=fake.state(),
                employee_number=str(fake.unique.random_number()),
                department=fake.word(),
                title=fake.job(),
                password=fake.password(),
                preferred_language=fake._locales[0],
            )
            user.save()
            users.append(user)
        except ldap.ALREADY_EXISTS:  # pragma: no cover
            pass
    return users


def fake_groups(nb=1, nb_users_max=1):
    fake = faker_generator(["en_US"])[0]
    users = User.query()
    groups = list()
    for _ in range(nb):
        try:
            group = Group(
                cn=fake.unique.word(),
                description=fake.sentence(),
            )
            nb_users = random.randrange(1, nb_users_max + 1)
            group.member = list({random.choice(users) for _ in range(nb_users)})
            group.save()
            groups.append(group)
        except ldap.ALREADY_EXISTS:  # pragma: no cover
            pass
    return groups