103 lines
2.7 KiB
Python
103 lines
2.7 KiB
Python
"""
|
||
会话路由
|
||
"""
|
||
from flask import Blueprint, request, jsonify
|
||
from flask_jwt_extended import jwt_required, get_jwt_identity
|
||
from datetime import datetime
|
||
from app.models import db, Session, Agent
|
||
|
||
sessions_bp = Blueprint('sessions', __name__)
|
||
|
||
|
||
@sessions_bp.route('/', methods=['GET'])
|
||
@jwt_required()
|
||
def get_sessions():
|
||
"""获取会话列表"""
|
||
user_id = get_jwt_identity()
|
||
sessions = Session.query.filter_by(user_id=user_id).all()
|
||
|
||
return jsonify({
|
||
'sessions': [s.to_dict() for s in sessions]
|
||
}), 200
|
||
|
||
|
||
@sessions_bp.route('/', methods=['POST'])
|
||
@jwt_required()
|
||
def create_session():
|
||
"""创建会话"""
|
||
user_id = get_jwt_identity()
|
||
data = request.get_json()
|
||
|
||
title = data.get('title', 'New Session')
|
||
agent_id = data.get('agent_id')
|
||
priority = data.get('priority', 5)
|
||
|
||
# 如果没有指定 Agent，分配一个
|
||
if not agent_id:
|
||
agent = Agent.query.filter_by(status='online').order_by(
|
||
Agent.current_sessions.asc()
|
||
).first()
|
||
if agent:
|
||
agent_id = agent.id
|
||
|
||
session = Session(
|
||
user_id=user_id,
|
||
primary_agent_id=agent_id,
|
||
title=title,
|
||
status='active'
|
||
)
|
||
|
||
db.session.add(session)
|
||
db.session.commit()
|
||
|
||
return jsonify({
|
||
'session': session.to_dict()
|
||
}), 201
|
||
|
||
|
||
@sessions_bp.route('/<session_id>', methods=['GET'])
|
||
@jwt_required()
|
||
def get_session(session_id):
|
||
"""获取会话详情"""
|
||
user_id = get_jwt_identity()
|
||
session = Session.query.filter_by(id=session_id, user_id=user_id).first()
|
||
|
||
if not session:
|
||
return jsonify({'error': 'Session not found'}), 404
|
||
|
||
return jsonify({'session': session.to_dict()}), 200
|
||
|
||
|
||
@sessions_bp.route('/<session_id>/close', methods=['PUT'])
|
||
@jwt_required()
|
||
def close_session(session_id):
|
||
"""关闭会话"""
|
||
user_id = get_jwt_identity()
|
||
session = Session.query.filter_by(id=session_id, user_id=user_id).first()
|
||
|
||
if not session:
|
||
return jsonify({'error': 'Session not found'}), 404
|
||
|
||
session.status = 'closed'
|
||
session.updated_at = datetime.utcnow()
|
||
db.session.commit()
|
||
|
||
return jsonify({'message': 'Session closed', 'session': session.to_dict()}), 200
|
||
|
||
|
||
@sessions_bp.route('/<session_id>/messages', methods=['GET'])
|
||
@jwt_required()
|
||
def get_session_messages(session_id):
|
||
"""获取会话消息"""
|
||
user_id = get_jwt_identity()
|
||
session = Session.query.filter_by(id=session_id, user_id=user_id).first()
|
||
|
||
if not session:
|
||
return jsonify({'error': 'Session not found'}), 404
|
||
|
||
messages = session.messages.order_by('created_at').all()
|
||
|
||
return jsonify({
|
||
'messages': [m.to_dict() for m in messages]
|
||
}), 200
|