Package commands :: Module add_user
[hide private]
[frames] | no frames]

Source Code for Module commands.add_user

 1  from flask_script import Command, Option 
 2  from coprs import db 
 3  from coprs import models 
 4  from coprs.views.misc import create_user_wrapper 
 5   
 6   
7 -class AddUserCommand(Command):
8 9 """ 10 You should not use regularly as that user will not be related to FAS account. 11 This should be used only for testing or adding special accounts e.g. proxy user. 12 """ 13
14 - def run(self, name, mail, **kwargs):
15 user = models.User.query.filter(models.User.username == name).first() 16 if user: 17 print("User named {0} already exists.".format(name)) 18 return 19 20 user = create_user_wrapper(name, mail) 21 if kwargs["api_token"]: 22 user.api_token = kwargs["api_token"] 23 if kwargs["api_login"]: 24 user.api_token = kwargs["api_login"] 25 26 db.session.add(user) 27 db.session.commit()
28 29 option_list = ( 30 Option("name"), 31 Option("mail"), 32 Option("--api_token", default=None, required=False), 33 Option("--api_login", default=None, required=False), 34 )
35