コード例 #1
0
ファイル: create.py プロジェクト: xSandie/snnudj
    def init():
        # 创建第一个管理员
        click.echo("Initializing roles and permissions")
        Roles.init_role()

        click.echo("Initializing the first admin")
        with db.auto_commit():
            admin = User()
            admin.username = '******'
            admin.userPhone = '13572856767'
            admin.password = '******'
            admin.admin = True
            admin.roleId = Roles.Administrator
            db.session.add(admin)
        click.echo("Done")

        click.echo("Initializing the first moderator")
        with db.auto_commit():
            admin = User()
            admin.username = '******'
            admin.userPhone = '15594989021'
            admin.password = '******'
            admin.admin = True
            admin.roleId = Roles.Moderator
            db.session.add(admin)
        click.echo("Done")
コード例 #2
0
    def create_test_user() -> dict:
        """Create a test fuel supplier user"""
        user = User()

        generated_name = 'test_{0}'.format(str(uuid.uuid4())[8:])
        user.username = generated_name

        user.first_name = 'Test'
        user.last_name = 'User'
        user.display_name = 'Test User'

        user.organization = Organization.objects.get_by_natural_key(
            DataCreationUtilities.create_test_organization()['name'])

        user.save()
        user.refresh_from_db()

        return {'id': user.id, 'username': user.username}