diff --git a/exam/ex01/cfg.py b/exam/ex01/cfg.py index 15a65db..891f03d 100644 --- a/exam/ex01/cfg.py +++ b/exam/ex01/cfg.py @@ -1,6 +1,6 @@ config = { - "use_sqlite": True + "use_sqlite": False , "mysql_server": "172.17.0.2" , "mysql_user": "wikipedia" , "mysql_password": "wikipediastuff" diff --git a/exam/ex01/graph.py b/exam/ex01/graph.py index 24f7609..79a8eea 100644 --- a/exam/ex01/graph.py +++ b/exam/ex01/graph.py @@ -49,7 +49,13 @@ class DijkstraHelper(object): def write_back(self, connection): cursor = connection.cursor() cursor.execute(sql.statements["delete_dijkstra"]) - cursor.executemany(sql.statements["insert_dijkstra_values"], list(self._nodes.items())) + + def sqlize(v): + if(config["use_sqlite"]): + return v + if(v == float("inf")): + return 2147483647 + cursor.executemany(sql.statements["insert_dijkstra_values"], [(k, sqlize(v)) for k,v in self._nodes.items()]) diff --git a/exam/ex01/main.py b/exam/ex01/main.py index 7b7e165..4cacf25 100644 --- a/exam/ex01/main.py +++ b/exam/ex01/main.py @@ -29,4 +29,4 @@ angela = get_page_id("Angela_Merkel", cache) dijkstra = DijkstraHelper.from_db(cache) dijkstra.dijkstra(angela) dijkstra.write_back(cache) -print(dijkstra._nodes) +print({k:v for k,v in dijkstra._nodes.items() if v != float("inf")})