Forráskód Böngészése

feat(decode): update default quota limit to MAX_QUOTA_LIMIT

- Changed the default quota limit from 100 to a new constant MAX_QUOTA_LIMIT (500) in the quota management logic.
- Updated the logic to ensure consistency in quota limit retrieval and record creation.
jihuaqiang 1 hónapja
szülő
commit
106d4947cd
1 módosított fájl, 6 hozzáadás és 4 törlés
  1. 6 4
      tasks/decode.py

+ 6 - 4
tasks/decode.py

@@ -9,6 +9,8 @@ from typing import Dict, Any, Optional
 import sys
 import sys
 import requests
 import requests
 
 
+MAX_QUOTA_LIMIT = 500
+
 logger.add(sink=sys.stderr, level="ERROR", backtrace=True, diagnose=True)
 logger.add(sink=sys.stderr, level="ERROR", backtrace=True, diagnose=True)
 
 
 # 错误码常量
 # 错误码常量
@@ -91,18 +93,18 @@ def _check_quota(scene: SceneEnum, capability: CapabilityEnum = CapabilityEnum.D
                     cursor.execute(select_sql, (quota_date, scene_value, capability_value, content_type_value))
                     cursor.execute(select_sql, (quota_date, scene_value, capability_value, content_type_value))
                     quota_record = cursor.fetchone()
                     quota_record = cursor.fetchone()
                     
                     
-                    # 2. 如果没找到,创建一条新记录,quota_limit 默认为 10
+                    # 2. 如果没找到,创建一条新记录,quota_limit 默认为 MAX_QUOTA_LIMIT
                     if not quota_record:
                     if not quota_record:
                         insert_sql = """
                         insert_sql = """
                             INSERT INTO workflow_daily_quota (scene, capability, content_type, quota_date, quota_limit, used_count, locked)
                             INSERT INTO workflow_daily_quota (scene, capability, content_type, quota_date, quota_limit, used_count, locked)
                             VALUES (%s, %s, %s, %s, %s, %s, %s)
                             VALUES (%s, %s, %s, %s, %s, %s, %s)
                         """
                         """
-                        cursor.execute(insert_sql, (scene_value, capability_value, content_type_value, quota_date, 100, 0, 0))
-                        quota_limit = 100
+                        cursor.execute(insert_sql, (scene_value, capability_value, content_type_value, quota_date, MAX_QUOTA_LIMIT, 0, 0))
+                        quota_limit = MAX_QUOTA_LIMIT
                         current_used_count = 0
                         current_used_count = 0
                         is_locked = 0
                         is_locked = 0
                     else:
                     else:
-                        quota_limit = quota_record.get('quota_limit', 100)
+                        quota_limit = quota_record.get('quota_limit', MAX_QUOTA_LIMIT)
                         current_used_count = quota_record.get('used_count', 0)
                         current_used_count = quota_record.get('used_count', 0)
                         is_locked = quota_record.get('locked', 0)
                         is_locked = quota_record.get('locked', 0)