forked from Github-Mirrors/canaille
60 lines
1.2 KiB
Python
60 lines
1.2 KiB
Python
|
import click
|
||
|
|
from canaille.app.commands import with_backendcontext
|
||
|
|
from flask.cli import with_appcontext
|
||
|
|
|
||
|
|
try:
|
||
|
|
HAS_FAKER = True
|
||
|
|
except ImportError: # pragma: no cover
|
||
|
|
HAS_FAKER = False
|
||
|
|
|
||
|
|
|
||
|
|
@click.group()
|
||
|
|
@click.option("--nb", default=1, help="Number of items to create")
|
||
|
|
@click.pass_context
|
||
|
|
@with_appcontext
|
||
|
|
def populate(ctx, nb):
|
||
|
|
"""
|
||
|
|
Populate the database with generated random data.
|
||
|
|
"""
|
||
|
|
ctx.ensure_object(dict)
|
||
|
|
|
||
|
|
ctx.obj["number"] = nb
|
||
|
|
|
||
|
|
|
||
|
|
@populate.command()
|
||
|
|
@click.pass_context
|
||
|
|
@with_appcontext
|
||
|
|
@with_backendcontext
|
||
|
|
def users(ctx):
|
||
|
|
"""
|
||
|
|
Populate the database with generated random users.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from canaille.core.populate import fake_users
|
||
|
|
|
||
|
|
fake_users(ctx.obj["number"])
|
||
|
|
|
||
|
|
|
||
|
|
@populate.command()
|
||
|
|
@click.pass_context
|
||
|
|
@click.option(
|
||
|
|
"--nb-users-max",
|
||
|
|
default=1,
|
||
|
|
help="The maximum number of users that will randomly be affected in the group",
|
||
|
|
)
|
||
|
|
@with_appcontext
|
||
|
|
@with_backendcontext
|
||
|
|
def groups(ctx, nb_users_max):
|
||
|
|
"""
|
||
|
|
Populate the database with generated random groups.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from canaille.core.populate import fake_groups
|
||
|
|
|
||
|
|
fake_groups(ctx.obj["number"], nb_users_max)
|
||
|
|
|
||
|
|
|
||
|
|
def register(cli):
|
||
|
|
if HAS_FAKER: # pragma: no branch
|
||
|
|
cli.add_command(populate)
|