From 9c8ae36ba3431b4547ab9eed37301cc376c0c9e6 Mon Sep 17 00:00:00 2001
From: mike cullerton <michaelc@cullerton.com>
Date: Wed, 3 Nov 2021 12:40:43 -0400
Subject: [PATCH] UVA IRB passes back `Q_COMPLETE` and `HSRNUMBER`

---
 crc/models/protocol_builder.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/crc/models/protocol_builder.py b/crc/models/protocol_builder.py
index 831f80a0..692ba4ab 100644
--- a/crc/models/protocol_builder.py
+++ b/crc/models/protocol_builder.py
@@ -41,20 +41,21 @@ class ProtocolBuilderStatus(enum.Enum):
 class ProtocolBuilderStudy(object):
     def __init__(
             self, STUDYID: int, TITLE: str, NETBADGEID: str,
-            DATE_MODIFIED: str
+            DATE_MODIFIED: str, Q_COMPLETE: str=None, HSRNUMBER: str=None
     ):
         self.STUDYID = STUDYID
         self.TITLE = TITLE
         self.NETBADGEID = NETBADGEID
         self.DATE_MODIFIED = DATE_MODIFIED
-
+        self.Q_COMPLETE = Q_COMPLETE
+        self.HSRNUMBER = HSRNUMBER
 
 class ProtocolBuilderStudySchema(ma.Schema):
     class Meta:
         model = ProtocolBuilderStudy
         unknown = INCLUDE
         fields = ["STUDYID", "TITLE", "NETBADGEID",
-                  "DATE_MODIFIED"]
+                  "DATE_MODIFIED", "Q_COMPLETE", "HSRNUMBER"]
 
     @post_load
     def make_pbs(self, data, **kwargs):