Query patterns

import encodebox as eb

eb.register_connector("connectors/postgres_local_sentinel/connector.yml")
db = eb.connect(..., connector="postgres_local_sentinel")

# Colonoscopy CPT list — any code matches
db.procedure.matching(
    ["45378", "45380", "45385"],
    code_type="C4",
    match="exactly",
)

# Diabetes OR hypertension as separate sets
db.diagnosis.matching(
    [["E11"], ["I10"]],
    code_type="10",
    match=["starts_with", "exactly"],
)

confirmed = (
    db.diagnosis
    .matching("E11", code_type="10", match="starts_with")
    .occurring(at_least=2)
    .first_per_patient()
)

index = (
    db.diagnosis
    .matching("E11", code_type="10", match="starts_with")
    .occurring(at_least=2, gap_days=30)
    .with_index_date()
    .first_per_patient()
)

db.diagnosis.matching("E11", code_type="10").in_position("primary")

cohort = (
    eb.Cohort("study", database=db)
    .entry(index, index="first")
    .include(db.demographic.age_at_index(18, 89))
    .include(db.enrollment.covering_index(days_before=365, days_after=0))
    .exclude(db.diagnosis.matching("C", match="starts_with"), window=(-365, 0))
)
print(cohort.attrition())

other = (
    db.diagnosis
    .matching("E10", code_type="10", match="starts_with")
    .occurring(at_least=2, gap_days=30)
    .with_index_date()
    .first_per_patient()
)

cohort = (
    eb.Cohort("t2dm", database=db)
    .entry(index, index="first")
    .include(db.demographic.age_at_index(18, 89))
    .include(db.enrollment.covering_index(days_before=365, days_after=0))
    .include(other, window=(-365, 0))
    .exclude(db.diagnosis.matching("C", match="starts_with"), window=(-365, 0))
)

dx = db.diagnosis.matching("E11", code_type="10")
px = db.procedure.matching("45378", code_type="C4", match="exactly")
combined = dx.union(px)