AoC Day6

# Please use the following encoding: utf-8, thanks!
import networkx as nx

''' Part 1'''
graph = nx.read_edgelist('./data/day6_input.txt', delimiter=')', create_using=nx.DiGraph)
orbit_count = sum(len(nx.ancestors(G=graph, source=node)) for node in graph.nodes)
print(orbit_count)
''' Part 2 '''
undir_graph = nx.read_edgelist('./data/day6_input.txt', delimiter=')', create_using=nx.Graph)
shortest = len(nx.shortest_path(undir_graph, source="YOU", target="SAN")) - 3
print(shortest)