From d0e89ce145d159159ae8b73a1bf45a9de3aaefaf Mon Sep 17 00:00:00 2001 From: Ian Weller Date: Mon, 21 Dec 2009 20:16:23 -0600 Subject: [PATCH] Restructure metadir --- src/mw/metadir.py | 47 +++++++++++++++++++++++------------------------ 1 file changed, 23 insertions(+), 24 deletions(-) diff --git a/src/mw/metadir.py b/src/mw/metadir.py index 06dbeb3..ab99288 100644 --- a/src/mw/metadir.py +++ b/src/mw/metadir.py @@ -57,34 +57,33 @@ class Metadir(object): self.config.set('remote', 'api_url', api_url) with open(self.config_loc, 'wb') as config_file: self.config.write(config_file) - # create cache + # create cache/ os.mkdir(os.path.join(self.location, 'cache')) - # create cache/page - fd = file(os.path.join(self.location, 'cache', 'page'), 'w') - fd.write(json.dumps({})) - # create cache/rv - fd = file(os.path.join(self.location, 'cache', 'rv'), 'w') + # create cache/pagedict + fd = file(os.path.join(self.location, 'cache', 'pagedict'), 'w') fd.write(json.dumps({})) + # create cache/pages/ + os.mkdir(os.path.join(self.location, 'cache', 'pages'), 0755) - def add_page_info(self, pageid, pagename, rvids): - lulz = file(os.path.join(self.location, 'cache', 'page'), 'r') - conf = json.loads(lulz.read()) - conf[pageid] = {'name': pagename, 'rv': rvids} - fd = file(os.path.join(self.location, 'cache', 'page'), 'w') - fd.write(json.dumps(conf)) + def pagedict_add(self, pagename, pageid): + fd = file(os.path.join(self.location, 'cache', 'pagedict'), 'r+') + pagedict = json.loads(fd.read()) + pagedict[pagename] = int(pageid) + fd.seek(0) + fd.write(json.dumps(pagedict)) + fd.truncate() - def add_rv_info(self, rv): - lulz = file(os.path.join(self.location, 'cache', 'rv'), 'r') - conf = json.loads(lulz.read()) + def pages_add_rev(self, pageid, rv): + pagefile = os.path.join(self.location, 'cache', 'pages', str(pageid)) + fd = file(pagefile, 'w+') + pagedata = json.loads(fd.read()) rvid = int(rv['revid']) - conf[rvid] = { + if pageid not in pagedata.keys(): + pagedata[pageid] = {} + pagedata[pageid][rvid] = { 'user': rv['user'], 'timestamp': rv['timestamp'], - 'content': rv['*'] + 'content': rv['*'], } - conf[rvid]['minor'] = 'minor' in rv - if 'comment' in rv: - conf[rvid]['comment'] = rv['comment'] - else: - conf[rvid]['comment'] = None - fd = file(os.path.join(self.location, 'cache', 'rv'), 'w') - fd.write(json.dumps(conf)) + fd.seek(0) + fd.write(json.dumps(pagedata)) + fd.truncate() -- 2.39.5