Flask Framework Cookbook - Second Edition

How to do it...

To add basic user roles to the application, perform the following steps:

1. First, add a field named roles to the User model in auth/models.py, as follows:

class User(db.Model): 
    id = db.Column(db.Integer, primary_key=True) 
    username = db.Column(db.String(60)) 
    pwdhash = db.Column(db.String()) 
    admin = db.Column(db.Boolean()) 
    notes = db.Column(db.UnicodeText) 
    roles = db.Column(db.String(4)) 
 
    def __init__(self, username, password, admin=False, notes='',        roles='R'): 
        self.username = username 
        self.pwdhash = generate_password_hash(password) 
        self.admin = admin 
        self.notes = notes 
        self.roles = self.admin and roles or '' 

In order to add a new field, you might need to run migration scripts. You can refer to the Migrating databases using ...