- Add update dao_update_annual_billing_for_current_and_future_years

- moved get_current_financial_year_start_year from service.utils to dao.date_utils
- Moved logic for data persistence from rest to dao when updating records in db

app/dao/annual_billing_dao.py

@@ -4,8 +4,24 @@ from app.dao.dao_utils import (
     version_class
 )
 from app.models import AnnualBilling
-from datetime import datetime
-from app.service.utils import get_current_financial_year_start_year
+from app.dao.date_util import get_current_financial_year_start_year
+
+
+@transactional
+def dao_create_or_update_annual_billing_for_year(service_id, free_sms_fragment_limit, financial_year_start=None):
+
+    if not financial_year_start:
+        financial_year_start = get_current_financial_year_start_year()
+
+    result = dao_get_free_sms_fragment_limit_for_year(service_id, financial_year_start)
+
+    if result:
+        result.free_sms_fragment_limit = free_sms_fragment_limit
+    else:
+        result = AnnualBilling(service_id=service_id, financial_year_start=financial_year_start,
+                               free_sms_fragment_limit=free_sms_fragment_limit)
+    db.session.add(result)
+    return result
 
 
 def dao_get_annual_billing(service_id):
@@ -14,16 +30,28 @@ def dao_get_annual_billing(service_id):
     ).order_by(AnnualBilling.financial_year_start).all()
 
 
-def dao_create_or_update_annual_billing_for_year(annual_billing):
-    db.session.add(annual_billing)
-    db.session.commit()
+@transactional
+def dao_update_annual_billing_for_current_and_future_years(service_id, free_sms_fragment_limit,
+                                                           financial_year_start=None):
+    if not financial_year_start:
+        financial_year_start = get_current_financial_year_start_year()
+
+    updated = AnnualBilling.query.filter(
+        AnnualBilling.service_id == service_id,
+        AnnualBilling.financial_year_start >= financial_year_start
+    ).update(
+        {'free_sms_fragment_limit': free_sms_fragment_limit}
+    )
 
 
-def dao_get_free_sms_fragment_limit_for_year(service_id, year):
+def dao_get_free_sms_fragment_limit_for_year(service_id, financial_year_start=None):
+
+    if not financial_year_start:
+        financial_year_start = get_current_financial_year_start_year()
 
     return AnnualBilling.query.filter_by(
         service_id=service_id,
-        financial_year_start=year
+        financial_year_start=financial_year_start
     ).first()