仓库模块完了
This commit is contained in:
@@ -8,7 +8,7 @@ import logging
|
||||
from sqlalchemy.orm import Session
|
||||
from sqlalchemy import and_, or_
|
||||
|
||||
from app.models.models import User, Algorithm, APIKey
|
||||
from app.models.models import User, Algorithm
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
@@ -209,26 +209,7 @@ class PermissionManager:
|
||||
logger.error(f"Error getting algorithm permissions: {str(e)}")
|
||||
return []
|
||||
|
||||
def check_api_key_access(self, db: Session, api_key_value: str, algorithm_id: str) -> bool:
|
||||
"""检查API密钥对算法的访问权限"""
|
||||
try:
|
||||
# 通过API密钥查找用户
|
||||
api_key = db.query(APIKey).filter(
|
||||
APIKey.key == api_key_value,
|
||||
APIKey.status == "active"
|
||||
).first()
|
||||
|
||||
if not api_key:
|
||||
return False
|
||||
|
||||
# 检查用户对算法的访问权限
|
||||
return self.check_algorithm_access(
|
||||
db, api_key.user_id, algorithm_id, PermissionType.EXECUTE
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error checking API key access: {str(e)}")
|
||||
return False
|
||||
|
||||
|
||||
def validate_user_algorithm_operation(self, db: Session, user_id: str, algorithm_id: str,
|
||||
operation: str) -> bool:
|
||||
|
||||
Reference in New Issue
Block a user