codeql/python/ql/test/library-tests/state_tracking/Lib.qll

import python
import semmle.python.dataflow.StateTracking

predicate callTo(CallNode call, string name) { call.getFunction().(NameNode).getId() = name }

class Initialized extends TrackableState {
    Initialized() { this = "initialized" }

    override predicate startsAt(ControlFlowNode f) { callTo(f, "initialize") }
}

class Frobnicated extends TrackableState {
    Frobnicated() { this = "frobnicated" }

    override predicate startsAt(ControlFlowNode f) { callTo(f, "frobnicate") }

    override predicate endsAt(ControlFlowNode f) { callTo(f, "defrobnicate") }
}