From 12b7e53262058efa76246dab8df1d7362ccc8aa5 Mon Sep 17 00:00:00 2001
From: Tanner Prestegard <tanner.prestegard@ligo.org>
Date: Fri, 22 Jun 2018 09:31:59 -0500
Subject: [PATCH] Adding a __unicode__ display for labelling model

---
 gracedb/superevents/models.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/gracedb/superevents/models.py b/gracedb/superevents/models.py
index 1d814f8e4..539a0f851 100644
--- a/gracedb/superevents/models.py
+++ b/gracedb/superevents/models.py
@@ -304,6 +304,8 @@ class Superevent(CleanSaveModel, ModelToDictMixin, AutoIncrementModel):
             letter_suffix
 
     # Custom methods ----------------------------------------------------------
+    # TODO: may want to add select or prefetch here to speed up access to group
+    # names, since that is used in calculating the graceid as well
     def get_external_events(self):
         """Returns a queryset of external events"""
         return self.events.filter(group__name=settings.EXTERNAL_ANALYSIS_GROUP)
@@ -372,6 +374,10 @@ class Labelling(m2mThroughBase):
         related_name='%(app_label)s_%(class)s_set',
         on_delete=models.CASCADE)
 
+    def __unicode__(self):
+        return "{superevent_id} | {label}".format(superevent_id=
+            self.superevent.superevent_id, label=self.label.name)
+
 
 class Signoff(CleanSaveModel, SignoffBase):
     """Class for superevent signoffs"""
-- 
GitLab