X-Git-Url: https://projects.mako.cc/source/mw/blobdiff_plain/c5075c4a66b4f0f461489f1cee62068b67e01ff0..e670001ebb36dc1592e94b60c271232e427f3d64:/src/mw/metadir.py diff --git a/src/mw/metadir.py b/src/mw/metadir.py index 49a91cf..4549a29 100644 --- a/src/mw/metadir.py +++ b/src/mw/metadir.py @@ -53,7 +53,7 @@ class Metadir(object): os.mkdir(self.location, 0755) # metadir versioning fd = file(os.path.join(self.location, 'version'), 'w') - fd.write('1') + fd.write('1') # XXX THIS API VERSION NOT LOCKED IN YET fd.close() # create config self.config = ConfigParser.RawConfigParser() @@ -79,14 +79,24 @@ class Metadir(object): fd.truncate() fd.close() - def pages_add_rev(self, pageid, rv): + def get_pageid_from_pagename(self, pagename): + fd = file(os.path.join(self.location, 'cache', 'pagedict'), 'r') + pagedict = json.loads(fd.read()) + if pagename in pagedict.keys(): + return pagedict[pagename] + else: + return None + + def pages_add_rv(self, pageid, rv): pagefile = os.path.join(self.location, 'cache', 'pages', str(pageid)) fd = file(pagefile, 'w+') - pagedata = json.loads(fd.read()) + pagedata_raw = fd.read() + if pagedata_raw == '': + pagedata = {} + else: + pagedata = json.loads(pagedata_raw) rvid = int(rv['revid']) - if pageid not in pagedata.keys(): - pagedata[pageid] = {} - pagedata[pageid][rvid] = { + pagedata[rvid] = { 'user': rv['user'], 'timestamp': rv['timestamp'], 'content': rv['*'], } @@ -94,3 +104,17 @@ class Metadir(object): fd.write(json.dumps(pagedata)) fd.truncate() fd.close() + + def pages_get_rv_list(self, pageid): + pagefile = os.path.join(self.location, 'cache', 'pages', str(pageid)) + fd = file(pagefile, 'r') + pagedata = json.loads(fd.read()) + rvs = [int(x) for x in pagedata.keys()] + rvs.sort() + return rvs + + def pages_get_rv(self, pageid, rvid): + pagefile = os.path.join(self.location, 'cache', 'pages', str(pageid)) + fd = file(pagefile, 'r') + pagedata = json.loads(fd.read()) + return pagedata[str(rvid)]