Customizing User model to add more details

The user model in Superset only contains the following fields:

username
firstName
lastName
userId
isActive
email

If I want to add more (let’s say country, location, etc.) how would I go about it? I looked at the UserAttribute class, but not quite sure how to use it.

Would I need to customize the schema for ab_user table? And if so, how can I do that?

[BigDataArtist]

Hi, can you please share the full code? we want to add a few more fields and we cannot do them into our DB. Is it possible to share the full code?

[jay-finn]

For anyone who runs into a similar issue, I resolved this by overriding the User class from flask_appbuilder.security.sqla.models.

from flask_appbuilder.security.sqla.models import User
from sqlalchemy import (
    Column,
    String
)
class CustomUser(User):
    __tablename__ = "ab_user"
    time_zone = Column(String(64))

This adds time_zone to the schema for ab_user. After this, override the add_user function from class SupersetSecurityManager. Note that if you’re implementing auth through an external OAuth provider, you’d already be doing this.

from superset.security import SupersetSecurityManager
class CustomSsoSecurityManager(SupersetSecurityManager):
  def add_user(
        self,
        username,
        first_name,
        last_name,
        email,
        role,
        password="",
        hashed_password="",
        time_zone=""
    ):