codeql/python/ql/test/library-tests/PointsTo/customise/test.ql

import python
import semmle.python.types.Extensions

/*
 * Customise: Claim any function called has_type_XXX return any class
 * whose name matches XXX
 */

class HasTypeFact extends CustomPointsToOriginFact {
  HasTypeFact() {
    exists(FunctionObject func, string name |
      func.getACall() = this and
      name = func.getName() and
      name.prefix("has_type_".length()) = "has_type_"
    )
  }

  override predicate pointsTo(Object value, ClassObject cls) {
    exists(FunctionObject func, string name |
      func.getACall() = this and
      name = func.getName() and
      name.prefix("has_type_".length()) = "has_type_"
    |
      cls.getName() = name.suffix("has_type_".length())
    ) and
    value = this
  }
}

from int line, ControlFlowNode f, Object o, ClassObject c
where
  f.getLocation().getStartLine() = line and
  exists(Comment ct | ct.getLocation().getStartLine() < line) and
  f.refersTo(o, c, _)
select line, f.toString(), o.toString(), c.toString()