|
@@ -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)
|
|
|
|
|
|