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

Source Code for Module commands.alter_user

 1  from coprs import db 
 2  from coprs import models 
 3  from flask_script import Command, Option, Group 
 4   
 5   
6 -class AlterUserCommand(Command):
7
8 - def run(self, name, **kwargs):
9 user = models.User.query.filter( 10 models.User.username == name).first() 11 if not user: 12 print("No user named {0}.".format(name)) 13 return 14 15 if kwargs["admin"]: 16 user.admin = True 17 if kwargs["no_admin"]: 18 user.admin = False 19 if kwargs["proven"]: 20 user.proven = True 21 if kwargs["no_proven"]: 22 user.proven = False 23 if kwargs["proxy"]: 24 user.proxy = True 25 if kwargs["no_proxy"]: 26 user.proxy = False 27 28 db.session.add(user) 29 db.session.commit()
30 31 option_list = ( 32 Option("name"), 33 Group( 34 Option("--admin", 35 action="store_true"), 36 Option("--no-admin", 37 action="store_true"), 38 exclusive=True 39 ), 40 Group( 41 Option("--proven", 42 action="store_true"), 43 Option("--no-proven", 44 action="store_true"), 45 exclusive=True 46 ), 47 Group( 48 Option("--proxy", 49 action="store_true"), 50 Option("--no-proxy", 51 action="store_true"), 52 exclusive=True 53 ) 54 )
55