Python Query Key Efficiently Definition