Add a __repr__ method to the Profile method

This commit is contained in:
Gergely Polonkai 2018-07-09 09:55:55 +02:00
parent acb7566d1b
commit 2136546390

View File

@ -165,7 +165,9 @@ class Profile(db.Model): # pylint: disable=too-few-public-methods
id = db.Column(db.Integer(), primary_key=True)
#: The ID of the local user this profile belongs to
user_id = db.Column(db.Integer(), db.ForeignKey('users.id'), nullable=True)
user_id = db.Column(db.Integer(), db.ForeignKey('users.id'), nullable=True, unique=True)
user = db.relationship(User, backref=db.backref('profile', lazy='dynamic'))
#: The username this profile belongs to. If ``None``, `user_id` must be set
username = db.Column(db.String(length=50), nullable=True)
@ -176,6 +178,16 @@ class Profile(db.Model): # pylint: disable=too-few-public-methods
#: The display name
display_name = db.Column(db.Unicode(length=80), nullable=False)
def __repr__(self):
if self.user:
username = self.user.username
domain = ''
else:
username = self.username
domain = '@' + self.domain
return f'<Profile {self.id}(@{username}{domain})>'
class Event(db.Model):
"""Database model for events