mirror of
https://github.com/tencentmusic/cube-studio.git
synced 2024-12-21 06:19:31 +08:00
28 lines
807 B
Python
28 lines
807 B
Python
|
|
||
|
|
from flask_appbuilder import Model
|
||
|
|
from sqlalchemy import Column, ForeignKey, Integer
|
||
|
|
from sqlalchemy.orm import relationship
|
||
|
|
|
||
|
|
from myapp import security_manager
|
||
|
|
from myapp.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 Myapp. 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]
|
||
|
|
)
|
||
|
|
|