from flask import Blueprint, request, jsonify
from sqlalchemy import desc

from utils import pgclass, dbhelper
from utils.misc import error

log = Blueprint('log', __name__)

# 列出log
@log.route("/list", methods = ["GET"])
def listlog():
    # variables
    if request.args.get("start") is None or request.args.get("count") is None or \
            not request.args.get("start").isdigit() or not request.args.get("count").isdigit(): 
        return error("Arguments error"), 400
    rst = int(request.args.get("start"))
    count = int(request.args.get("count"))

    # getctx
    with dbhelper.db.getsession() as session:
        table = pgclass.SQLlog
        res = session.query(table).order_by(desc(table.id)).offset(rst).limit(count).all()

    # mapping
    res = [ {"id":r.id, "created_at":r.created_at, "source":r.source, "message":r.message} for r in res ]

    return jsonify(res), 200

# 指定顯示特定一條log
@log.route("/<int:id>", methods = ["GET"])
def getlog(id:int):
    # db
    with dbhelper.db.getsession() as session:
        table = pgclass.SQLlog
        res = session.query(table).filter(table.id == id).first()
    if res is None:
        return error("Log not Found"), 404
    
    # mapping
    res = {"id":res.id, "created_at":res.created_at, "source":res.source, "message":res.message}

    return jsonify(res), 200