#!/usr/bin/python
# -*- coding: utf-8 -*-
import pywikibot as w
import re
# Code for saving redirects and other pages
def savepage(page, text, BRFANo, summary = '', minor = False):
"""Save text to a page and log exceptions."""
if summary != '':
w.setAction(summary + '. See [[Wikipedia:Bots/Requests for approval/PotatoBot ' + BRFANo + '|approval]]. Report errors and suggestions at [[User talk:PotatoBot]].')
try:
if not '#' in page.title():
page.put(text, minorEdit = minor)
w.output(' \03{green}saving %s -> \03{gray}%s\03{default}' % (page.title(), text))
return ''
else:
w.output(' \03{red}cannot save %s because it is a section\03{default}' % page.title())
return '# %s: this is a secion title' % page.title(aslink=True)
except w.LockedPage:
w.output(' \03{red}cannot save %s because it is locked\03{default}' % page.title())
return '# %s: page was locked\n' % page.title(aslink=True)
except w.EditConflict:
w.output(' \03{red}cannot save %s because of edit conflict\03{default}' % page.title())
return '# %s: edit conflict occurred\n' % page.title(aslink=True)
except w.SpamfilterError, error:
w.output(' \03{red}cannot save %s because of spam blacklist entry %s\03{default}' % (page.title(), error.url))
return '# %s: spam blacklist entry\n' % page.title(aslink=True)
except:
w.output(' \03{red}unknown error on saving %s\03{default}' % page.title())
return '# %s: unknown error occurred\n' % page.title(aslink=True)
def resolveredir(page):
"""Return target if input is a redirect, else return input."""
try:
if page.isRedirectPage():
try:
w.output(' \03{gray}resolving redir %s to %s\03{default}'\
% (page.title(), page.getRedirectTarget().title()))
return page.getRedirectTarget()
except:
w.output(' \03{yellow}target %s is a broken redir\03{default}' % page.title())
return w.Page(w.getSite(), page.title() + ' (broken redirect)')
else:
return page
except:
w.output(' \03{yellow}target %s is a bad link\03{default}' % page.title())
return w.Page(w.getSite(), page.title() + ' (bad link)') # workaround for wikipedia.py breaking wikiasite: links
def makeredir(redirpage, page, BRFANo, templates = ''):
"""Create a redirect and log existing page that isn't a redirect to the desired article."""
page = resolveredir(page)
if redirpage.exists():
comment = ''
if redirpage.isDisambig():
comment = ' (disambiguation)'
dab = redirpage
if redirpage.isRedirectPage():
try:
if redirpage.getRedirectTarget().title() == page.title() or \
redirpage.getRedirectTarget().sectionFreeTitle() == page.title():
# Already a redir to the desired article
return ''
elif redirpage.getRedirectTarget().isDisambig():
comment = ' (redirect to disambiguation)'
dab = redirpage.getRedirectTarget()
else:
comment = ' (redirect)'
except:
comment = ' (broken redir)'
if 'disambiguation' in comment and page in [resolveredir(p) for p in dab.linkedPages()]:
w.output(' link to %s already on dab page %s' % (page.title(), redirpage.title()))
return ''
elif redirpage.title() != page.title():
w.output(' \03{yellow}redir to %s failed, page %s already exists\03{default}' % (page.title(), redirpage.title()))
return '# %s: redirecting to %s failed, page already exists%s\n' % (redirpage.title(aslink=True), page.title(aslink=True), comment)
else:
return ''
# Else create redirect, or write page name to list if an error occurs
else:
return savepage(redirpage, '#REDIRECT %s %s' % (page.title(aslink=True), templates), BRFANo, 'Redirect to ' + page.title(aslink=True))
def findATCs(page, includeVet = True):
"""Look for ATC codes in infoboxes."""
ATCvet, prefix, suffix, supp = False, '', '', ''
ATCvetpos, prefixpos, suffixpos, supppos = -1, -1, -1, -1
templatenames = ('Drugbox', 'Chembox Identifiers')
templates = page.templatesWithParams()
for tuple in templates:
if tuple[0] in templatenames:
idx = templatenames.index(tuple[0])
templatepos = templates.index(tuple)
for param in tuple[1]:
value = param.partition('=')
if value[0].strip() == 'ATCvet':
ATCvet = value[2].strip() == 'yes' and includeVet
ATCvetpos = tuple[1].index(param)
elif value[0].strip() == ('ATC_prefix', 'ATCCode_prefix')[idx] and value[2].strip().lower != 'none':
prefix = value[2].strip()
prefixpos = tuple[1].index(param)
elif value[0].strip() == ('ATC_suffix', 'ATCCode_suffix')[idx]:
suffix = value[2].strip()
suffixpos = tuple[1].index(param)
elif value[0].strip() == ('ATC_supplemental', 'ATC_Supplemental')[idx]:
supp = value[2].strip()
supppos = tuple[1].index(param)
codes = (prefix != '') * [(ATCvet*'Q' + prefix + suffix)]
for tupleSupp in page.templatesWithParams(supp):
if tupleSupp[0] in ['ATC', 'ATCvet']:
codes.append((tupleSupp[0] == 'ATCvet')*'Q' + tupleSupp[1][0] + tupleSupp[1][1])
return (codes, ATCvetpos, prefixpos, suffixpos, supppos)
def addTemplateParam(page, newtemplates, BRFANo, summary = 'Updating template', minor = False):
text = page.get()
oldtemplates = page.templatesWithParams()
pointer = 0
for i in range(len(oldtemplates)):
search1 = re.compile(r'\{\{\s*(%s|%s)%s\s*\|' % (oldtemplates[i][0][0].upper(), oldtemplates[i][0][0].lower(),\
oldtemplates[i][0].replace(' ', '( |_)'))).search(text, pointer)
if search1:
pointer = end() - 1
if newtemplates[i] != oldtemplates[i]:
if newtemplates[i][0].strip() == oldtemplates[i][0].strip():
for j in range(len(oldtemplates[i][1])):
oldparam = oldtemplates[i][1][j].partition('=')
newparam = newtemplates[i][1][j].partition('=')
# Todo: unnamed params #
if newparam[0] == oldparam[0]:
span = re.compile(r'\|\s*%s\s*=\s*([^|}\s]*)\s*(}|\|)' % oldparam[0]).\
search(text, pointer).span(1)
pointer = span(1)
if newparam[2].strip() != oldparam[2].strip():
text = text[:span(0)] + newparam[2] + text[span(1):]
else:
text = text[:] + newtemplates[i][1][j] + text[:]
pointer = len(text[:] + newtemplates[i][1][j])
else:
w.output('\03{yellow}template list does not match page %s: %s vs. %s\03{default}' % \
(page.title(), newtemplates[i][0].strip(), oldtemplates[i][0].strip()))
return '# %s: template list did not match templates on page' % page.title(aslink=True)
if text != page.get():
return savepage(page, text, BRFANo, summary, minor)
else:
return ''
def fmtdate(date):
"""Format date in English w style."""
return '%d %s %d' % (date.day, ('', 'January', 'February', 'March', 'April', 'May', 'June', 'July', 'August',\
'September', 'October', 'November', 'December')[date.month], date.year)
