up b11 params

.gitignore

@@ -11,6 +11,7 @@ db.sqlite3
 db.sqlite3-journal
 db.sqlite3-shm
 db.sqlite3-wal
-
+test-api.py
+test_api.py
 .DS_Store
 ._.DS_Store

app/api/v1/calculation/calcuation.py

@@ -13,9 +13,17 @@ from app.schemas.valuation import (
 from app.schemas.base import Success, SuccessExtra
 from app.utils.app_user_jwt import get_current_app_user_id
 from app.utils.calculation_engine.economic_value_b1.economic_value_b1 import EconomicValueB1Calculator
+from app.utils.calculation_engine.economic_value_b1.sub_formulas.basic_value_b11 import calculate_popularity_score, \
+    calculate_infringement_score, calculate_patent_usage_score
+from app.utils.calculation_engine.economic_value_b1.sub_formulas.traffic_factor_b12 import calculate_search_index_s1
+
+from app.models.esg import ESG
+from app.models.industry import Industry
+from app.models.policy import Policy
 
 app_valuations_router = APIRouter(tags=["用户端估值评估"])
 
+
 @app_valuations_router.post("/calculation", summary="计算估值")
 async def calculate_valuation(
         data: UserValuationCreate,
@@ -30,10 +38,28 @@ async def calculate_valuation(
         # 创建计算器实例
         calculator = EconomicValueB1Calculator()
 
-        # 从用户数据中提取计算所需的参数
+        # 根据行业查询 ESG 基准分（优先用行业名称匹配，如用的是行业代码就把 name 改成 code）
-        input_data = _extract_calculation_params(data)
+        esg_obj = await ESG.filter(name=data.industry).first()
+        esg_score = float(esg_obj.number) if esg_obj else 0.0
 
-        # 调用计算引擎进行估值计算
+        # 根据行业查询 行业修正系数与ROE
+        industry_obj = await Industry.filter(name=data.industry).first()
+        # roe_score = industry_obj.roe
+        fix_num_score = industry_obj.fix_num
+
+        # 根据行业查询 政策匹配度
+        policy_obj = await Policy.filter(name=data.industry).first()
+        policy_match_score = policy_obj.score
+        # 提取计算参数
+        input_data = _extract_calculation_params(data)
+        # ESG关联价值 ESG分 (0-10分)
+        input_data["esg_score"] = esg_score
+        # 行业修正系数I
+        input_data["industry_coefficient"] = fix_num_score
+        # 政策匹配度
+        input_data["policy_match_score"] = policy_match_score
+
+        # 调用计算引擎
         calculation_result = calculator.calculate_complete_economic_value_b1(input_data)
 
         # 创建估值评估记录
@@ -42,7 +68,7 @@ async def calculate_valuation(
             data=data
         )
 
-        # 将计算结果添加到返回数据中
+        # 组装返回
         result_dict = json.loads(result.model_dump_json())
         result_dict['calculation_result'] = calculation_result
 
@@ -66,15 +92,22 @@ def _extract_calculation_params(data: UserValuationCreate) -> Dict[str, Any]:
     # 财务价值所需数据 从近三年收益计算
     three_year_income = data.three_year_income or [0, 0, 0]
 
+    # 法律强度L相关参数
+    # 普及地域分值 默认 7分
+    popularity_score = calculate_popularity_score(data.application_coverage)
+    # TODO 专利剩余年限【专利证书】 需要使用第三方API
+    patent = data.patent_application_no
+    patent_score = 0
+    # 侵权分 默认 6 TODO  需要使用第三方API 无侵权记录(10分)，历史侵权已解决(6分)，现存纠纷(2分)
+    # infringement_score = calculate_infringement_score()
+    infringement_score = 0
 
-    # 法律强度L相关参数 - 需要用户额外提供评分
-    # 这里使用默认值，实际应用中需要用户填写
-    patent_score = 5.0  # 专利分 (0-10分)
-    popularity_score = 5.0  # 普及地域分 (0-10分) 
-    infringement_score = 5.0  # 侵权分 (0-10分)
 
     # 发展潜力D相关参数
-    esg_score = 5.0  # ESG分 (0-10分)
+    # 专利使用量 TODO  需要使用第三方API
+    patent_score = calculate_patent_usage_score(0)
+
+
     # 创新投入比 = (研发费用/营收) * 100
     try:
         rd_investment = float(data.rd_investment or 0)
@@ -83,38 +116,43 @@ def _extract_calculation_params(data: UserValuationCreate) -> Dict[str, Any]:
     except (ValueError, TypeError):
         innovation_ratio = 0.0
 
-    # 行业系数I - 系统配置值
+    #
-    target_industry_roe = 0.15  # 目标行业平均ROE
-    benchmark_industry_roe = 0.12  # 基准行业ROE
-    
     # 流量因子B12相关参数
-    # 近30天搜索指数S1 - 从社交媒体数据计算
+    # 近30天搜索指数S1 - 从社交媒体数据计算 TODO  需要使用第三方API
-    search_index_s1 = 4500.0  # 默认值，实际应从API获取
+    search_index_s1 = calculate_search_index_s1() # 默认值，实际应从API获取
-    industry_average_s2 = 5000.0  # 行业均值，系统配置
+    # 行业均值S2 TODO 系统内置 未找到相关内容
-    
+    industry_average_s2 = 0.0
-    # 社交媒体传播度S3 - 从线上账号信息计算
+    # 社交媒体传播度S3 - TODO  需要使用第三方API,click_count view_count 未找到对应参数
-    social_media_spread_s3 = 1.07  # 默认值，实际应从API获取
+    # likes: 点赞数(API获取)
+    # comments: 评论数(API获取)
+    # shares: 转发数(API获取)
+    # followers: 粉丝数
+    # click_count: 商品链接点击量(用户填写) sales_volume 使用 sales_volume 销售量
+    # view_count: 内容浏览量(用户填写) link_views
 
     # 政策乘数B13相关参数
     # 政策契合度评分P - 根据行业和资助情况计算
-    policy_compatibility_score = 9.1  # 默认值，实际应系统计算
+    # 实施阶段
+    implementation_stage = data.implementation_stage
+    # 资金支持
+    funding_support = data.funding_status
 
     return {
         # 基础价值B11相关参数
-        'financial_value': financial_value,
+        'three_year_income': three_year_income,
         'patent_score': patent_score,
         'popularity_score': popularity_score,
         'infringement_score': infringement_score,
-        'esg_score': esg_score,
         'innovation_ratio': innovation_ratio,
-        'target_industry_roe': target_industry_roe,
-        'benchmark_industry_roe': benchmark_industry_roe,
 
         # 流量因子B12相关参数
         'search_index_s1': search_index_s1,
-        'industry_average_s2': industry_average_s2,
+        # 'industry_average_s2': industry_average_s2,
-        'social_media_spread_s3': social_media_spread_s3,
+        # 'social_media_spread_s3': social_media_spread_s3,
+        'click_count': int(data.click_count) or 100,
+        'view_count': int(data.link_views) or 100,
 
         # 政策乘数B13相关参数
-        'policy_compatibility_score': policy_compatibility_score
+        'implementation_stage': implementation_stage,
+        'funding_support':funding_support
     }

app/utils/calculation_engine/economic_value_b1/economic_value_b1.py

@@ -8,7 +8,7 @@
 from typing import Dict
 try:
     # 相对导入（当作为包使用时）
-    from .sub_formulas.basic_value_b11 import BasicValueB11Calculator
+    from .sub_formulas.basic_value_b11 import BasicValueB11Calculator, calculate_popularity_score
     from .sub_formulas.traffic_factor_b12 import TrafficFactorB12Calculator
     from .sub_formulas.policy_multiplier_b13 import PolicyMultiplierB13Calculator
 except ImportError:
@@ -37,12 +37,12 @@ class EconomicValueB1Calculator:
         
         经济价值B1 = 基础价值B11 × (1 + 流量因子B12) × 政策乘数B13
         
-        Args:
+        args:
             basic_value_b11: 基础价值B11 (系统计算)
             traffic_factor_b12: 流量因子B12 (系统计算)
             policy_multiplier_b13: 政策乘数B13 (系统计算)
             
-        Returns:
+        returns:
             float: 经济价值B1
         """
         economic_value = basic_value_b11 * (1 + traffic_factor_b12) * policy_multiplier_b13
@@ -50,27 +50,27 @@ class EconomicValueB1Calculator:
         return economic_value
     
     def calculate_complete_economic_value_b1(self, input_data: Dict) -> Dict:
+        
         """
         计算完整的经济价值B1，包含所有子公式
         
-        Args:
+        args:
             input_data: 输入数据字典，包含所有必要的参数
             
-        Returns:
+        returns:
             Dict: 包含所有中间计算结果和最终结果的字典
         """
         # 财务价值F 近三年年均收益列表 [1,2,3]
-
         financial_value = self.basic_value_calculator.calculate_financial_value_f()
         # 计算法律强度L   patent_score: 专利分 (0-10分) (用户填写)
         #              popularity_score: 普及地域分 (0-10分) (用户填写)
         #              infringement_score: 侵权分 (0-10分) (用户填写)
-
+        popularity_score = calculate_popularity_score()
         legal_strength = self.basic_value_calculator.calculate_legal_strength_l()
+
         # 发展潜力      patent_score: 专利分 (0-10分) (用户填写)
         #             esg_score: ESG分 (0-10分) (用户填写)
         #             innovation_ratio: 创新投入比 (研发费用/营收) * 100 (用户填写)
-
         development_potential = self.basic_value_calculator.calculate_development_potential_d()
         # 计算行业系数I  target_industry_roe: 目标行业平均ROE (系统配置)
         #             benchmark_industry_roe: 基准行业ROE (系统配置)

app/utils/calculation_engine/economic_value_b1/sub_formulas/basic_value_b11.py

@@ -19,13 +19,13 @@ class BasicValueB11Calculator:
 
         基础价值B11 = 财务价值F × (0.45 + 0.05 × 行业系数I) + 法律强度L × (0.35 + 0.05 × 行业系数I) + 发展潜力D × 0.2
 
-        Args:
+        args:
             financial_value: 财务价值F (用户填写)
             legal_strength: 法律强度L (用户填写)
             development_potential: 发展潜力D (用户填写)
             industry_coefficient: 行业系数I (系统配置)
 
-        Returns:
+        returns:
             float: 基础价值B11
         """
         basic_value = (financial_value * (0.45 + 0.05 * industry_coefficient) +
@@ -51,11 +51,11 @@ class BasicValueB11Calculator:
 
         财务价值F = [3年内年均收益 × (1 + 增长率)^5] / 5
 
-        Args:
+        args:
             annual_revenue_3_years: 近三年年均收益列表，单位：万元 (用户填写)
             growth_rate: 增长率，如果为None则自动计算 (用户填写)
 
-        Returns:
+        returns:
             float: 财务价值F
         """
         if len(annual_revenue_3_years) != 3:
@@ -76,10 +76,10 @@ class BasicValueB11Calculator:
         """
         根据三年收益数据计算增长率
 
-        Args:
+        args:
             revenue_data: 三年收益数据 (用户填写)
 
-        Returns:
+        returns:
             float: 增长率
         """
         if len(revenue_data) != 3:
@@ -104,12 +104,12 @@ class BasicValueB11Calculator:
 
         法律强度L = 专利分 × 0.4 + 普及分 × 0.3 + 侵权分 × 0.3
 
-        Args:
+        args:
             patent_score: 专利分 (0-10分) (用户填写)
             popularity_score: 普及地域分 (0-10分) (用户填写)
             infringement_score: 侵权分 (0-10分) (用户填写)
 
-        Returns:
+        returns:
             float: 法律强度L
         """
 
@@ -129,12 +129,12 @@ class BasicValueB11Calculator:
 
         发展潜力D = 专利分 × 0.5 + ESG分 × 0.2 + 创新投入比 × 0.3
 
-        Args:
+        args:
             patent_score: 专利分 (0-10分) (用户填写)
             esg_score: ESG分 (0-10分) (用户填写)
             innovation_ratio: 创新投入比 (研发费用/营收) * 100 (用户填写)
 
-        Returns:
+        returns:
             float: 发展潜力D
         """
 
@@ -152,11 +152,11 @@ class BasicValueB11Calculator:
 
         行业系数I = (目标行业平均ROE - 基准行业ROE) / 基准行业ROE
 
-        Args:
+        args:
             target_industry_roe: 目标行业平均ROE (系统配置)
             benchmark_industry_roe: 基准行业ROE (系统配置)
 
-        Returns:
+        returns:
             float: 行业系数I
         """
         if benchmark_industry_roe == 0:
@@ -177,10 +177,10 @@ def calculate_patent_score(patent_remaining_years: int) -> float:
     - 5-10年: 7分
     - <5年: 3分
 
-    Args:
+    args:
         patent_remaining_years: 专利剩余保护期（年） (用户填写)
 
-    Returns:
+    returns:
         float: 专利分
     """
     if patent_remaining_years > 10:
@@ -201,10 +201,10 @@ def calculate_patent_usage_score(patent_count: int) -> float:
     - 每引用一项: +2.5分
     - 10分封顶
 
-    Args:
+    args:
         patent_count: 专利数量 (用户填写)
 
-    Returns:
+    returns:
         float: 专利使用量分
     """
     score = min(patent_count * 2.5, 10.0)
@@ -218,10 +218,10 @@ def calculate_popularity_score(region_coverage: str) -> float:
 
     全球覆盖(10分)，全国覆盖(7分)，区域覆盖(4分)
 
-    Args:
+    args:
         region_coverage: 普及地域类型 (用户填写)
 
-    Returns:
+    returns:
         float: 普及地域分
     """
     coverage_scores = {
@@ -230,7 +230,7 @@ def calculate_popularity_score(region_coverage: str) -> float:
         "区域覆盖": 4.0
     }
 
-    return coverage_scores.get(region_coverage, 0.0)
+    return coverage_scores.get(region_coverage, 7.0)
 
 
 # 侵权记录评分
@@ -240,10 +240,10 @@ def calculate_infringement_score(infringement_status: str) -> float:
 
     无侵权记录(10分)，历史侵权已解决(6分)，现存纠纷(2分)
 
-    Args:
+    args:
         infringement_status: 侵权记录状态 (用户填写)
 
-    Returns:
+    returns:
         float: 侵权记录分
     """
     infringement_scores = {
@@ -252,7 +252,7 @@ def calculate_infringement_score(infringement_status: str) -> float:
         "现存纠纷": 2.0
     }
 
-    return infringement_scores.get(infringement_status, 0.0)
+    return infringement_scores.get(infringement_status, 6.0)
 
 
 # 示例使用

app/utils/calculation_engine/economic_value_b1/sub_formulas/policy_multiplier_b13.py

@@ -15,7 +15,7 @@ class PolicyMultiplierB13Calculator:
         Args:
             policy_compatibility_score: 政策契合度评分P (系统计算)
             
-        Returns:
+        returns:
             float: 政策乘数B13
         """
         # 
@@ -38,7 +38,7 @@ class PolicyMultiplierB13Calculator:
             implementation_stage_score: 实施阶段评分 (用户填写)
             funding_support_score: 资金支持度 (用户填写)
             
-        Returns:
+        returns:
             float: 政策契合度评分P
         """
         # 
@@ -57,7 +57,7 @@ class PolicyMultiplierB13Calculator:
         Args:
             industry: 所属行业 (用户填写)
             
-        Returns:
+        returns:
             float: 政策匹配度
         """
 
@@ -72,7 +72,7 @@ class PolicyMultiplierB13Calculator:
         Args:
             implementation_stage: 实施阶段 (用户填写)
             
-        Returns:
+        returns:
             float: 实施阶段评分
         """
         stage_scores = {
@@ -81,7 +81,7 @@ class PolicyMultiplierB13Calculator:
             "试点阶段": 4.0
         }
 
-        return stage_scores.get(implementation_stage, 0.0)
+        return stage_scores.get(implementation_stage, 10.0)
 
     def calculate_funding_support_score(self, funding_support: str) -> float:
         """
@@ -92,7 +92,7 @@ class PolicyMultiplierB13Calculator:
         Args:
             funding_support: 资金支持情况 (用户填写)
             
-        Returns:
+        returns:
             float: 资金支持度
         """
         funding_scores = {

app/utils/calculation_engine/economic_value_b1/sub_formulas/traffic_factor_b12.py

@@ -19,12 +19,12 @@ class TrafficFactorB12Calculator:
 
         流量因子B12 = ln(近30天搜索指数S1/行业均值S2) × 0.3 + 社交媒体传播度S3 × 0.7
         
-        Args:
+        args:
             search_index_s1: 近30天搜索指数S1 (API获取)
             industry_average_s2: 行业均值S2 (系统配置)
             social_media_spread_s3: 社交媒体传播度S3 (API获取)
             
-        Returns:
+        returns:
             float: 流量因子B12
         """
         # 避免除零和对数计算错误
@@ -41,29 +41,7 @@ class TrafficFactorB12Calculator:
         
         return traffic_factor
 
-    def calculate_search_index_s1(self,
-                                 baidu_index: float,
-                                 wechat_index: float,
-                                 weibo_index: float) -> float:
-        """
-        计算近30天搜索指数S1
 
-        近30天搜索指数S1 = 百度搜索指数 × 0.4 + 微信搜索指数 × 0.3 + 微博搜索指数 × 0.3
-        
-        Args:
-            baidu_index: 百度搜索指数 (API获取)
-            wechat_index: 微信搜索指数 (API获取)
-            weibo_index: 微博搜索指数 (API获取)
-            
-        Returns:
-            float: 近30天搜索指数S1
-        """
-        # 
-        search_index = (baidu_index * 0.4 + 
-                       wechat_index * 0.3 + 
-                       weibo_index * 0.3)
-        
-        return search_index
     
     def calculate_social_media_spread_s3(self,
                                        interaction_index: float,
@@ -74,12 +52,12 @@ class TrafficFactorB12Calculator:
 
         社交媒体传播度S3 = 互动量指数 × 0.4 + 覆盖人群指数 × 0.3 + 转化效率 × 0.3
 
-        Args:
+        args:
             interaction_index: 互动量指数 (API获取)
             coverage_index: 覆盖人群指数 (API获取)
             conversion_efficiency: 转化效率 (用户填写)
 
-        Returns:
+        returns:
             float: 社交媒体传播度S3
         """
         #
@@ -98,12 +76,12 @@ class TrafficFactorB12Calculator:
 
         互动量指数 = (点赞 + 评论 + 转发) / 1000
         
-        Args:
+        args:
             likes: 点赞数 (API获取)
             comments: 评论数 (API获取)
             shares: 转发数 (API获取)
             
-        Returns:
+        returns:
             float: 互动量指数
         """
         # 
@@ -117,10 +95,10 @@ class TrafficFactorB12Calculator:
 
         覆盖人群指数 = 粉丝数 / 10000
         
-        Args:
+        args:
             followers: 粉丝数 (API获取)
             
-        Returns:
+        returns:
             float: 覆盖人群指数
         """
         # 
@@ -137,11 +115,11 @@ class TrafficFactorB12Calculator:
         ：
         转化效率 = 商品链接点击量 / 内容浏览量
         
-        Args:
+        args:
             click_count: 商品链接点击量 (用户填写)
             view_count: 内容浏览量 (用户填写)
             
-        Returns:
+        returns:
             float: 转化效率
         """
         if view_count == 0:
@@ -163,10 +141,10 @@ class PlatformDataProcessor:
         
         输入格式：B站：12345678\n抖音：22334455
         
-        Args:
+        args:
             platform_data: 平台账号数据字符串 (用户填写)
             
-        Returns:
+        returns:
             Dict[str, str]: 平台账号字典
         """
         accounts = {}
@@ -188,10 +166,10 @@ class PlatformDataProcessor:
         """
         计算多平台互动数据
         
-        Args:
+        args:
             platform_data: 平台数据字典，格式为 {平台名: {likes: int, comments: int, shares: int, followers: int}} (API获取)
             
-        Returns:
+        returns:
             Tuple[float, float]: (互动量指数, 覆盖人群指数)
         """
         total_interactions = 0
@@ -227,11 +205,11 @@ def calculate_heat_score(daily_views: float, favorites: int) -> float:
     - 低热度：近7日日均浏览量 < 100，且收藏数 < 20 → 0.2
     - 无数据：无任何浏览与互动数据 → 0.0
     
-    Args:
+    args:
         daily_views: 近7日日均浏览量 (API获取)
         favorites: 收藏数 (API获取)
         
-    Returns:
+    returns:
         float: 热度分
     """
     if daily_views > 1000 or favorites > 100:
@@ -243,7 +221,29 @@ def calculate_heat_score(daily_views: float, favorites: int) -> float:
     else:
         return 0.0
 
+# 30天搜索指数S1
+def calculate_search_index_s1(baidu_index: float,
+                              wechat_index: float,
+                              weibo_index: float) -> float:
+    """
+    计算近30天搜索指数S1
 
+    近30天搜索指数S1 = 百度搜索指数 × 0.4 + 微信搜索指数 × 0.3 + 微博搜索指数 × 0.3
+
+    args:
+        baidu_index: 百度搜索指数 (API获取)
+        wechat_index: 微信搜索指数 (API获取)
+        weibo_index: 微博搜索指数 (API获取)
+
+    returns:
+        float: 近30天搜索指数S1
+    """
+    #
+    search_index = (baidu_index * 0.4 +
+                    wechat_index * 0.3 +
+                    weibo_index * 0.3)
+
+    return search_index
 # 示例使用
 if __name__ == "__main__":
     # 创建计算器实例