rework shop
This commit is contained in:
@@ -181,18 +181,29 @@ async def use_consumable(
|
||||
# Get participant
|
||||
participant = await require_participant(db, current_user.id, data.marathon_id)
|
||||
|
||||
# For skip and reroll, we need the assignment
|
||||
# For some consumables, we need the assignment
|
||||
assignment = None
|
||||
if data.item_code in ["skip", "reroll"]:
|
||||
if data.item_code in ["skip", "wild_card", "copycat"]:
|
||||
if not data.assignment_id:
|
||||
raise HTTPException(status_code=400, detail="assignment_id is required for skip/reroll")
|
||||
raise HTTPException(status_code=400, detail=f"assignment_id is required for {data.item_code}")
|
||||
|
||||
result = await db.execute(
|
||||
select(Assignment).where(
|
||||
Assignment.id == data.assignment_id,
|
||||
Assignment.participant_id == participant.id,
|
||||
# For copycat, we need bonus_assignments to properly handle playthrough
|
||||
if data.item_code == "copycat":
|
||||
result = await db.execute(
|
||||
select(Assignment)
|
||||
.options(selectinload(Assignment.bonus_assignments))
|
||||
.where(
|
||||
Assignment.id == data.assignment_id,
|
||||
Assignment.participant_id == participant.id,
|
||||
)
|
||||
)
|
||||
else:
|
||||
result = await db.execute(
|
||||
select(Assignment).where(
|
||||
Assignment.id == data.assignment_id,
|
||||
Assignment.participant_id == participant.id,
|
||||
)
|
||||
)
|
||||
)
|
||||
assignment = result.scalar_one_or_none()
|
||||
if not assignment:
|
||||
raise HTTPException(status_code=404, detail="Assignment not found")
|
||||
@@ -201,15 +212,29 @@ async def use_consumable(
|
||||
if data.item_code == "skip":
|
||||
effect = await consumables_service.use_skip(db, current_user, participant, marathon, assignment)
|
||||
effect_description = "Assignment skipped without penalty"
|
||||
elif data.item_code == "shield":
|
||||
effect = await consumables_service.use_shield(db, current_user, participant, marathon)
|
||||
effect_description = "Shield activated - next drop will be free"
|
||||
elif data.item_code == "boost":
|
||||
effect = await consumables_service.use_boost(db, current_user, participant, marathon)
|
||||
effect_description = f"Boost x{effect['multiplier']} activated for next complete"
|
||||
elif data.item_code == "reroll":
|
||||
effect = await consumables_service.use_reroll(db, current_user, participant, marathon, assignment)
|
||||
effect_description = "Assignment rerolled - you can spin again"
|
||||
effect_description = f"Boost x{effect['multiplier']} activated for current assignment"
|
||||
elif data.item_code == "wild_card":
|
||||
if data.game_id is None:
|
||||
raise HTTPException(status_code=400, detail="game_id is required for wild_card")
|
||||
effect = await consumables_service.use_wild_card(
|
||||
db, current_user, participant, marathon, assignment, data.game_id
|
||||
)
|
||||
effect_description = f"New challenge from {effect['game_name']}: {effect['challenge_title']}"
|
||||
elif data.item_code == "lucky_dice":
|
||||
effect = await consumables_service.use_lucky_dice(db, current_user, participant, marathon)
|
||||
effect_description = f"Lucky Dice rolled: x{effect['multiplier']} multiplier"
|
||||
elif data.item_code == "copycat":
|
||||
if data.target_participant_id is None:
|
||||
raise HTTPException(status_code=400, detail="target_participant_id is required for copycat")
|
||||
effect = await consumables_service.use_copycat(
|
||||
db, current_user, participant, marathon, assignment, data.target_participant_id
|
||||
)
|
||||
effect_description = f"Copied challenge: {effect['challenge_title']}"
|
||||
elif data.item_code == "undo":
|
||||
effect = await consumables_service.use_undo(db, current_user, participant, marathon)
|
||||
effect_description = f"Restored {effect['points_restored']} points and streak {effect['streak_restored']}"
|
||||
else:
|
||||
raise HTTPException(status_code=400, detail=f"Unknown consumable: {data.item_code}")
|
||||
|
||||
@@ -243,9 +268,11 @@ async def get_consumables_status(
|
||||
|
||||
# Get inventory counts for all consumables
|
||||
skips_available = await consumables_service.get_consumable_count(db, current_user.id, "skip")
|
||||
shields_available = await consumables_service.get_consumable_count(db, current_user.id, "shield")
|
||||
boosts_available = await consumables_service.get_consumable_count(db, current_user.id, "boost")
|
||||
rerolls_available = await consumables_service.get_consumable_count(db, current_user.id, "reroll")
|
||||
wild_cards_available = await consumables_service.get_consumable_count(db, current_user.id, "wild_card")
|
||||
lucky_dice_available = await consumables_service.get_consumable_count(db, current_user.id, "lucky_dice")
|
||||
copycats_available = await consumables_service.get_consumable_count(db, current_user.id, "copycat")
|
||||
undos_available = await consumables_service.get_consumable_count(db, current_user.id, "undo")
|
||||
|
||||
# Calculate remaining skips for this marathon
|
||||
skips_remaining = None
|
||||
@@ -256,12 +283,16 @@ async def get_consumables_status(
|
||||
skips_available=skips_available,
|
||||
skips_used=participant.skips_used,
|
||||
skips_remaining=skips_remaining,
|
||||
shields_available=shields_available,
|
||||
has_shield=participant.has_shield,
|
||||
boosts_available=boosts_available,
|
||||
has_active_boost=participant.has_active_boost,
|
||||
boost_multiplier=consumables_service.BOOST_MULTIPLIER if participant.has_active_boost else None,
|
||||
rerolls_available=rerolls_available,
|
||||
wild_cards_available=wild_cards_available,
|
||||
lucky_dice_available=lucky_dice_available,
|
||||
has_lucky_dice=participant.has_lucky_dice,
|
||||
lucky_dice_multiplier=participant.lucky_dice_multiplier,
|
||||
copycats_available=copycats_available,
|
||||
undos_available=undos_available,
|
||||
can_undo=participant.can_undo,
|
||||
)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user