From 23ead0de10a03fa9d2f87d0b07101787da36f02e Mon Sep 17 00:00:00 2001
From: Branson Stephens <branson.stephens@ligo.org>
Date: Thu, 25 Sep 2014 15:13:19 -0500
Subject: [PATCH] Fixed getByGraceid and graceid for the new distinguishing
 fields.

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

diff --git a/gracedb/models.py b/gracedb/models.py
index d0986a5b9..1d4b70faa 100644
--- a/gracedb/models.py
+++ b/gracedb/models.py
@@ -123,11 +123,11 @@ class Event(models.Model):
         ordering = ["-id"]
 
     def graceid(self):
-        if self.group.name == "Test":
+        if self.group.name == "Test" or self.search.name == "Test":
             return "T%04d" % self.id
-        elif self.analysisType == "HWINJ":
+        elif self.pipeline == "HardwareInjection":
             return "H%04d" % self.id
-        elif self.analysisType == "GRB":
+        elif self.group.name == "External":
             return "E%04d" % self.id
         return "G%04d" % self.id
 
@@ -194,11 +194,11 @@ class Event(models.Model):
             e = cls.objects.filter(id=int(id[1:])).select_subclasses()[0]
         except IndexError:
             raise cls.DoesNotExist("Event matching query does not exist")
-        if (id[0] == "T") and (e.group.name == "Test"):
+        if (id[0] == "T") and (e.group.name == "Test" or e.search.name == "Test"):
             return e
-        if (id[0] == "H") and (e.analysisType == "HWINJ"):
+        if (id[0] == "H") and (e.pipeline.name == "HardwareInjection"):
             return e
-        if (id[0] == "E") and (e.analysisType == "GRB"):
+        if (id[0] == "E") and (e.group.name == "External"):
             return e
         if (id[0] == "G"):
             return e
-- 
GitLab