122 lines
4.0 KiB
Python
122 lines
4.0 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import UTC, datetime, timedelta
|
|
import json
|
|
|
|
from app.const import BANCHOBOT_ID
|
|
from app.database.playlists import Playlist
|
|
from app.database.room import Room
|
|
from app.dependencies.database import get_redis, with_db
|
|
from app.dependencies.scheduler import get_scheduler
|
|
from app.log import logger
|
|
from app.models.metadata_hub import DailyChallengeInfo
|
|
from app.models.mods import APIMod
|
|
from app.models.room import RoomCategory
|
|
|
|
from .room import create_playlist_room
|
|
|
|
from sqlmodel import col, select
|
|
|
|
|
|
async def create_daily_challenge_room(
|
|
beatmap: int, ruleset_id: int, duration: int, required_mods: list[APIMod] = []
|
|
) -> Room:
|
|
async with with_db() as session:
|
|
today = datetime.now(UTC).date()
|
|
return await create_playlist_room(
|
|
session=session,
|
|
name=str(today),
|
|
host_id=BANCHOBOT_ID,
|
|
playlist=[
|
|
Playlist(
|
|
id=0,
|
|
room_id=0,
|
|
owner_id=BANCHOBOT_ID,
|
|
ruleset_id=ruleset_id,
|
|
beatmap_id=beatmap,
|
|
required_mods=required_mods,
|
|
)
|
|
],
|
|
category=RoomCategory.DAILY_CHALLENGE,
|
|
duration=duration,
|
|
)
|
|
|
|
|
|
@get_scheduler().scheduled_job("cron", hour=0, minute=0, second=0, id="daily_challenge")
|
|
async def daily_challenge_job():
|
|
from app.signalr.hub import MetadataHubs
|
|
|
|
now = datetime.now(UTC)
|
|
redis = get_redis()
|
|
key = f"daily_challenge:{now.date()}"
|
|
if not await redis.exists(key):
|
|
return
|
|
async with with_db() as session:
|
|
room = (
|
|
await session.exec(
|
|
select(Room).where(
|
|
Room.category == RoomCategory.DAILY_CHALLENGE,
|
|
col(Room.ends_at) > datetime.now(UTC),
|
|
)
|
|
)
|
|
).first()
|
|
if room:
|
|
return
|
|
|
|
try:
|
|
beatmap = await redis.hget(key, "beatmap") # pyright: ignore[reportGeneralTypeIssues]
|
|
ruleset_id = await redis.hget(key, "ruleset_id") # pyright: ignore[reportGeneralTypeIssues]
|
|
required_mods = await redis.hget(key, "required_mods") # pyright: ignore[reportGeneralTypeIssues]
|
|
|
|
if beatmap is None or ruleset_id is None:
|
|
logger.warning(
|
|
f"[DailyChallenge] Missing required data for daily challenge {now}."
|
|
" Will try again in 5 minutes."
|
|
)
|
|
get_scheduler().add_job(
|
|
daily_challenge_job,
|
|
"date",
|
|
run_date=datetime.now(UTC) + timedelta(minutes=5),
|
|
)
|
|
return
|
|
|
|
beatmap_int = int(beatmap)
|
|
ruleset_id_int = int(ruleset_id)
|
|
|
|
mods_list = []
|
|
if required_mods:
|
|
mods_list = json.loads(required_mods)
|
|
|
|
next_day = (now + timedelta(days=1)).replace(
|
|
hour=0, minute=0, second=0, microsecond=0
|
|
)
|
|
room = await create_daily_challenge_room(
|
|
beatmap=beatmap_int,
|
|
ruleset_id=ruleset_id_int,
|
|
required_mods=mods_list,
|
|
duration=int((next_day - now - timedelta(minutes=2)).total_seconds() / 60),
|
|
)
|
|
await MetadataHubs.broadcast_call(
|
|
"DailyChallengeUpdated", DailyChallengeInfo(room_id=room.id)
|
|
)
|
|
logger.success(
|
|
"[DailyChallenge] Added today's daily challenge: "
|
|
f"{beatmap=}, {ruleset_id=}, {required_mods=}"
|
|
)
|
|
return
|
|
except (ValueError, json.JSONDecodeError) as e:
|
|
logger.warning(
|
|
f"[DailyChallenge] Error processing daily challenge data: {e}"
|
|
" Will try again in 5 minutes."
|
|
)
|
|
except Exception as e:
|
|
logger.exception(
|
|
f"[DailyChallenge] Unexpected error in daily challenge job: {e}"
|
|
" Will try again in 5 minutes."
|
|
)
|
|
get_scheduler().add_job(
|
|
daily_challenge_job,
|
|
"date",
|
|
run_date=datetime.now(UTC) + timedelta(minutes=5),
|
|
)
|