Skip to content
Snippets Groups Projects
Commit 01450470 authored by Tanner Prestegard's avatar Tanner Prestegard Committed by GraceDB
Browse files

Add a few useful properties to the Superevent model

far: preferred event's FAR
default_superevent_id: S-type ID
gw_id: None if not confirmed as GW, GW ID otherwise
superevent_id is still gw_id if confirmed as GW, otherwise
default_superevent_id.
parent f30f34fa
No related branches found
No related tags found
No related merge requests found
......@@ -364,18 +364,35 @@ class Superevent(CleanSaveModel, AutoIncrementModel):
@property
def superevent_id(self):
if self.is_gw:
id_prefix = self.GW_ID_PREFIX
letter_suffix = self.gw_letter_suffix
return self.gw_id
else:
id_prefix = self.DEFAULT_ID_PREFIX
letter_suffix = self.base_letter_suffix
return self.default_superevent_id
@property
def default_superevent_id(self):
id_prefix = self.DEFAULT_ID_PREFIX
letter_suffix = self.base_letter_suffix
# Prepend category prefix (if not production)
pre_prefix = ""
if self.category != self.SUPEREVENT_CATEGORY_PRODUCTION:
pre_prefix = self.category
return pre_prefix + self.DEFAULT_ID_PREFIX + \
self.t_0_date.strftime(self.DATE_STR_FMT) + self.base_letter_suffix
@property
def gw_id(self):
if not self.is_gw:
return None
# Prepend category prefix (if not production)
pre_prefix = ""
if self.category != self.SUPEREVENT_CATEGORY_PRODUCTION:
id_prefix = self.category + id_prefix
pre_prefix = self.category
return id_prefix + self.t_0_date.strftime(self.DATE_STR_FMT) + \
letter_suffix
return pre_prefix + self.GW_ID_PREFIX + \
self.t_0_date.strftime(self.DATE_STR_FMT) + self.gw_letter_suffix
@property
def graceid(self):
......@@ -387,6 +404,11 @@ class Superevent(CleanSaveModel, AutoIncrementModel):
"""Alias for t_0"""
return self.t_0
@property
def far(self):
"""Alias the FAR from the preferred_event"""
return self.preferred_event.far
# Custom methods ----------------------------------------------------------
def get_external_events(self, related_fields=['group']):
"""Returns a queryset of external events"""
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment