api_key_sb.py

from supabase import create_client, Client
import os
import logging # 导入logging模块
from fastapi import FastAPI, Request, HTTPException, Depends, status
from dotenv import load_dotenv
from datetime import timezone, timedelta, datetime

load_dotenv()

# Supabase 配置
SUPABASE_URL = os.getenv("SUPABASE_URL")
SUPABASE_KEY = os.getenv("SUPABASE_KEY")

# 配置日志
logger = logging.getLogger(__name__)

def get_supabase_client() -> Client:
    """Initializes and returns a Supabase client instance."""
    return create_client(SUPABASE_URL, SUPABASE_KEY)

supabase: Client = get_supabase_client()

# async def get_api_key_info(api_type_name: str, lookup_value: str):
#     """
#     根据 API 类型名称和查找值从 Supabase 获取 API 密钥信息。
#     """
#     try:
#         # 直接从 airs_model_api_keys_view_v1 视图中查询
#         response = supabase.from_('airs_model_api_keys_view_v1').select('api_key, api_key_id').eq('api_type', api_type_name).eq('model_name', lookup_value).order('api_key_created_at', desc=False).limit(1).execute()
        
#         if response.data:
#             api_key_info = response.data[0]
#             # 视图返回的 id 是 api_key_id，需要将其映射为 'id' 以保持与之前逻辑的兼容性
#             formatted_api_key_info = {
#                 'api_key': api_key_info.get('api_key'),
#                 'id': api_key_info.get('api_key_id')
#             }
#             # 在返回api_key_info之前，更新其ran_at字段
#             await update_api_key_ran_at(formatted_api_key_info.get('id'))
#             return formatted_api_key_info
#         return None
#     except Exception as e:
#             logger.error(f"Error fetching API key info from view: {e}")
#             return None

async def get_api_key_info(lookup_value: str):
    """
    根据 API 类型名称和查找值从 Supabase 获取 API 密钥信息。
    """
    try:
        # 直接从 airs_model_api_keys_view_v1 视图中查询
        response = supabase.from_('airs_model_api_keys_view_v1').select('api_key, api_key_id').eq('model_name', lookup_value).order('api_key_created_at', desc=False).limit(1).execute()
        
        if response.data:
            api_key_info = response.data[0]
            # 视图返回的 id 是 api_key_id，需要将其映射为 'id' 以保持与之前逻辑的兼容性
            formatted_api_key_info = {
                'api_key': api_key_info.get('api_key'),
                'id': api_key_info.get('api_key_id')
            }
            # 在返回api_key_info之前，更新其ran_at字段
            await update_api_key_ran_at(formatted_api_key_info.get('id'))
            return formatted_api_key_info
        return None
    except Exception as e:
            logger.error(f"Error fetching API key info from view: {e}")
            return None

async def update_api_key_ran_at(api_key_id: int, ran_at_time: datetime = None):
    """
    Updates the 'ran_at' field for a given API key in Supabase.
    If ran_at_time is None, it defaults to the current Beijing time.
    """
    try:
        if ran_at_time is None:
            # 创建北京时间时区对象（UTC+8）
            beijing_tz = timezone(timedelta(hours=8))
            now_beijing = datetime.now(beijing_tz)  # 带时区的datetime对象
            current_local_time = now_beijing.isoformat()
        else:
            current_local_time = ran_at_time.isoformat()

        logger.info(f"尝试更新 API 密钥 {api_key_id} 的 ran_at 为 {current_local_time}")
        response = supabase.table("airs_api_keys_v1").update({"ran_at": current_local_time}).eq("id", api_key_id).execute()
        logger.info(f'API 密钥 {api_key_id} 的 ran_at 更新成功为 {current_local_time}. ')

        # 验证更新是否成功
        verification_response = supabase.from_('airs_api_keys_v1').select('ran_at').eq('id', api_key_id).single().execute()
        verified_ran_at = verification_response.data.get('ran_at')
        logger.info(f"验证：API 密钥 {api_key_id} 的实际 ran_at 值为 {verified_ran_at}")

    except Exception as e:
        logger.error(f"更新API密钥运行时间失败！错误信息：{str(e)}")
        raise HTTPException(status_code=500, detail=f"更新API密钥运行时间失败！错误信息：{str(e)}")