From: Yves Tennevin <esby@free.fr>
Date: Sun, 25 Apr 2010 09:52:47 +0000 (-0500)
Subject: Add CSRF support (MediaWiki bug 23076)
X-Git-Url: https://projects.mako.cc/source/mw/commitdiff_plain/e31cbf92b32521b46527e9b2a2bfecca408d05d1

Add CSRF support (MediaWiki bug 23076)

https://bugzilla.wikimedia.org/show_bug.cgi?id=23076
---

diff --git a/src/mw/clicommands.py b/src/mw/clicommands.py
index de08cb6..a42bcfb 100644
--- a/src/mw/clicommands.py
+++ b/src/mw/clicommands.py
@@ -57,6 +57,16 @@ class CommandBase(object):
         if result['login']['result'] == 'Success':
             # cookies are saved to a file
             print 'Login successful! (yay)'
+        elif result['login']['result'] == 'NeedToken':
+            print'Login with token'
+            result = self.api.call({'action': 'login',
+                               'lgname': user,
+                               'lgpassword': passwd,
+                               'lgtoken': result['login']['token']})
+            if result['login']['result'] == 'Success':
+                print 'Login successful! (yay)'
+    else:
+        print 'Login failed: %s' % result['login']['result']
         else:
             print 'Login failed: %s' % result['login']['result']