kschappell/codetutor
Public Access
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(backend): data loader handles test cases

Browse Source
This commit is contained in:
Kai Chappell
2025-05-21 20:41:04 +01:00
parent dd9a027c5a
commit f9c404b69e
1 changed files with 18 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
backend/scripts/load_data.py
View File

@@ -9,6 +9,7 @@ from typing import Any
import yaml import yaml
from sqlalchemy import select from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.orm import selectinload
# Add src to path for imports # Add src to path for imports
sys.path.insert(0, str(Path(__file__).parent.parent)) sys.path.insert(0, str(Path(__file__).parent.parent))
@@ -96,28 +97,41 @@ async def load_question(
data: dict[str, Any] = yaml.safe_load(f) data: dict[str, Any] = yaml.safe_load(f)
slug = data["slug"] slug = data["slug"]
result = await session.execute(select(Question).where(Question.slug == slug)) result = await session.execute(
select(Question)
.where(Question.slug == slug)
.options(
selectinload(Question.explanation),
selectinload(Question.solutions),
selectinload(Question.categories),
selectinload(Question.patterns),
)
)
existing = result.scalar_one_or_none() existing = result.scalar_one_or_none()
if existing: if existing:
question = existing question = existing
question.title = data["title"] question.title = data["title"]
question.difficulty = Difficulty(data["difficulty"]) question.difficulty = Difficulty(data["difficulty"].lower())
question.description = data["description"] question.description = data["description"]
question.constraints = data.get("constraints") question.constraints = data.get("constraints")
question.examples = data.get("examples") question.examples = data.get("examples")
question.leetcode_id = data.get("leetcode_id") question.leetcode_id = data.get("leetcode_id")
question.leetcode_url = data.get("leetcode_url") question.leetcode_url = data.get("leetcode_url")
question.function_signature = data.get("function_signature")
question.test_cases = data.get("test_cases")
else: else:
question = Question( question = Question(
title=data["title"], title=data["title"],
slug=slug, slug=slug,
difficulty=Difficulty(data["difficulty"]), difficulty=Difficulty(data["difficulty"].lower()),
description=data["description"], description=data["description"],
constraints=data.get("constraints"), constraints=data.get("constraints"),
examples=data.get("examples"), examples=data.get("examples"),
leetcode_id=data.get("leetcode_id"), leetcode_id=data.get("leetcode_id"),
leetcode_url=data.get("leetcode_url"), leetcode_url=data.get("leetcode_url"),
function_signature=data.get("function_signature"),
test_cases=data.get("test_cases"),
) )
session.add(question) session.add(question)
@@ -136,7 +150,7 @@ async def load_question(
# Handle explanation # Handle explanation
if "explanation" in data: if "explanation" in data:
exp_data = data["explanation"] exp_data = data["explanation"]
if question.explanation: if existing and existing.explanation:
explanation = question.explanation explanation = question.explanation
explanation.approach = exp_data["approach"] explanation.approach = exp_data["approach"]
explanation.intuition = exp_data["intuition"] explanation.intuition = exp_data["intuition"]