Files
superset2/superset/models/user_attributes.py
Beto Dealmeida 222b79df7e Template dashboard (#5550)
* Template dashboard

* Fix MySQL test

* Model for user attributes

* Redirect to welcome dash

* Fix lint

* Add missing file

* Add migration script

* Fix lint

* Fix more lint
2018-08-07 17:39:15 -07:00

37 lines
1.1 KiB
Python

# -*- coding: utf-8 -*-
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
from flask_appbuilder import Model
from sqlalchemy import Column, ForeignKey, Integer
from sqlalchemy.orm import relationship
from superset import security_manager
from superset.models.helpers import AuditMixinNullable
class UserAttribute(Model, AuditMixinNullable):
"""
Custom attributes attached to the user.
Extending the user attribute is tricky due to its dependency on the
authentication typew an circular dependencies in Superset. Instead, we use
a custom model for adding attributes.
"""
__tablename__ = 'user_attribute'
id = Column(Integer, primary_key=True) # pylint: disable=invalid-name
user_id = Column(Integer, ForeignKey('ab_user.id'))
user = relationship(
security_manager.user_model,
backref='extra_attributes',
foreign_keys=[user_id],
)
welcome_dashboard_id = Column(Integer, ForeignKey('dashboards.id'))
welcome_dashboard = relationship('Dashboard')