From: Ian Weller Date: Tue, 22 Dec 2009 08:34:27 +0000 (-0600) Subject: status: Add command X-Git-Url: https://projects.mako.cc/source/mw/commitdiff_plain/ac7e90d463ab412375247a74083fe8f7385729b5?ds=sidebyside status: Add command --- diff --git a/src/mw/clicommands.py b/src/mw/clicommands.py index 5bddd62..7ead096 100644 --- a/src/mw/clicommands.py +++ b/src/mw/clicommands.py @@ -16,6 +16,7 @@ # with this program. If not, see . ### +import codecs import getpass import mw.api import mw.metadir @@ -104,3 +105,32 @@ class FetchCommand(CommandBase): fd = file(os.path.join(self.metadir.root, filename + '.wiki'), 'w') fd.write(response[pageid]['revisions'][0]['*'].encode('utf-8')) + + +class StatusCommand(CommandBase): + def __init__(self): + CommandBase.__init__(self, 'status', 'check repo status') + self.shortcuts.append('st') + + def _do_command(self): + self._die_if_no_init() + check = [] + for root, dirs, files in os.walk(self.metadir.root): + if root == self.metadir.root: + dirs.remove('.mw') + for name in files: + check.append(os.path.join(root, name)) + check.sort() + for full in check: + name = os.path.split(full)[1] + if name[-5:] == '.wiki': + pagename = mw.api.filename_to_pagename(name[:-5]) + pageid = self.metadir.get_pageid_from_pagename(pagename) + if not pageid: + print '? %s' % os.path.relpath(full, self.metadir.root) + else: + rvid = self.metadir.pages_get_rv_list(pageid)[-1] + rv = self.metadir.pages_get_rv(pageid, rvid) + cur_content = codecs.open(full, 'r', 'utf-8').read() + if cur_content != rv['content']: + print 'U %s' % os.path.relpath(full, self.metadir.root) diff --git a/src/mw/metadir.py b/src/mw/metadir.py index 377e048..66648e7 100644 --- a/src/mw/metadir.py +++ b/src/mw/metadir.py @@ -79,6 +79,14 @@ class Metadir(object): fd.truncate() fd.close() + 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+') @@ -96,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)]