Skip to content
This repository was archived by the owner on Jun 15, 2018. It is now read-only.
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 6 additions & 1 deletion simplemediawiki.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,10 +99,11 @@ class MediaWiki(object):
_psuedo_namespaces = None

def __init__(self, api_url, cookie_file=None, cookiejar=None,
user_agent=DEFAULT_UA, http_user=None, http_password=None):
user_agent=DEFAULT_UA, http_user=None, http_password=None, domain=None):
self._api_url = api_url
self._http_user = http_user
self._http_password = http_password
self.domain = domain
if cookiejar:
self._cj = cookiejar
elif cookie_file:
Expand Down Expand Up @@ -239,13 +240,17 @@ def do_login(self, user, passwd, token=None):
'lgpassword': passwd}
if token:
data['lgtoken'] = token
if self.domain:
data['lgdomain'] = self.domain

result = self.call(data)
if result['login']['result'] == 'Success':
self._high_limits = None
return True
elif result['login']['result'] == 'NeedToken' and not token:
return do_login(self, user, passwd, result['login']['token'])
else:
print("Login Failed: %s" % result['login']['result'])
return False

return do_login(self, user, passwd)
Expand Down