FEAT: Add sample XYZ Auto BI rate index to sample database.

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()