aboutsummaryrefslogtreecommitdiff
path: root/src/srht_contrib/utils/identity.py
blob: adc9a9d8ab89e91f8c2566b544aa00b28694c6ef (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
from __future__ import annotations

from sqlalchemy import select
from sqlalchemy.orm import Session

from srht_contrib.models import ActorAlias


class ActorIdentityResolver:
    def __init__(self, configured_aliases: dict[str, list[str]] | None = None) -> None:
        self.configured_aliases = configured_aliases or {}

    def canonicalize(self, actor: str, db: Session | None = None) -> str:
        normalized = actor.strip()
        if not normalized:
            return normalized

        for canonical, aliases in self.configured_aliases.items():
            if normalized == canonical or normalized in aliases:
                return canonical

        if db is not None:
            alias = db.scalar(select(ActorAlias).where(ActorAlias.alias == normalized))
            if alias is not None:
                return alias.canonical_actor

        return normalized