import python
import semmle.python.ApiGraphs

abstract class LoggingSinks extends DataFlow::Node { }

class PrintMethod extends LoggingSinks {
  PrintMethod() {
    exists(DataFlow::Node call |
      call = API::builtin("print").getACall() and
      call.getScope().inSource() and
      this = call
    )
  }
}

class LoggingFramework extends LoggingSinks {
  LoggingFramework() {
    exists(DataFlow::Node call, API::Node node |
      (
        (
          node = API::moduleImport("logging")
          or
          node = API::moduleImport("logging").getMember("getLogger").getReturn()
        ) and
        call = node.getMember(["info", "debug", "warning", "error"]).getACall()
      ) and
      call.getScope().inSource() and
      this = call
    )
  }
}