From a95e94936deb93d07b2641933c55ab18ea38727e Mon Sep 17 00:00:00 2001 From: Bryan Schumaker Date: Sun, 14 Nov 2010 17:49:23 -0500 Subject: [PATCH] Load preference trees Preference trees should be able to load from disk if associated with a file. --- libsaria/trees.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/libsaria/trees.py b/libsaria/trees.py index 170604dd..af275873 100644 --- a/libsaria/trees.py +++ b/libsaria/trees.py @@ -226,8 +226,16 @@ class PersPrefTree(PrefTree): def init_pref(self, pref, value): PrefTree.init_pref(self, pref, value) - data.save(self.file, ".tree") + if self.file != None: + data.save(self, self.file, ".tree") def set_pref(self, pref, value): PrefTree.set_pref(self, pref, value) - data.save(self.file, ".tree") + if self.file != None: + data.save(self, self.file, ".tree") + +def get_pref_tree(file): + tree = data.load(file, ".tree") + if tree == None: + tree = PersPrefTree(file) + return tree