Files
Yiqun/worker/orca-agent/src/server/routes/submission.py

39 lines
1.1 KiB
Python

from flask import Blueprint, jsonify
from prometheus_swarm.database import get_db
from src.database.models import Submission
import logging
import os
logger = logging.getLogger(__name__)
bp = Blueprint("submission", __name__)
@bp.get("/submission/<swarmBountyId>")
def fetch_submission(swarmBountyId):
logger.info(f"Fetching submission for swarmBountyId: {swarmBountyId}")
db = get_db()
submission = (
db.query(Submission)
.filter(
Submission.swarmBountyId == int(swarmBountyId),
)
.first()
)
logger.info(f"Submission: {submission}")
logger.info(f"Submission: {submission}")
if submission:
github_username = os.getenv("GITHUB_USERNAME")
return jsonify(
{
"taskId": submission.task_id,
"swarmBountyId": submission.swarmBountyId,
"status": submission.status,
"prUrl": submission.pr_url,
"githubUsername": github_username,
}
)
else:
return jsonify({"error": "Submission not found"}), 409