Now user_add accepts role as arg

From a fix by Olivier Thauvin <olivier.thauvin@latmos.ipsl.fr>

Now user_add accepts role as arg
From a fix by Olivier Thauvin <olivier.thauvin@latmos.ipsl.fr>
hitier
1 parent 82642adb
Showing 1 changed file with 12 additions and 5 deletions Show diff stats
app/commands/commands.py
@@ -12,7 +12,7 @@ from sqlalchemy.orm import Session
 from sqlalchemy import create_engine
 from app.models import db, Agent, Service, Project, Capacity, Period, Charge
-from app.auth.models import User
+from app.auth.models import User, _nameToRole
 from . import bp
@@ -220,18 +220,25 @@ def create_db():
 @click.argument('name')
 @click.argument('login')
 @click.argument('password')
-def user_add(email, name, login, password):
+@click.argument('role')
+def user_add(email, name, login, password, role):
     """ Add a new user in db."""
-    user = User.query.filter(User.name==name).one_or_none()
-    if( user ):
+    user = User.query.filter(User.name == name).one_or_none()
+    if (user):
         current_app.logger.error(f"user already exists {name}")
         return
-    user = User(email=email, name=name, login=login, password=password)
+    user = User(email=email, name=name, login=login, password=password, role=role)
     db.session.add(user)
     db.session.commit()
     current_app.logger.info(f"added {name}")
+@bp.cli.command('show_roles')
+def show_roles():
+    """ List all available roles for a user"""
+    print("\n".join(list(_nameToRole)))
+
+
 @bp.cli.command('user_show_all')
 def user_show_all():
     """ Show all users in db."""