refactor(sv_quant_api): 优化 rebalance 判断逻辑

-重构了 rebalance 函数，提高了代码可读性和维护性
- 新增 check_nan 函数，用于检查数据是否为空
- 新增 do_re_balance 函数，用于执行再平衡操作
- 优化了测试用例生成和执行的流程

test_case_exec.txt

 生成测试用例：
-python test_case_gen\Sample\gentestcase.py test_case_gen\Sample\testcase_example.xlsx
+python test_case_gen\Sample\gentestcase.py test_case_gen\Sample\testcase_example_0526.xlsx
 
 
 执行测试用例：
@@ -11,3 +11,5 @@ python reb_test.py sv test TEST_CASE_4_portfolio.json TEST_CASE_4_recomm.json te
 python reb_test.py sv test TEST_CASE_5_portfolio.json TEST_CASE_5_recomm.json test_data\class.json test_data\record_portfolio.json 20230812
 python reb_test.py sv test TEST_CASE_6_portfolio.json TEST_CASE_6_recomm.json test_data\class.json test_data\record_portfolio.json 20230812
 python reb_test.py sv test TEST_CASE_7_portfolio.json TEST_CASE_7_recomm.json test_data\class.json test_data\record_portfolio.json 20230812
+python reb_test.py sv test TEST_CASE_8_portfolio.json TEST_CASE_8_recomm.json test_data\class.json test_data\record_portfolio.json 20230812
+python reb_test.py sv test TEST_CASE_10_portfolio.json TEST_CASE_10_recomm.json test_data\class.json test_data\record_portfolio.json 20230812

vender/sv_quant_api.py

 import calendar
 import json
+import math
 from datetime import datetime, timedelta
 
 from vender.quant_api_base import base_Quant_api
 
 
-class Quant_api(base_Quant_api):
-
-    def _rebalance(self, pt):
-
-        """
-        是否需要 rebalance
-        :param pt:
-        :return: 返回 model_portfolio 的 rp_id 表示需要 rebalance ，若是不需要 rebalance 返回 0
-        """
-
-        def last_busi_date(day_str: str):
+def last_busi_date(day_str: str):
     year = int(day_str[:4])
     month = int(day_str[4:6])
     _, last_day = calendar.monthrange(year, month)
@@ -25,17 +16,54 @@ class Quant_api(base_Quant_api):
         last_day_date -= timedelta(days=1)
     return last_day_date.strftime('%Y-%m-%d %H:%M:%S.%f')[:-3]
 
-        model_portfolio = self.get_latest_portfolio(pt['recomm_id'])
-        if not pt.get('rp_id') or pt.get('rp_id') != model_portfolio['rp_id']:
-            # 如果推薦生成日data_Date在最後一次觸發再平衡时间之后，视为需要再平衡
-            if (not pt.get('last_recomm_date') or type(pt.get('last_recomm_date')) != str or
-                    model_portfolio['data_Date'] > pt.get('last_recomm_date').replace('-', '')):
+
+def check_nan(data):
+    """
+    检查输入数据是否为空
+    :param data:
+    :return:
+    """
+    if data is None:
+        return True
+    elif type(data) == float and math.isnan(data):
+        return True
+    elif type(data) == str and (data == 'nan' or data.strip() == ''):
+        return True
+    return False
+
+
+def do_re_balance(model_portfolio):
     note = json.loads(model_portfolio.get('note'))
     if not note.get('recomm_reason'):
         raise RuntimeError('未輸入推薦理由')
     return model_portfolio.get('rp_id'), '月初調倉日', last_busi_date(
         model_portfolio['data_Date']), note.get(
         'recomm_reason')
+
+
+class Quant_api(base_Quant_api):
+
+    def _rebalance(self, pt):
+
+        """
+        是否需要 rebalance
+        :param pt:
+        :return: 返回 model_portfolio 的 rp_id 表示需要 rebalance ，若是不需要 rebalance 返回 0
+        """
+
+        model_portfolio = self.get_latest_portfolio(pt['recomm_id'])
+        # 用户没买过投组
+        if check_nan(pt.get('rp_id')):
+            return do_re_balance(model_portfolio)
+        else:
+            # 买过的投组不是最新投组
+            if str(pt.get('rp_id')) != str(model_portfolio.get('rp_id')):
+                if check_nan(pt.get('last_recomm_date')):
+                    return do_re_balance(model_portfolio)
+                else:
+                    # 最后检查日不为空且投组的日期大于最后检查日
+                    if model_portfolio['data_Date'] > pt.get('last_recomm_date').replace('-', ''):
+                        return do_re_balance(model_portfolio)
         return 0
 
     def _dividend(self, pt):