#!/usr/bin/env python3
"""
涨停分析 API（静态文件版本）
从 data/zt/ 目录读取预生成的 JSON 文件，不依赖 Elasticsearch

启动方式：
    python zt_api_static.py

端口：8800（与原 report_zt_api.py 相同，可替换使用）
"""

import os
import json
from flask import Flask, request, jsonify, send_from_directory
from flask_cors import CORS
from datetime import datetime
import logging

app = Flask(__name__)
CORS(app)

# 配置
DATA_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'data', 'zt')

# 日志
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# 内存缓存
_dates_cache = None
_stocks_cache = None


def load_dates():
    """加载日期索引"""
    global _dates_cache
    if _dates_cache is None:
        dates_file = os.path.join(DATA_DIR, 'dates.json')
        if os.path.exists(dates_file):
            with open(dates_file, 'r', encoding='utf-8') as f:
                _dates_cache = json.load(f)
        else:
            _dates_cache = {'dates': [], 'total': 0}
    return _dates_cache


def load_daily_analysis(date):
    """加载指定日期的分析数据"""
    daily_file = os.path.join(DATA_DIR, 'daily', f'{date}.json')
    if os.path.exists(daily_file):
        with open(daily_file, 'r', encoding='utf-8') as f:
            return json.load(f)
    return None


def load_stocks_for_search():
    """加载股票数据用于搜索"""
    global _stocks_cache
    if _stocks_cache is None:
        stocks_file = os.path.join(DATA_DIR, 'stocks.jsonl')
        if os.path.exists(stocks_file):
            _stocks_cache = []
            with open(stocks_file, 'r', encoding='utf-8') as f:
                for line in f:
                    if line.strip():
                        _stocks_cache.append(json.loads(line))
            logger.info(f"已加载 {len(_stocks_cache)} 条股票记录用于搜索")
        else:
            _stocks_cache = []
    return _stocks_cache


# ==================== API 路由 ====================

@app.route('/api/v1/dates/available', methods=['GET'])
def get_available_dates():
    """获取所有可用日期"""
    try:
        data = load_dates()

        # 转换为日历事件格式
        events = []
        for d in data.get('dates', []):
            events.append({
                'title': f"{d['count']}只",
                'start': d['formatted_date'],
                'end': d['formatted_date'],
                'className': 'bg-gradient-primary',
                'allDay': True,
                'date': d['date'],
                'count': d['count']
            })

        return jsonify({
            'success': True,
            'events': events,
            'total': len(events)
        })

    except Exception as e:
        logger.error(f"获取日期列表失败: {e}")
        return jsonify({'success': False, 'error': str(e)}), 500


@app.route('/api/v1/analysis/daily/<date>', methods=['GET'])
def get_daily_analysis(date):
    """获取指定日期的分析数据"""
    try:
        data = load_daily_analysis(date)

        if data is None:
            return jsonify({
                'success': False,
                'error': f'日期 {date} 的数据不存在'
            }), 404

        # 返回数据（与原接口格式兼容）
        return jsonify({
            'success': True,
            'data': {
                'date': data['date'],
                'formatted_date': data['formatted_date'],
                'total_stocks': data['total_stocks'],
                'sector_data': data['sector_data'],
                'chart_data': data['chart_data'],
                'word_freq_data': data['word_freq_data'],
                'sector_relations_top10': data['sector_relations_top10']
            },
            'from_cache': True,
            'cache_source': 'static_file'
        })

    except Exception as e:
        logger.error(f"获取日期 {date} 分析数据失败: {e}")
        return jsonify({'success': False, 'error': str(e)}), 500


@app.route('/api/v1/stocks/batch-detail', methods=['POST'])
def get_stocks_batch_detail():
    """批量获取股票详情"""
    try:
        data = request.json
        stock_codes = data.get('codes', [])
        date = data.get('date')

        if not stock_codes or not date:
            return jsonify({'success': False, 'error': '缺少参数'}), 400

        # 从日分析数据中获取股票详情
        daily_data = load_daily_analysis(date)
        if not daily_data:
            return jsonify({'success': False, 'error': f'日期 {date} 数据不存在'}), 404

        # 过滤指定股票
        stocks = [s for s in daily_data.get('stocks', []) if s.get('scode') in stock_codes]

        return jsonify({
            'success': True,
            'data': stocks
        })

    except Exception as e:
        logger.error(f"批量获取股票详情失败: {e}")
        return jsonify({'success': False, 'error': str(e)}), 500


@app.route('/api/v1/stocks/search/hybrid', methods=['POST'])
def search_stocks():
    """
    关键词搜索股票
    支持搜索：股票代码、股票名称、涨停原因（brief）、板块
    """
    try:
        data = request.json
        query = data.get('query', '').strip().lower()
        date = data.get('date')
        date_range = data.get('date_range', {})
        page = data.get('page', 1)
        page_size = data.get('page_size', 20)

        if not query:
            return jsonify({'success': False, 'error': '搜索关键词不能为空'}), 400

        # 加载搜索数据
        all_stocks = load_stocks_for_search()

        # 过滤
        results = []
        for stock in all_stocks:
            # 日期过滤
            stock_date = stock.get('date', '')
            if date and stock_date != date:
                continue
            if date_range:
                if date_range.get('start') and stock_date < date_range['start']:
                    continue
                if date_range.get('end') and stock_date > date_range['end']:
                    continue

            # 关键词匹配
            match_score = 0

            # 精确匹配股票代码（最高优先级）
            if query == stock.get('scode', '').lower():
                match_score = 100

            # 精确匹配股票名称
            elif query == stock.get('sname', '').lower():
                match_score = 90

            # 部分匹配股票名称
            elif query in stock.get('sname', '').lower():
                match_score = 80

            # 匹配板块
            elif any(query in sector.lower() for sector in stock.get('core_sectors', [])):
                match_score = 70

            # 匹配涨停原因
            elif query in stock.get('brief', '').lower():
                match_score = 60

            if match_score > 0:
                stock_copy = stock.copy()
                stock_copy['_score'] = match_score
                results.append(stock_copy)

        # 按分数和日期排序
        results.sort(key=lambda x: (-x['_score'], -int(x.get('date', '0'))))

        # 分页
        total = len(results)
        start = (page - 1) * page_size
        end = start + page_size
        page_results = results[start:end]

        return jsonify({
            'success': True,
            'data': {
                'stocks': page_results,
                'total': total,
                'page': page,
                'page_size': page_size,
                'total_pages': (total + page_size - 1) // page_size,
                'search_mode': 'keyword'
            }
        })

    except Exception as e:
        logger.error(f"搜索失败: {e}")
        return jsonify({'success': False, 'error': str(e)}), 500


@app.route('/api/v1/init/data', methods=['GET'])
def init_data():
    """初始化数据（获取最新日期的分析数据）"""
    try:
        dates_data = load_dates()
        dates = dates_data.get('dates', [])

        if not dates:
            return jsonify({'success': False, 'error': '无可用数据'}), 404

        latest_date = dates[0]['date']
        analysis = load_daily_analysis(latest_date)

        if not analysis:
            return jsonify({'success': False, 'error': '数据加载失败'}), 500

        # 转换日期为日历事件格式
        events = [{
            'title': f"{d['count']}只",
            'start': d['formatted_date'],
            'end': d['formatted_date'],
            'className': 'bg-gradient-primary',
            'allDay': True,
            'date': d['date'],
            'count': d['count']
        } for d in dates]

        return jsonify({
            'success': True,
            'data': {
                'latest_date': latest_date,
                'formatted_date': analysis['formatted_date'],
                'analysis': {
                    'date': analysis['date'],
                    'formatted_date': analysis['formatted_date'],
                    'total_stocks': analysis['total_stocks'],
                    'sector_data': analysis['sector_data'],
                    'chart_data': analysis['chart_data'],
                    'word_freq_data': analysis['word_freq_data'],
                    'sector_relations_top10': analysis['sector_relations_top10']
                },
                'available_dates': events
            }
        })

    except Exception as e:
        logger.error(f"初始化数据失败: {e}")
        return jsonify({'success': False, 'error': str(e)}), 500


@app.route('/api/v1/health', methods=['GET'])
def health_check():
    """健康检查"""
    dates_data = load_dates()
    return jsonify({
        'success': True,
        'status': 'healthy',
        'mode': 'static_file',
        'data_dir': DATA_DIR,
        'total_dates': dates_data.get('total', 0),
        'updated_at': dates_data.get('updated_at', 'unknown')
    })


@app.route('/api/v1/cache/reload', methods=['POST'])
def reload_cache():
    """重新加载缓存"""
    global _dates_cache, _stocks_cache
    _dates_cache = None
    _stocks_cache = None

    # 重新加载
    load_dates()
    load_stocks_for_search()

    return jsonify({
        'success': True,
        'message': '缓存已重新加载'
    })


# 静态文件服务（可选，用于直接访问 JSON 文件）
@app.route('/data/zt/<path:filename>')
def serve_data_file(filename):
    """直接提供静态 JSON 文件"""
    return send_from_directory(DATA_DIR, filename)


if __name__ == '__main__':
    # 预加载数据
    logger.info("预加载数据...")
    load_dates()
    load_stocks_for_search()
    logger.info("数据加载完成")

    app.run(debug=True, host='0.0.0.0', port=8800)