def run(self):
email = prompt('Email')
# user = users.first(email=email)
user = User.query.filter_by(email=email).first()
if user:
_security_datastore = LocalProxy(lambda: current_app.extensions['security'].datastore)
admin_role = _security_datastore.find_role('admin')
_security_datastore.add_role_to_user(user, admin_role)
db.session.commit()
print '\nUser given admin role sucessfully'
return
print '\nNo user found'
def run(self):
email = prompt('Email')
# user = users.first(email=email)
user = User.query.filter_by(email=email).first()
if user:
_security_datastore = LocalProxy(
lambda: current_app.extensions['security'].datastore)
admin_role = _security_datastore.find_role('admin')
_security_datastore.add_role_to_user(user, admin_role)
db.session.commit()
print '\nUser given admin role sucessfully'
return
print '\nNo user found'
def create_user_with_role(username, rolename):
_datastore = LocalProxy(
lambda: current_app.extensions['security'].datastore)
user, role = _datastore._prepare_role_modify_args(username, rolename)
if not user:
user = create_test_user(email=username, password='******')
if not role:
role = _datastore.create_role(name=rolename)
_datastore.add_role_to_user(user, role)
return user
def create_user_with_role(username, rolename):
_datastore = LocalProxy(
lambda: current_app.extensions['security'].datastore)
user, role = _datastore._prepare_role_modify_args(username, rolename)
if not user:
user = create_test_user(email=username, password='******')
if not role:
role = _datastore.create_role(name=rolename)
_datastore.add_role_to_user(user, role)
return user
def user_registered_signal_handler(app, user, confirm_token):
"""Executed imediately after user registers a new account.
Adds a default 'user' role to the new user
:param app: Flask application object
:param user: Newly registered Flask-Login User object
:param confirm_token: Users confirm token
"""
_security_datastore = LocalProxy(lambda: app.extensions["security"].datastore)
default_role = _security_datastore.find_role("user")
_security_datastore.add_role_to_user(user, default_role)
db.session.commit()
Bundle.refresh_user_bundle(user, current_wednesday())
Bundle.refresh_user_bundle(user, next_wednesday())
Bundle.refresh_user_bundle(user, two_wednesdays())
def user_registered_signal_handler(app, user, confirm_token):
"""Executed imediately after user registers a new account.
Adds a default 'user' role to the new user
:param app: Flask application object
:param user: Newly registered Flask-Login User object
:param confirm_token: Users confirm token
"""
_security_datastore = LocalProxy(
lambda: app.extensions['security'].datastore)
default_role = _security_datastore.find_role('user')
_security_datastore.add_role_to_user(user, default_role)
db.session.commit()
Bundle.refresh_user_bundle(user, current_wednesday())
Bundle.refresh_user_bundle(user, next_wednesday())
Bundle.refresh_user_bundle(user, two_wednesdays())
