transfer from monorepo
This commit is contained in:
38
worker/orca-agent/src/server/routes/submission.py
Normal file
38
worker/orca-agent/src/server/routes/submission.py
Normal file
@ -0,0 +1,38 @@
|
||||
from flask import Blueprint, jsonify
|
||||
from prometheus_swarm.database import get_db
|
||||
from src.dababase.models import Submission
|
||||
import logging
|
||||
import os
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
bp = Blueprint("submission", __name__)
|
||||
|
||||
|
||||
@bp.get("/submission/<roundNumber>")
|
||||
def fetch_submission(roundNumber):
|
||||
logger.info(f"Fetching submission for round: {roundNumber}")
|
||||
db = get_db()
|
||||
submission = (
|
||||
db.query(Submission)
|
||||
.filter(
|
||||
Submission.round_number == int(roundNumber),
|
||||
)
|
||||
.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,
|
||||
"roundNumber": submission.round_number,
|
||||
"status": submission.status,
|
||||
"prUrl": submission.pr_url,
|
||||
"githubUsername": github_username,
|
||||
}
|
||||
)
|
||||
else:
|
||||
return jsonify({"error": "Submission not found"}), 409
|
Reference in New Issue
Block a user