http://www.mediawiki.org/wiki/Special:Code/pywikipedia/11141
Revision: 11141
Author: xqt
Date: 2013-03-01 12:19:30 +0000 (Fri, 01 Mar 2013)
Log Message:
-----------
remove unused -dry option, use site.image_repository() as commons site,
skip third commonscat parameter at de-wiki (which should be handled further);
some doc cleanup
Modified Paths:
--------------
trunk/pywikipedia/commonscat.py
Modified: trunk/pywikipedia/commonscat.py
===================================================================
--- trunk/pywikipedia/commonscat.py 2013-02-28 17:18:29 UTC (rev 11140)
+++ trunk/pywikipedia/commonscat.py 2013-03-01 12:19:30 UTC (rev 11141)
@@ -205,13 +205,14 @@
'zh': u'機器人：更改 commonscat 連結，從 %(oldcat)s 至 %(newcat)s',
}
+
class CommonscatBot:
def __init__(self, generator, always, summary=None):
self.generator = generator
self.always = always
- self.dry = False
self.summary = summary
+ self.site = pywikibot.getSite()
def run(self):
for page in self.generator:
@@ -264,35 +265,34 @@
# show what was changed
pywikibot.showDiff(page.get(), text)
pywikibot.output(u'Comment: %s' %comment)
- if not self.dry:
- if not self.always:
- choice = pywikibot.inputChoice(
- u'Do you want to accept these changes?',
- ['Yes', 'No', 'Always', 'Quit'],
- ['y', 'N', 'a', 'q'], 'N')
- if choice == 'a':
- self.always = True
- elif choice == 'q':
- import sys
- sys.exit()
- if self.always or choice == 'y':
- try:
- # Save the page
- page.put(text, comment=comment,
- minorEdit=minorEdit, botflag=botflag)
- except pywikibot.LockedPage:
- pywikibot.output(u"Page %s is locked; skipping."
- % page.title(asLink=True))
- except pywikibot.EditConflict:
- pywikibot.output(
- u'Skipping %s because of edit conflict'
- % (page.title()))
- except pywikibot.SpamfilterError, error:
- pywikibot.output(
+ if not self.always:
+ choice = pywikibot.inputChoice(
+ u'Do you want to accept these changes?',
+ ['Yes', 'No', 'Always', 'Quit'],
+ ['y', 'N', 'a', 'q'], 'N')
+ if choice == 'a':
+ self.always = True
+ elif choice == 'q':
+ import sys
+ sys.exit()
+ if self.always or choice == 'y':
+ try:
+ # Save the page
+ page.put(text, comment=comment,
+ minorEdit=minorEdit, botflag=botflag)
+ except pywikibot.LockedPage:
+ pywikibot.output(u"Page %s is locked; skipping."
+ % page.title(asLink=True))
+ except pywikibot.EditConflict:
+ pywikibot.output(
+ u'Skipping %s because of edit conflict'
+ % (page.title()))
+ except pywikibot.SpamfilterError, error:
+ pywikibot.output(
u'Cannot change %s because of spam blacklist entry %s'
- % (page.title(), error.url))
- else:
- return True
+ % (page.title(), error.url))
+ else:
+ return True
return False
@classmethod
@@ -325,15 +325,15 @@
return False
def updateInterwiki (self, wikipediaPage = None, commonsPage = None):
- '''
- Update the interwiki's at commons from a wikipedia page. The bot just
- replaces the interwiki links at the commons page with the interwiki's from
- the wikipedia page. This should probably be more intelligent. We could use
- add all the interwiki's and remove duplicates. Or only remove language links
- if multiple language links to the same language exist.
+ '''Update the interwiki's at commons from a wikipedia page. The bot just
+ replaces the interwiki links at the commons page with the interwiki's
+ from the wikipedia page. This should probably be more intelligent. We
+ could use add all the interwiki's and remove duplicates. Or only remove
+ language links if multiple language links to the same language exist.
- This function is disabled for the moment untill i figure out what the best
- way is to update the interwiki's.
+ This function is disabled for the moment until i figure out what the
+ best way is to update the interwiki's.
+
'''
interwikis = {}
comment= u''
@@ -356,11 +356,11 @@
commonsPage.put(newtext=newtext, comment=comment)
def addCommonscat (self, page):
+ '''Take a page. Go to all the interwiki page looking for a commonscat
+ template. When all the interwiki's links are checked and a proper
+ category is found add it to the page.
+
'''
- Take a page. Go to all the interwiki page looking for a commonscat template.
- When all the interwiki's links are checked and a proper category is found
- add it to the page.
- '''
pywikibot.output(u'Working on ' + page.title());
#Get the right templates for this page
primaryCommonscat, commonscatAlternatives = self.getCommonscatTemplate(
@@ -416,6 +416,8 @@
newtemplate=u'', newcat=u'', linktitle=u'',
description=u''):
''' Change the current commonscat template and target. '''
+ if oldcat == '3=S' or linktitle == '3=S':
+ return #additional param on de-wiki, TODO: to be handled
if not linktitle and (page.title().lower() in oldcat.lower() or
oldcat.lower() in page.title().lower()):
linktitle = oldcat
@@ -501,12 +503,13 @@
if pywikibot.verbose:
pywikibot.output("getCommonscat: " + name )
try:
- commonsSite = pywikibot.getSite("commons", "commons")
+ commonsSite = self.site.image_repository()
#This can throw a pywikibot.BadTitle
commonsPage = pywikibot.Page(commonsSite, "Category:" + name)
if not commonsPage.exists():
- logpages = commonsSite.logpages(mode='delete', title=commonsPage.title())
+ logpages = commonsSite.logpages(mode='delete',
+ title=commonsPage.title())
try:
logitem = logpages.next()
(logpage, loguser, logtimestamp, logcomment) = logitem
@@ -519,15 +522,19 @@
elif m.group('newcat2'):
return self.checkCommonscatLink(m.group('newcat2'))
else:
- pywikibot.output(u'getCommonscat: Deleted by %s. Couldn\'t find move target in \" %s \"' % (loguser, logcomment))
+ pywikibot.output(
+ u'getCommonscat: Deleted by %s. Couldn\'t find move target in \" %s \"'
+ % (loguser, logcomment))
return u''
except StopIteration:
if pywikibot.verbose:
- pywikibot.output(u"getCommonscat: The category doesnt exist and nothing found in the deletion log.")
+ pywikibot.output(
+ u"getCommonscat: The category doesnt exist and nothing found in the deletion log.")
return u''
elif commonsPage.isRedirectPage():
if pywikibot.verbose:
- pywikibot.output(u"getCommonscat: The category is a redirect")
+ pywikibot.output(
+ u"getCommonscat: The category is a redirect")
return self.checkCommonscatLink(
commonsPage.getRedirectTarget().title(withNamespace=False))
elif "Category redirect" in commonsPage.templates():