refactor(fc2c-i): sweep blueprints + ml tasks onto shared get_session (covers tasks/ml.py, a survey gap)
This commit is contained in:
@@ -2,27 +2,15 @@
|
||||
|
||||
from quart import Blueprint, jsonify, request
|
||||
|
||||
from ..extensions import make_engine, make_session_factory
|
||||
from ..extensions import get_session
|
||||
from ..services.ml.aliases import AliasService
|
||||
|
||||
aliases_bp = Blueprint("aliases", __name__, url_prefix="/api")
|
||||
|
||||
_engine = None
|
||||
_Session = None
|
||||
|
||||
|
||||
def _session_factory():
|
||||
global _engine, _Session
|
||||
if _engine is None:
|
||||
_engine = make_engine()
|
||||
_Session = make_session_factory(_engine)
|
||||
return _Session
|
||||
|
||||
|
||||
@aliases_bp.route("/aliases", methods=["GET"])
|
||||
async def list_aliases():
|
||||
Session = _session_factory()
|
||||
async with Session() as session:
|
||||
async with get_session() as session:
|
||||
rows = await AliasService(session).list_all()
|
||||
return jsonify(
|
||||
[
|
||||
@@ -43,8 +31,7 @@ async def create_alias():
|
||||
required = {"alias_string", "alias_category", "canonical_tag_id"}
|
||||
if not body or not required.issubset(body):
|
||||
return jsonify({"error": f"required: {sorted(required)}"}), 400
|
||||
Session = _session_factory()
|
||||
async with Session() as session:
|
||||
async with get_session() as session:
|
||||
await AliasService(session).create(
|
||||
body["alias_string"],
|
||||
body["alias_category"],
|
||||
@@ -58,8 +45,7 @@ async def create_alias():
|
||||
"/aliases/<alias_string>/<alias_category>", methods=["DELETE"]
|
||||
)
|
||||
async def remove_alias(alias_string: str, alias_category: str):
|
||||
Session = _session_factory()
|
||||
async with Session() as session:
|
||||
async with get_session() as session:
|
||||
await AliasService(session).remove(alias_string, alias_category)
|
||||
await session.commit()
|
||||
return "", 204
|
||||
|
||||
Reference in New Issue
Block a user