From f07efb0be326da5894880bf5a7b69cd56f03758a Mon Sep 17 00:00:00 2001
From: Gene Dan <genedan@gmail.com>
Date: Tue, 14 Jan 2025 17:26:37 -0600
Subject: [PATCH] FEAT: Add sample XYZ Auto BI rate index to sample database.

---
 faslr/samples/db/generate_sample_db.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/faslr/samples/db/generate_sample_db.py b/faslr/samples/db/generate_sample_db.py
index 5c074d0..be2cbb5 100644
--- a/faslr/samples/db/generate_sample_db.py
+++ b/faslr/samples/db/generate_sample_db.py
@@ -16,6 +16,8 @@
 
 from faslr.schema import (
     CountryTable,
+    IndexTable,
+    IndexValuesTable,
     LocationTable,
     StateTable,
     LOBTable,
@@ -24,6 +26,10 @@
     ProjectViewData
 )
 
+from faslr.utilities.sample import (
+    XYZ_RATE_INDEX
+)
+
 from sqlalchemy.orm import sessionmaker
 from uuid import uuid4
 
@@ -125,6 +131,26 @@
 
 session.add_all(obj_list)
 
+new_index = IndexTable(
+    description=XYZ_RATE_INDEX['Name'][0],
+    scope='Global'
+)
+
+session.add(new_index)
+session.flush()
+
+df_rate_changes = pd.DataFrame(data={'year': XYZ_RATE_INDEX['Origin'], 'change': XYZ_RATE_INDEX['Change']})
+df_rate_changes['index_id'] = new_index.index_id
+
+rate_change_list = df_rate_changes.to_dict('records')
+
+obj_list = []
+for record in rate_change_list:
+    data_obj = IndexValuesTable(**record)
+    obj_list.append(data_obj)
+
+session.add_all(obj_list)
+
 session.commit()