103 lines
3.4 KiB
Python
103 lines
3.4 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.db.models import ScholarProfile
|
|
from app.services.domains.portability.normalize import _normalize_optional_text
|
|
from app.services.domains.scholars import application as scholar_service
|
|
|
|
|
|
async def _load_user_scholar_map(
|
|
db_session: AsyncSession,
|
|
*,
|
|
user_id: int,
|
|
) -> dict[str, ScholarProfile]:
|
|
result = await db_session.execute(select(ScholarProfile).where(ScholarProfile.user_id == user_id))
|
|
profiles = list(result.scalars().all())
|
|
return {profile.scholar_id: profile for profile in profiles}
|
|
|
|
|
|
def _apply_imported_scholar_values(
|
|
*,
|
|
profile: ScholarProfile,
|
|
display_name: str | None,
|
|
profile_image_override_url: str | None,
|
|
is_enabled: bool,
|
|
) -> bool:
|
|
updated = False
|
|
if display_name and profile.display_name != display_name:
|
|
profile.display_name = display_name
|
|
updated = True
|
|
if profile.profile_image_override_url != profile_image_override_url:
|
|
profile.profile_image_override_url = profile_image_override_url
|
|
updated = True
|
|
if bool(profile.is_enabled) != bool(is_enabled):
|
|
profile.is_enabled = bool(is_enabled)
|
|
updated = True
|
|
return updated
|
|
|
|
|
|
def _new_scholar_profile(
|
|
*,
|
|
user_id: int,
|
|
scholar_id: str,
|
|
display_name: str | None,
|
|
profile_image_override_url: str | None,
|
|
is_enabled: bool,
|
|
) -> ScholarProfile:
|
|
return ScholarProfile(
|
|
user_id=user_id,
|
|
scholar_id=scholar_id,
|
|
display_name=display_name,
|
|
profile_image_override_url=profile_image_override_url,
|
|
is_enabled=bool(is_enabled),
|
|
)
|
|
|
|
|
|
async def _upsert_imported_scholars(
|
|
db_session: AsyncSession,
|
|
*,
|
|
user_id: int,
|
|
scholars: list[dict[str, Any]],
|
|
) -> tuple[dict[str, ScholarProfile], dict[str, int]]:
|
|
scholar_map = await _load_user_scholar_map(db_session, user_id=user_id)
|
|
counters = {"scholars_created": 0, "scholars_updated": 0, "skipped_records": 0}
|
|
for item in scholars:
|
|
try:
|
|
scholar_id = scholar_service.validate_scholar_id(str(item["scholar_id"]))
|
|
display_name = scholar_service.normalize_display_name(str(item.get("display_name") or ""))
|
|
override_url = scholar_service.normalize_profile_image_url(
|
|
_normalize_optional_text(item.get("profile_image_override_url"))
|
|
)
|
|
except (KeyError, scholar_service.ScholarServiceError):
|
|
counters["skipped_records"] += 1
|
|
continue
|
|
|
|
is_enabled = bool(item.get("is_enabled", True))
|
|
existing = scholar_map.get(scholar_id)
|
|
if existing is None:
|
|
profile = _new_scholar_profile(
|
|
user_id=user_id,
|
|
scholar_id=scholar_id,
|
|
display_name=display_name,
|
|
profile_image_override_url=override_url,
|
|
is_enabled=is_enabled,
|
|
)
|
|
db_session.add(profile)
|
|
scholar_map[scholar_id] = profile
|
|
counters["scholars_created"] += 1
|
|
continue
|
|
|
|
if _apply_imported_scholar_values(
|
|
profile=existing,
|
|
display_name=display_name,
|
|
profile_image_override_url=override_url,
|
|
is_enabled=is_enabled,
|
|
):
|
|
counters["scholars_updated"] += 1
|
|
|
|
await db_session.flush()
|
|
return scholar_map, counters
|