获取预估销售数据改为正则

ai/utils/show_process.py

@@ -2,6 +2,7 @@ from ai.models import *
 from ai.utils.show_prompt import *
 import datetime
 from django_redis import get_redis_connection
+import re
 
 # ai排片主流程
 def show_main_process():
@@ -31,8 +32,8 @@ def show_main_process():
         _show = _show.strip()
         print('_show:', _show)
         # 获取销售额
-        _sales = next((s for s in result.split('\n') if s.startswith('预估销售数据')), '').replace('预估销售数据：', '')
-        _sales = _sales.replace('约', '').replace('元', '')
+        _sales = next((s for s in result.split('\n') if re.search('预估销售数据', s)), '')
+        _sales = str(re.findall(r'\d+\.?\d*', _sales)[0])
         print('_sales:', _sales)
         # 处理返回结果
         try: