Add __repr__ to Activity and Generator class

1 files changed, 18 insertions, 0 deletions

diff --git a/mediagoblin/db/models.py b/mediagoblin/db/models.py
index 4d7830d2..ffd7b4f1 100644
--- a/mediagoblin/db/models.py
+++ b/mediagoblin/db/models.py
@@ -1070,6 +1070,12 @@ class Generator(Base):
     updated = Column(DateTime, default=datetime.datetime.now)
     object_type = Column(Unicode, nullable=False)
 
+    def __repr__(self):
+        return "<{klass} {name}>".format(
+            klass=self.__class__.__name__,
+            name=self.name
+        )
+
     def serialize(self, request):
         return {
             "id": self.id,
@@ -1168,6 +1174,18 @@ class Activity(Base, ActivityMixin):
         foreign_keys="Activity.actor", post_update=True)
     get_generator = relationship(Generator)
 
+    def __repr__(self):
+        if self.content is None:
+            return "<{klass} verb:{verb}>".format(
+                klass=self.__class__.__name__,
+                verb=self.verb
+            )
+        else:
+            return "<{klass} {content}>".format(
+                klass=self.__class__.__name__,
+                content=self.content
+            )
+
     @property
     def get_object(self):
         if self.object is None: