From 5ca8ee415e05291da1257fa749448148109c5892 Mon Sep 17 00:00:00 2001
From: Brian Moe <brian.moe@ligo.org>
Date: Wed, 9 Jan 2013 15:16:23 -0600
Subject: [PATCH] Added extended attributes for MBTA events

---
 gracedb/translator.py | 15 +++++++++++++++
 gracedb/views.py      |  2 +-
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/gracedb/translator.py b/gracedb/translator.py
index 76c5ef36b..0739dc04f 100644
--- a/gracedb/translator.py
+++ b/gracedb/translator.py
@@ -221,6 +221,21 @@ def handle_uploaded_data(event, datafilename,
         event.nevents = coinc_table[0].nevents
         event.likelihood = coinc_table[0].likelihood
 
+        # extended attributes
+        coinc_inspiral_table = glue.ligolw.table.getTablesByName(
+                            xmldoc,
+                            glue.ligolw.lsctables.CoincInspiralTable.tableName)
+        coinc_inspiral_table = coinc_inspiral_table[0]
+        event.ifos             = coinc_inspiral_table[0].ifos
+        event.end_time         = coinc_inspiral_table[0].end_time
+        event.end_time_ns      = coinc_inspiral_table[0].end_time_ns
+        event.mass             = coinc_inspiral_table[0].mass
+        event.mchirp           = coinc_inspiral_table[0].mchirp
+        #event.minimum_duration = coinc_inspiral_table[0].minimum_duration
+        event.snr              = coinc_inspiral_table[0].snr
+        event.false_alarm_rate = coinc_inspiral_table[0].false_alarm_rate
+        event.combined_far     = coinc_inspiral_table[0].combined_far
+
         # XXX xml_filename unused
         #xml_filename = os.path.join(output_dir, coinc_table_filename)
 
diff --git a/gracedb/views.py b/gracedb/views.py
index f8be89098..9515fc37e 100644
--- a/gracedb/views.py
+++ b/gracedb/views.py
@@ -205,7 +205,7 @@ def _createEventFromForm(request, form):
         group = Group.objects.filter(name=form.cleaned_data['group'])
         atype = form.cleaned_data['type']
         # Create Event
-        if atype in ['LM', 'HM']:
+        if atype in ['LM', 'HM', 'MBTA']:
             event = CoincInspiralEvent()
         elif atype == "CWB":
             event = MultiBurstEvent()
-- 
GitLab