log.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
# @Version     : Python 3.11.4
# @Software    : Sublime Text 4
# @Author      : StudentCWZ
# @Email       : StudentCWZ@outlook.com
# @Date        : 2023/11/19 16:05
# @File        : log.py
# @Description : Defines the log API endpoints.
"""

from flask import Blueprint, jsonify, request
from flask_restful import Api
from loguru import logger
from pydantic import ValidationError

from application.schemas import ParseLogRequestItem
from application.services import LogService


# Define the log API blueprint
log_api = Blueprint('log_api', __name__)
api_logs = Api(log_api)


@log_api.route('/logs', methods=['POST'])
def parse():
    """
    Parses the logs based on the request.

    Returns:
        The total number of logs parsed or an error message if the input data is invalid.
    """
    json_data = request.get_json(force=True)

    # If no data provided, return error message
    if not json_data:
        logger.error('No input data provided')
        return jsonify({"message": "No input data provided"}), 400

    try:
        # Validate input data
        item = ParseLogRequestItem(**json_data)
    except ValidationError as e:
        # If validation fails, return error message
        return jsonify({"message": "Invalid input data", "errors": e.errors()}), 400

    # If validation passes, parse logs and return total number of logs parsed
    total = LogService.parse(item)
    return jsonify(search_total=total), 200