106 lines
3.5 KiB
Python
106 lines
3.5 KiB
Python
import logging
|
|
|
|
from fastapi import HTTPException
|
|
|
|
from ..clients.jellyfin import JellyfinClient
|
|
from ..db import (
|
|
create_user_if_missing,
|
|
get_user_by_username,
|
|
set_user_email,
|
|
set_user_auth_provider,
|
|
set_user_jellyseerr_id,
|
|
)
|
|
from ..runtime import get_runtime_settings
|
|
from .jellyfin_identity import link_user
|
|
from .user_cache import (
|
|
build_jellyseerr_candidate_map,
|
|
extract_jellyseerr_user_email,
|
|
find_matching_jellyseerr_user,
|
|
get_cached_jellyseerr_users,
|
|
match_jellyseerr_user_id,
|
|
save_jellyfin_users_cache,
|
|
)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
async def sync_jellyfin_users() -> int:
|
|
runtime = get_runtime_settings()
|
|
client = JellyfinClient(runtime.jellyfin_base_url, runtime.jellyfin_api_key)
|
|
if not client.configured():
|
|
raise HTTPException(status_code=400, detail="Jellyfin not configured")
|
|
users = await client.get_users()
|
|
if not isinstance(users, list):
|
|
return 0
|
|
save_jellyfin_users_cache(users)
|
|
# Jellyfin is the canonical source for local user objects; Seerr IDs are
|
|
# matched as enrichment when possible.
|
|
jellyseerr_users = get_cached_jellyseerr_users()
|
|
candidate_map = build_jellyseerr_candidate_map(jellyseerr_users or [])
|
|
imported = 0
|
|
for user in users:
|
|
if not isinstance(user, dict):
|
|
continue
|
|
name = user.get("Name")
|
|
if not name:
|
|
continue
|
|
matched_id = match_jellyseerr_user_id(name, candidate_map) if candidate_map else None
|
|
matched_seerr_user = find_matching_jellyseerr_user(name, jellyseerr_users or [])
|
|
matched_email = extract_jellyseerr_user_email(matched_seerr_user)
|
|
created = create_user_if_missing(
|
|
name,
|
|
"jellyfin-user",
|
|
role="user",
|
|
email=matched_email,
|
|
auth_provider="jellyfin",
|
|
jellyseerr_user_id=matched_id,
|
|
)
|
|
if created:
|
|
imported += 1
|
|
else:
|
|
existing = get_user_by_username(name)
|
|
if (
|
|
existing
|
|
and str(existing.get("role") or "user").strip().lower() != "admin"
|
|
and str(existing.get("auth_provider") or "local").strip().lower() != "jellyfin"
|
|
):
|
|
set_user_auth_provider(name, "jellyfin")
|
|
if matched_id is not None:
|
|
set_user_jellyseerr_id(name, matched_id)
|
|
if matched_email:
|
|
set_user_email(name, matched_email)
|
|
if user.get("Id"):
|
|
local_user = get_user_by_username(name)
|
|
if local_user and local_user.get("auth_provider") == "jellyfin":
|
|
link_user(name, str(user["Id"]), runtime.jellyfin_base_url)
|
|
return imported
|
|
|
|
|
|
async def run_daily_jellyfin_sync() -> None:
|
|
while True:
|
|
delay = _seconds_until_midnight()
|
|
await _sleep_seconds(delay)
|
|
try:
|
|
imported = await sync_jellyfin_users()
|
|
logger.info("Jellyfin daily sync complete: imported=%s", imported)
|
|
except HTTPException as exc:
|
|
logger.warning("Jellyfin daily sync skipped: %s", exc.detail)
|
|
except Exception:
|
|
logger.exception("Jellyfin daily sync failed")
|
|
|
|
|
|
def _seconds_until_midnight() -> float:
|
|
from datetime import datetime, timedelta
|
|
|
|
now = datetime.now()
|
|
next_midnight = (now + timedelta(days=1)).replace(
|
|
hour=0, minute=0, second=0, microsecond=0
|
|
)
|
|
return max((next_midnight - now).total_seconds(), 0.0)
|
|
|
|
|
|
async def _sleep_seconds(delay: float) -> None:
|
|
import asyncio
|
|
|
|
await asyncio.sleep(delay)
|