don't need to have an account on mediawiki

This commit is contained in:
Romain Bignon 2011-02-19 18:54:26 +01:00
commit 10beaf2c5d
2 changed files with 18 additions and 22 deletions

View file

@ -28,13 +28,14 @@ __all__ = ['MediawikiBackend']
class MediawikiBackend(BaseBackend, ICapContent): class MediawikiBackend(BaseBackend, ICapContent):
NAME = 'mediawiki' NAME = 'mediawiki'
MAINTAINER = u'Clément Schreiner' MAINTAINER = u'Clément Schreiner'
EMAIL = '0.6' EMAIL = 'clemux@clemux.info'
VERSION = '0.6'
LICENSE = 'GPLv3' LICENSE = 'GPLv3'
DESCRIPTION = 'Mediawiki wiki software application' DESCRIPTION = 'Mediawiki wiki software application'
CONFIG = ValuesDict(Value('url', label='URL of the Mediawiki website'), CONFIG = ValuesDict(Value('url', label='URL of the Mediawiki website'),
Value('apiurl', label='URL of the Mediawiki website\'s API'), Value('apiurl', label='URL of the Mediawiki website\'s API'),
Value('username', label='Login'), Value('username', label='Login', default=''),
Value('password', label='Password', masked=True)) Value('password', label='Password', default='', masked=True))
BROWSER = MediawikiBrowser BROWSER = MediawikiBrowser
def create_default_browser(self): def create_default_browser(self):
@ -53,7 +54,6 @@ class MediawikiBackend(BaseBackend, ICapContent):
def iter_revisions(self, _id, max_results=10): def iter_revisions(self, _id, max_results=10):
for rev in self.browser.iter_wiki_revisions(_id, max_results): for rev in self.browser.iter_wiki_revisions(_id, max_results):
yield rev yield rev
def push_content(self, content, message=None, minor=False): def push_content(self, content, message=None, minor=False):
self.browser.set_wiki_source(content, message, minor) self.browser.set_wiki_source(content, message, minor)

View file

@ -35,7 +35,7 @@ __all__ = ['MediawikiBrowser']
# Browser # Browser
class MediawikiBrowser(BaseBrowser): class MediawikiBrowser(BaseBrowser):
ENCODING = 'utf-8' ENCODING = 'utf-8'
def __init__(self, url, apiurl, *args, **kwargs): def __init__(self, url, apiurl, *args, **kwargs):
url_parsed = urlsplit(url) url_parsed = urlsplit(url)
self.PROTOCOL = url_parsed.scheme self.PROTOCOL = url_parsed.scheme
@ -43,7 +43,7 @@ class MediawikiBrowser(BaseBrowser):
self.BASEPATH = url_parsed.path self.BASEPATH = url_parsed.path
if self.BASEPATH.endswith('/'): if self.BASEPATH.endswith('/'):
self.BASEPATH = self.BASEPATH[:-1] self.BASEPATH = self.BASEPATH[:-1]
prefix = '%s://%s%s' % (self.PROTOCOL, self.DOMAIN, self.BASEPATH) prefix = '%s://%s%s' % (self.PROTOCOL, self.DOMAIN, self.BASEPATH)
self.apiurl = apiurl self.apiurl = apiurl
@ -51,7 +51,7 @@ class MediawikiBrowser(BaseBrowser):
def get_wiki_source(self, page): def get_wiki_source(self, page):
assert isinstance(self.apiurl, basestring) assert isinstance(self.apiurl, basestring)
data = {'action': 'query', data = {'action': 'query',
'prop': 'revisions|info', 'prop': 'revisions|info',
'titles': page, 'titles': page,
@ -61,7 +61,7 @@ class MediawikiBrowser(BaseBrowser):
} }
result = self.API_get(data) result = self.API_get(data)
pageid = result['query']['pages'].keys()[0] pageid = result['query']['pages'].keys()[0]
if pageid == "-1": if pageid == "-1":
@ -70,7 +70,7 @@ class MediawikiBrowser(BaseBrowser):
def get_token(self, page, _type): def get_token(self, page, _type):
''' _type can be edit, delete, protect, move, block, unblock, email or import''' ''' _type can be edit, delete, protect, move, block, unblock, email or import'''
if not self.is_logged(): if len(self.username) > 0 and not self.is_logged():
self.login() self.login()
data = {'action': 'query', data = {'action': 'query',
@ -84,14 +84,14 @@ class MediawikiBrowser(BaseBrowser):
return None return None
return result['query']['pages'][str(pageid)][_type+'token'] return result['query']['pages'][str(pageid)][_type+'token']
def set_wiki_source(self, content, message=None, minor=False): def set_wiki_source(self, content, message=None, minor=False):
if not self.is_logged(): if len(self.username) > 0 and not self.is_logged():
self.login() self.login()
page = content.id page = content.id
token = self.get_token(page, 'edit') token = self.get_token(page, 'edit')
data = {'action': 'edit', data = {'action': 'edit',
'title': page, 'title': page,
'token': token, 'token': token,
@ -111,8 +111,6 @@ class MediawikiBrowser(BaseBrowser):
} }
result = self.API_post(data) result = self.API_post(data)
return result['parse']['text']['*'] return result['parse']['text']['*']
def is_logged(self): def is_logged(self):
data = {'action': 'query', data = {'action': 'query',
@ -120,7 +118,7 @@ class MediawikiBrowser(BaseBrowser):
} }
result = self.API_get(data) result = self.API_get(data)
return result['query']['userinfo']['id'] != 0 return result['query']['userinfo']['id'] != 0
def login(self): def login(self):
assert isinstance(self.username, basestring) assert isinstance(self.username, basestring)
assert isinstance(self.password, basestring) assert isinstance(self.password, basestring)
@ -132,15 +130,15 @@ class MediawikiBrowser(BaseBrowser):
} }
result = self.API_post(data) result = self.API_post(data)
if result['login']['result'] == 'WrongPass': if result['login']['result'] == 'WrongPass':
raise BrowserIncorrectPassword raise BrowserIncorrectPassword()
if result['login']['result'] == 'NeedToken': if result['login']['result'] == 'NeedToken':
data['lgtoken'] = result['login']['token'] data['lgtoken'] = result['login']['token']
result2 = self.API_post(data) result2 = self.API_post(data)
def iter_wiki_revisions(self, page, nb_entries): def iter_wiki_revisions(self, page, nb_entries):
'''Yield 'Revision' objects for the last <nb_entries> revisions of the specified page.''' '''Yield 'Revision' objects for the last <nb_entries> revisions of the specified page.'''
if not self.is_logged(): if len(self.username) > 0 and not self.is_logged():
self.login() self.login()
data = {'action': 'query', data = {'action': 'query',
'titles': page, 'titles': page,
@ -151,7 +149,7 @@ class MediawikiBrowser(BaseBrowser):
result = self.API_get(data) result = self.API_get(data)
pageid = str(result['query']['pages'].keys()[0]) pageid = str(result['query']['pages'].keys()[0])
for rev in result['query']['pages'][pageid]['revisions']: for rev in result['query']['pages'][pageid]['revisions']:
rev_content = Revision(str(rev['revid'])) rev_content = Revision(str(rev['revid']))
rev_content.comment = rev['comment'] rev_content.comment = rev['comment']
@ -163,8 +161,6 @@ class MediawikiBrowser(BaseBrowser):
else: else:
rev_content.minor = False rev_content.minor = False
yield rev_content yield rev_content
def home(self): def home(self):
'''We don't need to change location, we're using the JSON API here.''' '''We don't need to change location, we're using the JSON API here.'''