2025-summer/team-3
12
Fork 0
Code Issues Pull requests Projects Releases Activity

add caching to Graph.load()

Browse source
This commit is contained in:
Laura Klünder 2016-12-16 11:38:25 +01:00
parent 6cc8c27ffa
commit 55cac31bbe
1 changed files with 18 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

18
src/c3nav/routing/graph.py
View file

@ -12,6 +12,8 @@ from c3nav.routing.point import GraphPoint
class Graph: class Graph:
graph_cached = None
graph_cached_date = None
default_filename = os.path.join(settings.DATA_DIR, 'graph.pickle') default_filename = os.path.join(settings.DATA_DIR, 'graph.pickle')
def __init__(self): def __init__(self):
@ -132,10 +134,26 @@ class Graph:
@classmethod @classmethod
def load(cls, filename=None): def load(cls, filename=None):
do_cache = False
if filename is None: if filename is None:
do_cache = True
filename = cls.default_filename filename = cls.default_filename
graph_mtime = None
if do_cache:
graph_mtime = os.path.getmtime(filename)
if cls.graph_cached is not None:
if cls.graph_cached_date == graph_mtime:
return cls.graph_cached
with open(filename, 'rb') as f: with open(filename, 'rb') as f:
graph = cls.unserialize(pickle.load(f)) graph = cls.unserialize(pickle.load(f))
if do_cache:
cls.graph_cached_date = graph_mtime
cls.graph_cached = graph
print(cls.graph_cached, cls.graph_cached_date)
graph.print_stats() graph.print_stats()
return graph return graph