anti
614780f144
- Add _MixinBase abstract class to _helpers.py: declares _session(), _deserialize_attacker(), _assert_pending(), _check_and_bump_version(), and list_running_topology_deckies() so mypy can see cross-mixin contracts - Add _require(val, msg) helper for narrowing T | None → T - Inherit _MixinBase in all 26 leaf mixin classes - Wrap SQLAlchemy column method calls (.is_(), .like(), .notin_(), .in_(), .contains()) with col() from sqlmodel — fixes attr-defined false positives caused by pydantic plugin typing class-level fields as Python value types - Wrap select(Model.field) with select(col(Model.field)) for column projections - Add pyproject.toml [[tool.mypy.overrides]] to disable arg-type in sqlmodel_repo.*: pydantic plugin resolves .where(Model.field == v) as where(bool), a false positive; call-arg still catches real argument errors - Remove 9 stale # type: ignore comments (logging, helpers, credentials) - Fix telemetry.py traced() overload no-redef + misc - Fix logs.py datetime/str operator and nullable PK comparison with col() - sqlmodel_repo/ now has 0 mypy errors
52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
"""Per-session profile rows (keystroke-dynamics features land here at
|
|
ingestion-time post-V2)."""
|
|
from __future__ import annotations
|
|
|
|
from typing import Any, Optional
|
|
|
|
from sqlalchemy import select
|
|
|
|
from decnet.web.db.models import SessionProfile
|
|
|
|
|
|
from decnet.web.db.sqlmodel_repo._helpers import _MixinBase
|
|
|
|
class SessionProfilesMixin(_MixinBase):
|
|
async def upsert_session_profile(
|
|
self,
|
|
sid: str,
|
|
data: dict[str, Any],
|
|
) -> None:
|
|
"""
|
|
Write (or update) the session_profile row for *sid*.
|
|
|
|
Pre-v1, the typical call is the empty-write path at session close:
|
|
`upsert_session_profile(sid, {"log_id": <id>})` — all keystroke
|
|
feature columns stay NULL until the V2 ingestion job populates them.
|
|
"""
|
|
async with self._session() as session:
|
|
result = await session.execute(
|
|
select(SessionProfile).where(SessionProfile.sid == sid)
|
|
)
|
|
existing = result.scalar_one_or_none()
|
|
if existing:
|
|
for k, v in data.items():
|
|
setattr(existing, k, v)
|
|
session.add(existing)
|
|
else:
|
|
session.add(SessionProfile(sid=sid, **data))
|
|
await session.commit()
|
|
|
|
async def get_session_profile(
|
|
self,
|
|
sid: str,
|
|
) -> Optional[dict[str, Any]]:
|
|
async with self._session() as session:
|
|
result = await session.execute(
|
|
select(SessionProfile).where(SessionProfile.sid == sid)
|
|
)
|
|
row = result.scalar_one_or_none()
|
|
if not row:
|
|
return None
|
|
return row.model_dump(mode="json")
|