diff --git a/lemur/certificates/models.py b/lemur/certificates/models.py index 9b44dda2..345d6836 100644 --- a/lemur/certificates/models.py +++ b/lemur/certificates/models.py @@ -159,6 +159,9 @@ class Certificate(db.Model): """ return "arn:aws:iam::{}:server-certificate/{}".format(account_number, self.name) + def __repr__(self): + return "Certificate(name={name})".format(name=self.name) + @event.listens_for(Certificate.destinations, 'append') def update_destinations(target, value, initiator): diff --git a/lemur/destinations/models.py b/lemur/destinations/models.py index 9c5dbc0e..3e04eae5 100644 --- a/lemur/destinations/models.py +++ b/lemur/destinations/models.py @@ -23,3 +23,6 @@ class Destination(db.Model): @property def plugin(self): return plugins.get(self.plugin_name) + + def __repr__(self): + return "Destination(label={label})".format(label=self.label) diff --git a/lemur/domains/models.py b/lemur/domains/models.py index ed7123fc..dae55bb7 100644 --- a/lemur/domains/models.py +++ b/lemur/domains/models.py @@ -17,3 +17,6 @@ class Domain(db.Model): id = Column(Integer, primary_key=True) name = Column(String(256)) sensitive = Column(Boolean, default=False) + + def __repr__(self): + return "Domain(name={name})".format(name=self.name) diff --git a/lemur/endpoints/models.py b/lemur/endpoints/models.py index 6cffcb70..7b16058c 100644 --- a/lemur/endpoints/models.py +++ b/lemur/endpoints/models.py @@ -81,3 +81,6 @@ class Endpoint(db.Model): issues.append({'name': 'revoked', 'value': 'There is a revoked certificate attached to this endpoint consider replacing it.'}) return issues + + def __repr__(self): + return "Endpoint(name={name})".format(name=self.name) diff --git a/lemur/notifications/models.py b/lemur/notifications/models.py index 97475686..c2cb4ffc 100644 --- a/lemur/notifications/models.py +++ b/lemur/notifications/models.py @@ -33,3 +33,6 @@ class Notification(db.Model): @property def plugin(self): return plugins.get(self.plugin_name) + + def __repr__(self): + return "Notification(label={label})".format(label=self.label) diff --git a/lemur/roles/models.py b/lemur/roles/models.py index adf7b70a..7b9731d9 100644 --- a/lemur/roles/models.py +++ b/lemur/roles/models.py @@ -29,3 +29,6 @@ class Role(db.Model): user_id = Column(Integer, ForeignKey('users.id')) users = relationship("User", secondary=roles_users, passive_deletes=True, backref="role") certificates = relationship("Certificate", secondary=roles_certificates, backref="role") + + def __repr__(self): + return "Role(name={name})".format(name=self.name) diff --git a/lemur/sources/models.py b/lemur/sources/models.py index f3145b5c..92f7eb08 100644 --- a/lemur/sources/models.py +++ b/lemur/sources/models.py @@ -27,3 +27,6 @@ class Source(db.Model): @property def plugin(self): return plugins.get(self.plugin_name) + + def __repr__(self): + return "Source(label={label})".format(label=self.label) diff --git a/lemur/users/models.py b/lemur/users/models.py index b513f1a9..62165cb5 100644 --- a/lemur/users/models.py +++ b/lemur/users/models.py @@ -81,5 +81,8 @@ class User(db.Model): if role.name == 'admin': return True + def __repr__(self): + return "User(username={username})".format(username=self.username) + listen(User, 'before_insert', hash_password)