Doing queries in Python