scientific-programming-exer.../exam/ex01/main.py

35 lines
970 B
Python
Raw Normal View History

2019-02-02 10:18:57 +00:00
import logging
logging.basicConfig(level=logging.DEBUG)
from cache import get_cache
from receive import receive_links, receive_link_graph
2019-02-02 15:06:57 +00:00
from dijkstra import prepare_dijkstra, dijkstra
2019-02-15 10:47:50 +00:00
from connectivity import shortest_path
2019-02-02 15:06:57 +00:00
2019-02-21 16:14:17 +00:00
from graph import DijkstraHelper
2019-02-27 10:41:44 +00:00
from db_util import get_page_id, get_page_title
2019-02-21 16:14:17 +00:00
2019-02-25 13:43:15 +00:00
cache = get_cache()
receive_link_graph("Angela_Merkel", cache, 2, lang="en")
2019-02-02 15:06:57 +00:00
cursor = cache.cursor()
cursor.execute("SELECT COUNT(source) FROM links")
print(cursor.fetchall())
2019-02-15 10:47:50 +00:00
#prepare_dijkstra(cache)
#dijkstra("Angela_Merkel", cache)
2019-02-21 16:14:17 +00:00
#
angela = get_page_id("Angela_Merkel", cache)
2019-02-02 15:06:57 +00:00
2019-02-21 16:14:17 +00:00
dijkstra = DijkstraHelper.from_db(cache)
dijkstra.dijkstra(angela)
dijkstra.write_back(cache)
2019-02-27 10:41:44 +00:00
#print({k:v for k,v in dijkstra._nodes.items() if v != float("inf")})
print([get_page_title(id_, cache) for id_ in shortest_path("Angela_Merkel", "Germany", cache)])
print([get_page_title(id_, cache) for id_ in shortest_path("Angela_Merkel", "2012_Nobel_Peace_Prize", cache)])