Fix for type abbrev. Closes #304 (#307)

unitedstates · Apr 14, 2021 · 2d6194f · 2d6194f
1 parent 3d78df2
commit 2d6194f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/server_py/flatgov/bills/models.py b/server_py/flatgov/bills/models.py
@@ -7,6 +7,8 @@
 from iteration_utilities import flatten
 from django.utils.translation import gettext_lazy as _
 
+from bills.templatetags.bill_filters import stagesFormat
+
 MAX_RELATED_BILLS = 30
 
 class Bill(models.Model):
@@ -35,7 +37,10 @@ def __str__(self):
         return self.bill_congress_type_number
 
     def get_type_abbrev(self) -> str:
-        return ''.join([letter.upper() + '.' for letter in self.type])
+        if self.type:
+            return stagesFormat.get(self.type.upper(), '')
+        else:
+            return ''
 
     def get_related_bill_numbers(self):
         return self.related_dict.keys()