基金分析工具

# 阿里云 FC 函数（Python 3.9）—— 数据托管服务
# 功能：GET /read 公开读，POST /write 校验 Token 后写入 OSS
# 依赖：oss2（FC Python 环境默认已内置）
import json, os
import oss2

# ====== 配置区（必填） ======
OSS_ACCESS_KEY_ID = os.environ.get('OSS_AK_ID', '你的阿里云 AccessKey ID')
OSS_ACCESS_KEY_SECRET = os.environ.get('OSS_AK_SECRET', '你的阿里云 AccessKey Secret')
OSS_ENDPOINT = os.environ.get('OSS_ENDPOINT', 'oss-cn-hangzhou.aliyuncs.com')
OSS_BUCKET = os.environ.get('OSS_BUCKET', '你的 OSS 桶名')
OSS_OBJECT_KEY = 'fund_data.json'   # OSS 里存储数据的文件名
SHARED_TOKEN = os.environ.get('SHARED_TOKEN', '自己设一串复杂密码')  # 与前端填写的 Token 一致
# ==========================

def _get_bucket():
    auth = oss2.Auth(OSS_ACCESS_KEY_ID, OSS_ACCESS_KEY_SECRET)
    return oss2.Bucket(auth, OSS_ENDPOINT, OSS_BUCKET)

def _cors_headers():
    return {
        'Access-Control-Allow-Origin': '*',
        'Access-Control-Allow-Methods': 'GET, POST, OPTIONS',
        'Access-Control-Allow-Headers': 'Content-Type, X-Shared-Token',
        'Content-Type': 'application/json; charset=utf-8'
    }

def handler(event, context):
    evt = json.loads(event)
    path = evt.get('path', '/')
    method = (evt.get('method') or evt.get('httpMethod', 'GET')).upper()
    headers = evt.get('headers', {}) or {}

    # 浏览器预检请求
    if method == 'OPTIONS':
        return {'statusCode': 204, 'headers': _cors_headers(), 'body': ''}

    bucket = _get_bucket()

    # === 公开读：任何人都能 GET，用于分享链接 ===
    if method == 'GET' and path.endswith('/read'):
        try:
            data = bucket.get_object(OSS_OBJECT_KEY).read().decode('utf-8')
            return {'statusCode': 200, 'headers': _cors_headers(), 'body': data}
        except oss2.exceptions.NoSuchKey:
            return {'statusCode': 200, 'headers': _cors_headers(),
                    'body': json.dumps({'funds': [], 'savedAt': None})}
        except Exception as e:
            return {'statusCode': 500, 'headers': _cors_headers(),
                    'body': json.dumps({'error': str(e)})}

    # === 鉴权写：必须带正确的 X-Shared-Token ===
    if method == 'POST' and path.endswith('/write'):
        if headers.get('X-Shared-Token') != SHARED_TOKEN:
            return {'statusCode': 401, 'headers': _cors_headers(),
                    'body': json.dumps({'error': 'Token 错误'})}
        try:
            body = evt.get('body', '{}')
            if evt.get('isBase64Encoded'):
                body = __import__('base64').b64decode(body).decode('utf-8')
            bucket.put_object(OSS_OBJECT_KEY, body.encode('utf-8'))
            return {'statusCode': 200, 'headers': _cors_headers(),
                    'body': json.dumps({'ok': True})}
        except Exception as e:
            return {'statusCode': 500, 'headers': _cors_headers(),
                    'body': json.dumps({'error': str(e)})}

    return {'statusCode': 404, 'headers': _cors_headers(),
            'body': json.dumps({'error': 'Not Found'})}