http://www.mediawiki.org/wiki/Special:Code/pywikipedia/10276

Revision: 10276 Author: xqt Date: 2012-06-03 08:56:51 +0000 (Sun, 03 Jun 2012) Log Message: ----------- copy script part of the login script and keep the library on pwb folder

Added Paths: ----------- branches/rewrite/scripts/login.py

Copied: branches/rewrite/scripts/login.py (from rev 10274, branches/rewrite/pywikibot/login.py) =================================================================== --- branches/rewrite/scripts/login.py (rev 0) +++ branches/rewrite/scripts/login.py 2012-06-03 08:56:51 UTC (rev 10276) @@ -0,0 +1,281 @@ +#!/usr/bin/python +# -*- coding: utf-8 -*- +""" +Script to log the robot in to a wiki account. + +Suggestion is to make a special account to use for robot use only. Make +sure this robot account is well known on your home wiki before using. + +Parameters: + + -all Try to log in on all sites where a username is defined in + user-config.py. + + + -force Ignores if the user is already logged in, and tries to log in. + + -pass Useful in combination with -all when you have accounts for + several sites and use the same password for all of them. + Asks you for the password, then logs in on all given sites. + + -pass:XXXX Uses XXXX as password. Be careful if you use this + parameter because your password will be shown on your + screen, and will probably be saved in your command line + history. This is NOT RECOMMENDED for use on computers + where others have either physical or remote access. + Use -pass instead. + + -sysop Log in with your sysop account. + +If not given as parameter, the script will ask for your username and +password (password entry will be hidden), log in to your home wiki using +this combination, and store the resulting cookies (containing your password +hash, so keep it secured!) in a file in the data subdirectory. + +All scripts in this library will be looking for this cookie file and will +use the login information if it is present. + +To log out, throw away the *.lwp file that is created in the data +subdirectory. +""" +# +# (C) Rob W.W. Hooft, 2003 +# (C) Pywikipedia bot team, 2003-2010 +# +# Distributed under the terms of the MIT license. +# +__version__ = '$Id$' + +import logging +import pywikibot +from pywikibot import config, deprecate_arg +from pywikibot.exceptions import NoSuchSite, NoUsername + +_logger = "wiki.login" + + +# On some wikis you are only allowed to run a bot if there is a link to +# the bot's user page in a specific list. +# If bots are listed in a template, the templates name must be given as +# second parameter, otherwise it must be None +botList = { + 'wikipedia': { + 'en': [u'Wikipedia:Bots/Status', 'BotS'], + 'simple': [u'Wikipedia:Bots', '/links'] + }, + 'gentoo': { + 'en': [u'Help:Bots', None], + } +} + + +class LoginManager: + @deprecate_arg("username", "user") + @deprecate_arg("verbose", None) + def __init__(self, password=None, sysop=False, site=None, user=None): + if site is not None: + self.site = site + else: + self.site = pywikibot.Site() + if user: + self.username = user + elif sysop: + try: + self.username = config.sysopnames\ + [self.site.family.name][self.site.code] + except KeyError: + raise NoUsername( +u"""ERROR: Sysop username for %(fam_name)s:%(wiki_code)s is undefined. +If you have a sysop account for that site, please add a line to user-config.py: + +sysopnames['%(fam_name)s']['%(wiki_code)s'] = 'myUsername'""" + % {'fam_name': self.site.family.name, + 'wiki_code': self.site.code}) + else: + try: + self.username = config.usernames\ + [self.site.family.name][self.site.code] + except: + raise NoUsername( +u"""ERROR: Username for %(fam_name)s:%(wiki_code)s is undefined. +If you have an account for that site, please add a line to user-config.py: + +usernames['%(fam_name)s']['%(wiki_code)s'] = 'myUsername'""" + % {'fam_name': self.site.family.name, + 'wiki_code': self.site.code}) + self.password = password + if getattr(config, 'password_file', ''): + self.readPassword() + + def botAllowed(self): + """ + Checks whether the bot is listed on a specific page to comply with + the policy on the respective wiki. + """ + if self.site.family.name in botList \ + and self.site.code in botList[self.site.family.name]: + botListPageTitle, botTemplate = botList[self.site.family.name][self.site.code] + botListPage = pywikibot.Page(self.site, botListPageTitle) + if botTemplate: + for template in botListPage.templatesWithParams(): + if template[0] == botTemplate \ + and template[1][0] == self.username: + return True + else: + for linkedPage in botListPage.linkedPages(): + if linkedPage.title(withNamespace=False) == self.username: + return True + return False + else: + # No bot policies on other sites + return True + + def getCookie(self, remember=True, captcha = None): + """ + Login to the site. + + remember Remember login (default: True) + captchaId A dictionary containing the captcha id and answer, if any + + Returns cookie data if succesful, None otherwise. + """ + # NOT IMPLEMENTED - see data/api.py for implementation + + def storecookiedata(self, data): + """ + Store cookie data. + + The argument data is the raw data, as returned by getCookie(). + + Returns nothing. + """ + # THIS IS OVERRIDDEN IN data/api.py + filename = config.datafilepath('pywikibot.lwp') + pywikibot.debug(u"Storing cookies to %s" % filename, + _logger) + f = open(filename, 'w') + f.write(data) + f.close() + + def readPassword(self): + """ + Read passwords from a file. + + DO NOT FORGET TO REMOVE READ ACCESS FOR OTHER USERS!!! + Use chmod 600 password-file. + All lines below should be valid Python tuples in the form + (code, family, username, password) or (username, password) + to set a default password for an username. Default usernames + should occur above specific usernames. + + Example: + + ("my_username", "my_default_password") + ("my_sysop_user", "my_sysop_password") + ("en", "wikipedia", "my_en_user", "my_en_pass") + """ + password_f = open(config.password_file) + for line in password_f: + if not line.strip(): continue + entry = eval(line) + if len(entry) == 2: #for default userinfo + if entry[0] == self.username: self.password = entry[1] + elif len(entry) == 4: #for userinfo included code and family + if entry[0] == self.site.code and \ + entry[1] == self.site.family.name and \ + entry[2] == self.username: + self.password = entry[3] + password_f.close() + + def login(self, retry = False): + if not self.password: + # As we don't want the password to appear on the screen, we set + # password = True + self.password = pywikibot.input( + u'Password for user %(name)s on %(site)s:' + % {'name': self.username, 'site': self.site}, + password = True) + +# self.password = self.password.encode(self.site.encoding()) + + pywikibot.output(u"Logging in to %(site)s as %(name)s" + % {'name': self.username, 'site': self.site}) + try: + cookiedata = self.getCookie() + except pywikibot.data.api.APIError, e: + pywikibot.error(u"Login failed (%s)." % e.code) + if retry: + self.password = None + return self.login(retry = True) + else: + return False + self.storecookiedata(cookiedata) + pywikibot.log(u"Should be logged in now") +## # Show a warning according to the local bot policy +## FIXME: disabled due to recursion; need to move this to the Site object after +## login +## if not self.botAllowed(): +## logger.error( +## u"Username '%(name)s' is not listed on [[%(page)s]]." +## % {'name': self.username, +## 'page': botList[self.site.family.name][self.site.code]}) +## logger.error( +##"Please make sure you are allowed to use the robot before actually using it!") +## return False + return True + + def showCaptchaWindow(self, url): + pass + +def main(*args): + password = None + sysop = False + logall = False + forceLogin = False + for arg in pywikibot.handleArgs(*args): + if arg.startswith("-pass"): + if len(arg) == 5: + password = pywikibot.input(u'Password for all accounts:', + password = True) + else: + password = arg[6:] + elif arg == "-sysop": + sysop = True + elif arg == "-all": + logall = True + elif arg == "-force": + forceLogin = True + else: + pywikibot.showHelp('login') + return + if logall: + if sysop: + namedict = config.sysopnames + else: + namedict = config.usernames + for familyName in namedict: + for lang in namedict[familyName]: + try: + site = pywikibot.getSite(code=lang, fam=familyName) + if site.logged_in(sysop) \ + and site.user() == site.username(sysop): + pywikibot.output(u"Login successful on %(site)s." % locals()) + else: + pywikibot.output(u"Not logged in on %(site)s." % locals()) + except NoSuchSite: + pywikibot.output( + lang + u'.' + familyName + +u' is not a valid site, please remove it from your config') + + else: + site = pywikibot.Site(sysop=sysop) + if site.logged_in(sysop) and site.user() == site.username(sysop): + pywikibot.output(u"Login successful on %(site)s." % locals()) + else: + pywikibot.output(u"Not logged in on %(site)s." % locals()) + +if __name__ == "__main__": + try: + main() + finally: + pywikibot.stopme()