From d294df4761d3ba226005afa4181108e37cc46cc5 Mon Sep 17 00:00:00 2001
From: Keiran Price <keiran.price@stfc.ac.uk>
Date: Thu, 24 Oct 2019 10:25:16 +0100
Subject: [PATCH] #105: Create serializer method

---
 common/models/db_models.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/common/models/db_models.py b/common/models/db_models.py
index 5d3fce23..223709c4 100644
--- a/common/models/db_models.py
+++ b/common/models/db_models.py
@@ -1,6 +1,8 @@
 import enum
 
 from datetime import datetime
+from decimal import Decimal
+
 from sqlalchemy import Index, Column, BigInteger, String, DateTime, ForeignKey, Integer, Float, FetchedValue, \
     TypeDecorator, Boolean
 from sqlalchemy.ext.declarative import declarative_base
@@ -53,6 +55,19 @@ def to_dict(self):
             dictionary[column.name] = str(attribute) if isinstance(attribute, datetime) else attribute
         return dictionary
 
+    def _make_serializable(self, field):
+        """
+        Given a field, convert to a JSON serializable type
+        :param field: The field to be converted
+        :return: The converted field
+        """
+        if isinstance(field, datetime):
+            return str(field)
+        elif isinstance(field, Decimal):
+            return float(field)
+        else:
+            return field
+
     def to_nested_dict(self, includes):
         """
         Given related models return a nested dictionary with the child or parent rows nested.