59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
from flask import Blueprint, request, jsonify
|
|
from sqlalchemy.orm import Session
|
|
|
|
import api
|
|
from api import etag_response
|
|
from db import get_db
|
|
from db.models.Markdown import Markdown
|
|
from db.models.Path import Path
|
|
from api import limiter
|
|
import logging
|
|
logger = logging.getLogger(__name__)
|
|
|
|
tree_bp = Blueprint('tree', __name__, url_prefix='/api/tree')
|
|
|
|
|
|
def build_tree(db: Session, parent_id: int = None):
|
|
path_nodes = db.query(Path).filter(Path.parent_id == parent_id).all()
|
|
md_nodes = db.query(
|
|
Markdown.id,
|
|
Markdown.title,
|
|
Markdown.order,
|
|
Markdown.shortcut,
|
|
Markdown.setting_id
|
|
).filter(Markdown.path_id == parent_id).all()
|
|
t0 = [
|
|
{
|
|
"id": node.id,
|
|
"title": node.title,
|
|
"order": node.order,
|
|
"setting_id": node.setting_id,
|
|
"type": "markdown"
|
|
} for node in md_nodes
|
|
]
|
|
t1 = [
|
|
{**node.to_dict(), "type": "path", "children": build_tree(db, node.id)} for node in path_nodes
|
|
]
|
|
for node in t1:
|
|
for child in node["children"]:
|
|
if "title" in child.keys() and child["title"] == "index":
|
|
node["index"] = True
|
|
break
|
|
return t0 + t1
|
|
|
|
@tree_bp.route('/', methods=['GET'])
|
|
@limiter.limit(api.get_rate_limit)
|
|
@etag_response
|
|
def get_tree():
|
|
with get_db() as session:
|
|
children = build_tree(session, 1)
|
|
root = session.query(Path).get(1)
|
|
|
|
return jsonify(
|
|
{
|
|
**root.to_dict(),
|
|
"type": "path",
|
|
"index": any("title" in child.keys() and child["title"] == "index" for child in children),
|
|
"children": children
|
|
}
|
|
), 200 |