Intelligently search in Python code