ipython_startup.py

from typing import Any

from sqlalchemy.orm import Session

from railbird import util
from railbird.datatypes import models

util, Session

try:
    container: Any
    session: Session
    context["container"]  # noqa: F821
except Exception:
    context: dict
    container = object()
    session = Session()
    context = None


context["shot_count"] = (
    session.query(models.ShotModel)
    .where(
        models.ShotModel.annotation_any(
            models.ShotModel.annotation_all(
                "(l.3) True Positive - Likely Confounding", "(l.2) True Positive"
            ),
            "(l.1) False Positive",
        )
    )
    .count()
)