shortest path code snippets in python