425a7047c3
Public-facing page at / shows the current show's playlist with tracks obscured until the admin marks them as announced. Tracks reveal in real-time via a new unauthenticated /ws/public WebSocket. Server-side censorship on /public/playlist strips track details from unannounced items and sanitizes announced tracks to only expose frontend-needed fields (no raw_json, track_id, etc). Past episodes are browsable with fully revealed but sanitized tracklists. Also fixes RuntimeError on backfill shutdown by closing the httpx client on the same event loop that created it. Made-with: Cursor
111 lines
3.2 KiB
Python
111 lines
3.2 KiB
Python
import argparse
|
|
import asyncio
|
|
import logging
|
|
from datetime import date
|
|
|
|
import uvicorn
|
|
|
|
from ntr_fetcher.api import create_app
|
|
from ntr_fetcher.backfill import run_backfill
|
|
from ntr_fetcher.config import Settings
|
|
from ntr_fetcher.db import Database
|
|
from ntr_fetcher.poller import Poller
|
|
from ntr_fetcher.soundcloud import SoundCloudClient
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format="%(asctime)s %(levelname)s %(name)s: %(message)s",
|
|
)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def _parse_args() -> argparse.Namespace:
|
|
parser = argparse.ArgumentParser(description="NtR SoundCloud Fetcher")
|
|
parser.add_argument(
|
|
"--init", action="store_true",
|
|
help="Run historical backfill instead of starting the server",
|
|
)
|
|
parser.add_argument(
|
|
"--show", type=int,
|
|
help="Anchor episode number (required with --init)",
|
|
)
|
|
parser.add_argument(
|
|
"--aired", type=date.fromisoformat,
|
|
help="Air date of anchor episode as YYYY-MM-DD (required with --init)",
|
|
)
|
|
args = parser.parse_args()
|
|
if args.init and (args.show is None or args.aired is None):
|
|
parser.error("--init requires both --show and --aired")
|
|
return args
|
|
|
|
|
|
def run() -> None:
|
|
args = _parse_args()
|
|
settings = Settings()
|
|
|
|
db = Database(settings.db_path)
|
|
db.initialize()
|
|
logger.info("Database initialized at %s", settings.db_path)
|
|
|
|
if args.init:
|
|
sc = SoundCloudClient()
|
|
|
|
async def _backfill_and_close():
|
|
try:
|
|
await run_backfill(
|
|
db=db,
|
|
soundcloud=sc,
|
|
soundcloud_user=settings.soundcloud_user,
|
|
show_day=settings.show_day,
|
|
show_hour=settings.show_hour,
|
|
anchor_episode=args.show,
|
|
anchor_aired=args.aired,
|
|
)
|
|
finally:
|
|
await sc.close()
|
|
|
|
asyncio.run(_backfill_and_close())
|
|
logger.info("Backfill complete")
|
|
return
|
|
|
|
sc = SoundCloudClient()
|
|
poller = Poller(
|
|
db=db,
|
|
soundcloud=sc,
|
|
soundcloud_user=settings.soundcloud_user,
|
|
show_day=settings.show_day,
|
|
show_hour=settings.show_hour,
|
|
poll_interval=settings.poll_interval_seconds,
|
|
rotation_delay_hours=settings.show_rotation_delay_hours,
|
|
)
|
|
|
|
app = create_app(
|
|
db=db,
|
|
poller=poller,
|
|
admin_token=settings.admin_token,
|
|
show_day=settings.show_day,
|
|
show_hour=settings.show_hour,
|
|
rotation_delay_hours=settings.show_rotation_delay_hours,
|
|
web_user=settings.web_user,
|
|
web_password=settings.web_password,
|
|
secret_key=settings.secret_key,
|
|
ping_target=settings.ping_target,
|
|
ping_message=settings.ping_message,
|
|
)
|
|
|
|
@app.on_event("startup")
|
|
async def start_poller():
|
|
logger.info("Starting poller (interval=%ds)", settings.poll_interval_seconds)
|
|
asyncio.create_task(poller.run_supervised())
|
|
|
|
@app.on_event("shutdown")
|
|
async def shutdown():
|
|
logger.info("Shutting down")
|
|
await sc.close()
|
|
|
|
uvicorn.run(app, host=settings.host, port=settings.port)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run()
|