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")
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}