Query Function In Python